[BOJ] 2493. 탑

스택을 이용하여 비교하는 문제이다.

탑의 첫번째 부터 확인하며 스택이 비어있다면 0(벽), 만일 스택이 안비어있다면 스택의 top이 현재보다 크다면 top의 인덱스 출력 그렇지 않다면 top을 pop해주는 것을 반복해주며 최종적으로는 push를 계속 해주는 과정을 반복해주었다. 로직을 천천히 생각해봐야할 필요가 있는 문제였다.

package selfStudy.chap04;
 
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Stack;
import java.util.StringTokenizer;
 
public class Main_bj_2493_탑 {
 
    static class Tower{
        int index;
        long height;
 
        Tower(int index,long height){
            this.index=index;
            this.height=height;
        }
    }
 
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N= Integer.parseInt(br.readLine());
 
        Stack<Tower> stack = new Stack<>();
 
        StringTokenizer st = new StringTokenizer(br.readLine());
        StringBuilder sb = new StringBuilder();
 
        for(int i=1;i<=N;i++) {
            long height= Long.parseLong(st.nextToken());
 
            while(!stack.isEmpty()) {
                if(stack.peek().height>=height) {        //스택탑이 현재보다 크다면
                    sb.append(stack.peek().index+" ");
                    break;
                }else {    //현재가 스택탑보다 크다면
                    stack.pop();
                }
            }
            if(stack.isEmpty()) {
                sb.append("0 ");            
            }
            stack.push(new Tower(i,height));
 
        }
        System.out.println(sb.toString());
    }
}