[Java] 백준 2493번: 탑

[문제 바로 가기] - 탑

💡 일단 생각하기!

Stack으로 풀이하는 문제인건 알았지만 처음부터 다 넣고 비교를 해서 시간초과가 떴다. 생각해보면 이 방식은 Stack으로 풀든 배열로 풀든 비슷한 시간이 나왔을텐데 Stack으로 푼다고 시간이 줄어들진 않았을텐데 😅

아무튼 다른 사람의 풀이를 참고해서 풀이했다.
처음에 입력 받을 때 입력 받은 값과 stack의 가장 위에 있는 값을 비교한다. 이때, 입력 받은 값이 더 작다면 가장 위에 있는 값의 인덱스를 출력하고 다음 입력 값으로 넘어간다. 더 크다면 가장 위에 있는 값을 pop한다.
stack이 비어 있을 경우에는 0을 출력하고 마지막엔 입력 받은 값을 push 해준다.

풀이

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


/**
 * 백준 2493번 탑
 * 
 * 하나씩 입력 받으면서 가장 위에 있는 값과 비교
 * 작을 경우 해당 위치 출력하고 종료
 * 클 경우 가장 위에 있는 값 pop
 * stack이 비었으면 0 출력하고 마지막에 stack에 입력 받은 값 push  
 */

public class Main {
	public static class Top {
		int value, height;
		
		public Top (int value, int height) {
			this.value = value;
			this.height = height;
		}
	}
	
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(br.readLine());
		
		StringTokenizer st = new StringTokenizer(br.readLine());
		StringBuilder sb = new StringBuilder();
		
		Stack<Top> stack = new Stack<>();
		
		for (int i = 0; i < N; i++) {
			int num = Integer.parseInt(st.nextToken());
			
			while (!stack.isEmpty()) {
				if (num <= stack.peek().value) {
					sb.append(stack.peek().height + " ");
					break;
				}
				stack.pop();
			}

			if (stack.isEmpty()) sb.append("0 ");
			
			stack.push(new Top(num, i + 1));
		}
		
		System.out.println(sb);
	}
}