[SWEA - D3] 1230. 암호문3

 

 

나의 코드

for t in range(1, 11):
    N = int(input()) # 첫 번째 줄 : 원본 암호문의 길이 N 
    arr = list(map(int,input().split())) # 두 번째 줄 : 원본 암호문
    cmd = int(input()) # 세 번째 줄 : 명령어의 개수
    cmd_arr = list(input().split()) # 네 번째 줄 : 명령어
     
    for i in range(len(cmd_arr)):
        if cmd_arr[i] == 'A' :
            cnt = int(cmd_arr[i+1])
            for j in range(cnt):
                tmp = int(cmd_arr[i+2+j])
                arr.append(tmp)
         
        if cmd_arr[i] == 'I':
            idx = int(cmd_arr[i+1])
            cnt = int(cmd_arr[i+2])
            for j in range(cnt):
                tmp = int(cmd_arr[i+3+j])
                arr.insert(idx+j,tmp)
             
        if cmd_arr[i] == 'D':
            idx = int(cmd_arr[i+1])
            cnt = int(cmd_arr[i+2])
            for j in range(cnt):
                arr.pop(idx+1)
     
    result = arr[:10]
    print("#"+str(t) , *result)

 

 

입력받은 명령어를 List 로 생성

반복문으로 배열에 원소를 확인하여 

조건문으로 A,I,D 처리 

 

arr.append(x) : 맨끝에 요소를 추가
arr.insert(i,x) : i 번째 위치에 x 요소 추가 
arr.pop(x) : x 번째 요소 리턴 후 삭제