[SWEA] 5108 - 숫자 추가

ttaho·2022년 11월 9일

SWEA

목록 보기
8/38

문제

N개의 10억 이하 자연수로 이뤄진 수열이 주어진다.

이 수열은 완성된 것이 아니라 M개의 숫자를 지정된 위치에 추가하면 완성된다고 한다.

완성된 수열에서 인덱스 L의 데이터를 출력하는 프로그램을 작성하시오.

다음은 숫자를 추가하는 예이다.

[입력]

첫 줄에 테스트케이스의 수 T가 주어진다. 1<=T<=50

다음 줄부터 테스트 케이스의 별로 첫 줄에 수열의 길이 N, 추가 횟수 M, 출력할 인덱스 번호 L이 주어지고, 다음 줄에 수열이 주어진다.

그 다음 M개의 줄에 걸쳐 추가할 인덱스와 숫자 정보가 주어진다. 5<=N<=1000, 1<=M<=1000, 6<=L<=N+M

[출력]

각 줄마다 "#T" (T는 테스트 케이스 번호)를 출력한 뒤, 답을 출력한다.

풀이

인덱스와 숫자정보를 입력받아서 리스트에 insert(인덱스, 숫자정보)로 해당위치에 숫자를 넣는다.
연결리스트로 풀어야 할 것 같은데 다음에 해보기로 하자

코드

T = int(input())
for test_case in range(1, T+1):
    N, M, L = map(int,input().split())
    a = list(map(int, input().split()))
    for i in range(M):
        index, data = map(int, input().split())
        a.insert(index,data)
    print(f'#{test_case} {a[L]}')
profile
SW Engineer

0개의 댓글