티스토리 뷰

알고리즘/백준

백준 5430 (AC) - java

김다미김태리신시아 2023. 9. 26. 00:02

https://www.acmicpc.net/problem/5430

 

5430번: AC

각 테스트 케이스에 대해서, 입력으로 주어진 정수 배열에 함수를 수행한 결과를 출력한다. 만약, 에러가 발생한 경우에는 error를 출력한다.

www.acmicpc.net

문제 :

 

선영이는 주말에 할 일이 없어서 새로운 언어 AC를 만들었다. AC는 정수 배열에 연산을 하기 위해 만든 언어이다. 이 언어에는 두 가지 함수 R(뒤집기)과 D(버리기)가 있다.

함수 R은 배열에 있는 수의 순서를 뒤집는 함수이고, D는 첫 번째 수를 버리는 함수이다. 배열이 비어있는데 D를 사용한 경우에는 에러가 발생한다.

함수는 조합해서 한 번에 사용할 수 있다. 예를 들어, "AB"는 A를 수행한 다음에 바로 이어서 B를 수행하는 함수이다. 예를 들어, "RDD"는 배열을 뒤집은 다음 처음 두 수를 버리는 함수이다.

배열의 초기값과 수행할 함수가 주어졌을 때, 최종 결과를 구하는 프로그램을 작성하시오.

 

유형 : 자료구조(덱)

 

접근

  • 입력 형태가 조금 난해하다. 자바 언어의 경우 StringTokenizer 클래스의 delim을 활용하여 숫자만 뽑아내면 된다.
  • 연산의 형태가 R 즉 뒤집기라면 removeFirst -> removeLast로 삭제 방향을 바꿔주면 된다.

전체 코드

import java.util.*;
import java.io.*;

public class Main {
    static int n = 0;
    static Deque<Integer> q;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");
        n = Integer.parseInt(st.nextToken());

        StringBuilder sb = new StringBuilder();
        for(int i=1;i<=n;i++){
            char[] cal = br.readLine().toCharArray();

            int m = Integer.parseInt(br.readLine());
            st = new StringTokenizer(br.readLine(),",[]");
            q = new ArrayDeque<>();
            for(int j=1;j<=m;j++){
                q.add(Integer.parseInt(st.nextToken()));
            }

            boolean dir = true;
            boolean error = false;
            for(int j=0;j<cal.length;j++){
                if(cal[j] == 'R'){
                    dir = !dir;
                }

                else{
                    if(q.isEmpty()){
                        sb.append("error\n");
                        error = true;
                        break;
                    }

                    if(dir){
                        q.removeFirst();
                    }

                    else{
                        q.removeLast();
                    }
                }
            }

            if(!error){
                StringBuilder tmp = new StringBuilder();
                tmp.append('[');
                if(dir){
                    while(!q.isEmpty()){
                        tmp.append(q.removeFirst());

                        if(!q.isEmpty()){
                            tmp.append(",");
                        }
                    }
                }

                else{
                    while(!q.isEmpty()){
                        tmp.append(q.removeLast());

                        if(!q.isEmpty()){
                            tmp.append(",");
                        }
                    }
                }
                tmp.append("]\n");

                sb.append(tmp.toString());
            }
        }

        System.out.print(sb);
        br.close();
    }
}

'알고리즘 > 백준' 카테고리의 다른 글

백준 17396 (백도어) - java  (0) 2023.09.27
백준 1719 (택배) - java  (0) 2023.09.26
백준 1405 (미친 로봇) - java  (0) 2023.09.25
백준 2887 (행성 터널) - java  (0) 2023.09.21
백준 17822 (배열 돌리기) - java  (0) 2023.09.18