Categories: SWEA

SWEA 1230 암호문3 D3 (자바 Java)

java

SWEA 1230 암호문3

문제 정보

문제 링크

문제 출처 : SW Expert Academy
문제 번호 : 1230
문제 제목 : 암호문3
문제 난이도 : D3
풀이 언어 : Java

문제

0 ~ 999999 사이의 수로 표현되는 암호문이 있고, 이 암호문을 N개 모아 놓은 암호문 뭉치가 있다.

암호문 뭉치를 급히 수정해야 할 일이 발생했는데, 다음과 같이 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개 암호문을 출력하는 문제

접근

LinkedList 를 활용해 문제를 풀었다.

풀이

각 테스트 케이스에서, 암호문의 길이(N)를 입력 받는다. 그 후, LinkedList를 사용하여 입력 받은 숫자로 암호문을 초기화한다.
명령어의 수(commandCount)를 입력 받은 후, 각 명령어를 처리한다.
I (x, y, s): 위치 x에 y 개의 숫자를 삽입한다. 숫자들은 이어지는 입력으로부터 받는다.
D (x, y): 위치 x에서 시작하여 y 개의 숫자를 삭제한다.
A (y, s): 리스트의 끝에 y 개의 숫자를 추가한다. 추가할 숫자들은 이어지는 입력으로부터 받는다.
각 테스트 케이스의 결과로, 수정된 암호문의 처음 10개 숫자를 출력한다.

후기

LinkedList 를 활용한 여러 문제들을 더 풀어볼 계획이다.

코드

Java

import java.util.Scanner;
import java.io.FileInputStream;
import java.util.List;
import java.util.LinkedList;
 
class Solution
{
    public static void main(String args[]) throws Exception
    {
        Scanner sc = new Scanner(System.in);
        int T = 10;
         
        for(int test_case = 1; test_case <= T; test_case++)
        {
            int N = sc.nextInt();
            List<Integer> list = new LinkedList<>();
 
            for (int i = 0; i < N; i++) {
                list.add(sc.nextInt());
            }
 
            int commandCount = sc.nextInt();
 
 
            for (int i = 0; i < commandCount; i++) {
                String command = sc.next();
 
                switch (command) {
                    case "I":
                        int x = sc.nextInt();
                        int y = sc.nextInt();
                        for (int j = 0; j < y; j++) {
                            list.add(x + j, sc.nextInt());
                        }
                        break;
 
                    case "D":
                        x = sc.nextInt();
                        y = sc.nextInt();
                        for (int j = 0; j < y; j++) {
                            list.remove(x);
                        }
                        break;
 
                    case "A":
                        y = sc.nextInt();
                        for (int j = 0; j < y; j++) {
                            list.add(sc.nextInt());
                        }
                        break;
                }
            }
 
            System.out.print("#" + test_case + " ");
 
            for (int i = 0; i < 10 && i < list.size(); i++) {
                System.out.print(list.get(i) + " ");
            }
            System.out.println();
        }
    }
}