[Baekjoon Online Judge] 1874번: 스택 수열

1874번: 스택 수열

요구사항

 - 스택 (stack)은 기본적인 자료구조 중 하나로, 컴퓨터 프로그램을 작성할 때 자주 이용되는 개념이다.

 - 스택은 자료를 넣는 (push) 입구와 자료를 뽑는 (pop) 입구가 같아 제일 나중에 들어간 자료가 제일 먼저 나오는 (LIFO, Last in First out) 특성을 가지고 있다.

 - 1부터 n까지의 수를 스택에 넣었다가 뽑아 늘어놓음으로써, 하나의 수열을 만들 수 있다. 이때, 스택에 push하는 순서는 반드시 오름차순을 지키도록 한다고 하자.

 - 임의의 수열이 주어졌을 때 스택을 이용해 그 수열을 만들 수 있는지 없는지, 있다면 어떤 순서로 push와 pop 연산을 수행해야 하는지를 알아낼 수 있다.

 - 이를 계산하는 프로그램을 작성하라.

입력

 - 첫 줄에 n (1 ≤ n ≤ 100,000)이 주어진다. 둘째 줄부터 n개의 줄에는 수열을 이루는 1이상 n이하의 정수가 하나씩 순서대로 주어진다.

 - 물론 같은 정수가 두 번 나오는 일은 없다.

출력

 - 입력된 수열을 만들기 위해 필요한 연산을 한 줄에 한 개씩 출력한다.

 - push연산은 +로, pop 연산은 -로 표현하도록 한다. 불가능한 경우 NO를 출력한다.

---

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;

public class Baekjoon1874 {

    public static void main(String[] args) throws IOException {

        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));

        int n = Integer.parseInt(bufferedReader.readLine()); // 1 부터 n 까지의 수
        int[] nums = new int[n]; // 출력 순서를 확인할 수 있는 array

        for (int i = 0; i < n; i++)
            nums[i] = Integer.parseInt(bufferedReader.readLine());

        Stack<Integer> stack = new Stack<>();
        StringBuilder stringBuilder = new StringBuilder();
        int index = 0;
        for (int i = 1; i <= n; i++) {
            stack.push(i); // stack에 값을 채움
            stringBuilder.append("+\n");

            while (!stack.isEmpty() && stack.peek() == nums[index]) {
                stack.pop();
                index++;
                stringBuilder.append("-\n");
            }
        }

        if (!stack.isEmpty()) System.out.print("NO");
        else System.out.print(stringBuilder.toString());

        bufferedReader.close();
    }
}

 

입력으로 제공된 순서를 저장하는 nums int 배열을 사용했다. 1 부터 n의 수를 stack에 push한 뒤 peek으로 완성된 배열과 비교하여 확인하고 만족하면 stack을 pop한다. 

 

n 까지의 수 입력을 완료했고 가능한 상황이라면 stack이 비어있을 것이다. 간단하게 StringBuilder를 문자열로 출력한다.

비어있지 않다면 불가능한 경우이기 때문에 NO를 출력한다.

 

한 가지 의문인 점은 처음 문제를 해결할 때 nums 배열을 ArrayList로 사용하였다. 여러번 시도해보았지만 결국 통과하지 못하였고, 일반적인 배열을 사용하니 통과하게 되었다. 단순히 array를 편리하게 다루기 위해 시도하였는데 왜 이런 결과가 나오는지 이해가 잘 가지 않는다..

 

추가적으로 Stack은 모든 메소드들이 synchronized로 thread safe하기 때문에 다수의 쓰레드가 사용하게 되면 메소드를 lock, unlock하는 추가적인 시간이 소요된다. 이를 고려해야 하는 상황이 온다면 양쪽에서 값을 넣고 뺄 수 있는 Deque의 사용방법을 익혀두어야 겠다.