[백준 - 2493번] 탑 - Java //Wello Horld//

 

이번에는 BOJ의 2493번 문제 "탑"을 풀어보도록 하자

 

 

Stack을 이용해서 풀면 간단하게 풀리는 문제이다. 일단, 탑의 높이(Height)와 그 때의 높이(Position)를 받아올 클래스를 만들어서 Stack이 해당 클래스를 받아오도록 만든다. 여기서 유의해야 될 점은 탑들의 높이는 1이상 100,000,000 이하의 정수이므로 long으로 받아와야 된다는 것과 탑들의 높이가 같을 때도 신호를 받을 수 있다는 점이다.

 

성공한 코드는 아래와 같다

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

public class sample {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int N = Integer.parseInt(br.readLine());
        Stack<tower> stack = new Stack<tower>();
        StringTokenizer st = new StringTokenizer(br.readLine());
        for(int i = 1; i <= N; i++){
            long a = Long.parseLong(st.nextToken());
            while(!stack.isEmpty()){
                if(stack.peek().height >= a){
                    bw.write(stack.peek().position + " ");
                    break;
                }
                stack.pop();
            }
            if(stack.isEmpty()) bw.write("0 ");
            stack.push(new tower(a, i));
        }

        bw.flush();
        br.close();
        bw.close();
    }
    static class tower{
        long height;
        int position;
        public tower(long height, int position){
            this.height = height;
            this.position = position;
        }
    }
}

 

문제 : https://www.acmicpc.net/problem/2493

2493번: 탑