[SWEA] 5122 - 수열 편집

문제

N개의 10억 이하 자연수로 이뤄진 수열이 주어진다. 이 수열은 완성된 것이 아니라 M번의 편집을 거쳐 완성된다고 한다.

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

다음은 미완성 수열과 편집의 예이다.

[입력]

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

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

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

[출력]

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

풀이

편집 조건들을 order리스트에 저장해주고 첫번째 원소 알파벳에 따라 조건을 나눈다.

코드

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 _ in range(M):
        order = list(map(str, input().split()))
        if order[0] == 'I':
            a.insert(int(order[1]), int(order[2]))
        elif order[0] == 'D':
            del a[int(order[1])]
        elif order[0] == 'C':
            a[int(order[1])] = int(order[2])
    if L >= len(a):
        print(f'#{test_case} -1')
    else:
        print(f'#{test_case} {a[L]}')