SWEA_D3_1229

📁 [D3_1229] 암호문2

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.StringTokenizer;
 
public class Solution {
    static StringTokenizer st;
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
 
        for (int T = 1; T <= 10; T++) {
            sb.append("#");
            sb.append(T).append(" ");
 
            // 원본 암호문 길이
            int N = Integer.parseInt(br.readLine());
            st = new StringTokenizer(br.readLine(), " ");
 
            ArrayList<Integer> arr = new ArrayList<Integer>();
 
            // 원본 암호문
            for (int i = 0; i < N; i++) {
                arr.add(Integer.parseInt(st.nextToken()));
            }
 
            // 명령어 개수
            int cnt = Integer.parseInt(br.readLine());
 
            String[] arr_2 = new String[cnt];
 
            // 'I,D'를 경계로 나누기 -> 배열 arr_2에 저장
            st = new StringTokenizer(br.readLine(), "I|D");
            for (int i = 0; i < cnt; i++) {
                arr_2[i] = st.nextToken();
            }
 
            // y개의 숫자만큼 arr_int 배열에 저장
            for (int i = 0; i < cnt; i++) {
                String[] arr_int = arr_2[i].split(" ");
 
                // D인 경우 배열의 크기가 3 ( 공백, x, y)
                if (arr_int.length == 3) {
                    // 삭제할 위치 (x값)
                    int loc = Integer.parseInt(arr_int[1]);
                     
                    int temp = 0;
                    // x 위치에서 y 개수 만큼 삭제하기
                    for(int j=0; j<Integer.parseInt(arr_int[2]); j++) {
                        arr.remove(loc);
                         
                    }
                     
                }
 
                // I인경우 배열의 크기가 4이상
                else {
                    // 추가될 위치 (x값)
                    int loc = Integer.parseInt(arr_int[1]);
 
                    // loc위치 계속 증가해줄 변수
                    int temp = 0;
                    // x 위치에서 y 개수 만큼 추가시키기
                    for (int j = 3; j < 3 + Integer.parseInt(arr_int[2]); j++) {
                        arr.add(loc + temp, Integer.parseInt(arr_int[j]));
                        temp++;
                    }
                }
            }
 
            for (int i = 0; i < 10; i++) {
                sb.append(arr.get(i)).append(" ");
            }
            sb.append("\n");
        }
 
        System.out.println(sb);
 
    }
}

🤔 나의 생각

전체 값들을 ‘I’와 ‘D’를 경계로 나누고
‘D’ 배열의 특징은 길이가 3으로 고정인 것이다.
그 특징을 이용해 ‘I’는 추가를 ‘D’는 삭제를 하는 방식으로 작성했다.
여기서 약간 주의할 점은 ‘D’를 구현할 때 ArrayList는 삭제되면 한 칸씩 자동으로 댕겨진다. 그래서 위치를 옮겨줄 필요가 없다.