[SWEA] 1229 - 암호문2

문제

0 ~ 999999 사이의 수를 나열하여 만든 암호문이 있다. 암호문을 급히 수정해야 할 일이 발생했는데,

이 암호문은 특수 제작된 처리기로만 수정이 가능하다. 이 처리기는 다음과 같이 2개의 기능을 제공한다.

1. I(삽입) x, y, s : 앞에서부터 x의 위치 바로 다음에 y개의 숫자를 삽입한다. s는 덧붙일 숫자들이다.[ ex) I 3 2 123152 487651 ]

2. D(삭제) x, y : 앞에서부터 x의 위치 바로 다음부터 y개의 숫자를 삭제한다.[ ex) D 4 4 ]

위의 규칙에 맞게 작성된 명령어를 나열하여 만든 문자열이 주어졌을 때, 암호문을 수정하고, 수정된 결과의 처음 10개 숫자를 출력하는 프로그램을 작성하여라.

[입력]

첫 번째 줄 : 원본 암호문의 길이 N ( 100 ≤ N ≤ 200 의 정수)

두 번째 줄 : 원본 암호문

세 번째 줄 : 명령어의 개수 ( 10 ≤ N ≤ 20 의 정수)

네 번째 줄 : 명령어

위와 같은 네 줄이 한 개의 테스트 케이스이며, 총 10개의 테스트 케이스가 주어진다.

[출력]

#기호와 함께 테스트 케이스의 번호를 출력하고, 공백 문자 후 수정된 암호문의 처음 10개 항을 출력한다.

풀이

1228 암호문1에서 삭제 조건을 추가한다.
order를 순회하면서 order[idx] == 'D'이면 idx+2번 만큼 password[idx+1]를 pop시킨다.
암호문1 풀이 ->SWEA 1228 - 암호문1

코드

for test_case in range(1,11):
    N = int(input())
    password = list(map(int, input().split()))
    order_N = int(input())
    order = list(input().split())
    for i in range(len(order)):
        if order[i] == 'I':
            a = order[i+3:i+3+int(order[i+2])]
            for j in range(-1, -(len(a)+1), -1):
                password.insert(int(order[i+1]), a[j])
        elif order[i] == 'D':
            for idx in range(int(order[i+2])):
                password.pop(int(order[i+1]))
    print(f'#{test_case}', end=" ")
    for i in range(10):
        print(password[i], end=" ")
    print()