문제
0 ~ 999999 사이의 수를 나열하여 만든 암호문이 있다.
암호문을 급히 수정해야 할 일이 발생했는데, 이 암호문은 특수 제작된 처리기로만 수정이 가능하다.
이 처리기는 다음과 같이 3개의 기능을 제공한다.
-
I(삽입) x, y, s : 앞에서부터 x의 위치 바로 다음에 y개의 숫자를 삽입한다. s는 덧붙일 숫자들이다.[ ex) I 3 2 123152 487651 ]
-
D(삭제) x, y : 앞에서부터 x의 위치 바로 다음부터 y개의 숫자를 삭제한다.[ ex) D 4 4 ]
-
A(추가) y, s : 암호문의 맨 뒤에 y개의 숫자를 덧붙인다. s는 덧붙일 숫자들이다. [ ex) A 2 421257 796813 ]
위의 규칙에 맞게 작성된 명령어를 나열하여 만든 문자열이 주어졌을 때, 암호문을 수정하고, 수정된 결과의 처음 10개 숫자를 출력하는 프로그램을 작성하여라.
[입력]
첫 번째 줄 : 원본 암호문의 길이 N ( 2000 ≤ N ≤ 4000 의 정수)
두 번째 줄 : 원본 암호문
세 번째 줄 : 명령어의 개수 ( 250 ≤ M ≤ 500 의 정수)
네 번째 줄 : 명령어
위와 같은 네 줄이 한 개의 테스트 케이스이며, 총 10개의 테스트 케이스가 주어진다.
[출력]
#기호와 함께 테스트 케이스의 번호를 출력하고, 공백 문자 후 수정된 암호문의 처음 10개 항을 출력한다.
[제약 사항]
실행 시간 60ms 이하
풀이
SWEA 1229 - 암호문2에서 조건 A를 추가해준다.
암호문2 풀이 ->SWEA 1229 - 암호문2
order[idx] == 'A'이면 order[idx] 수 만큼 b리스트에 추가해준 후, 리스트 덧셈 성질을 사용하여 password += b를 해준다.
코드
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]))
elif order[i] == 'A':
b = order[i+2:i+2+int(order[i+1])]
password += b
print(f'#{test_case}', end=" ")
for i in range(10):
print(password[i], end=" ")
print()
