[BOJ/Java] 2493. 탑

[BOJ/Java] 2493. 탑

2493번: 탑

---

풀이

package edu.ssafy.im.BOJ.Gold.G5.No2493;

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

public class Main {

    public static void main(String[] args) throws IOException {
        new Main().sol(); // Main 클래스의 sol 메서드 호출
    }

    private void sol() throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); // 입력을 받기 위한 BufferedReader 객체 생성
        StringBuilder sb = new StringBuilder(); // 출력 결과를 저장하기 위한 StringBuilder 객체 생성
        int n = Integer.parseInt(br.readLine()); // 탑의 수를 입력 받음
        StringTokenizer st = new StringTokenizer(br.readLine()); // 탑들의 높이를 공백을 기준으로 분리하여 StringTokenizer 생성
        Stack<Tower> stack = new Stack<>(); // 탑의 정보를 저장하기 위한 Stack 객체 생성
        for (int i = 1; i <= n; i++) { // 탑의 수만큼 반복
            int h = Integer.parseInt(st.nextToken()); // 현재 탑의 높이를 입력 받음
            if(stack.isEmpty()) { // 스택이 비어있는 경우
                sb.append("0 "); // 현재 탑에서 발사한 레이저 신호를 수신하는 탑이 없으므로 0을 추가하고 StringBuilder에 저장
                stack.push(new Tower(i, h)); // 현재 탑의 정보를 스택에 저장
            } else { // 스택이 비어있지 않은 경우
                while (true) { // 레이저 신호를 수신하는 탑을 찾을 때까지 반복
                    if (stack.isEmpty()) { // 스택이 비어있는 경우
                        sb.append("0 "); // 현재 탑에서 발사한 레이저 신호를 수신하는 탑이 없으므로 0을 추가하고 StringBuilder에 저장
                        stack.push(new Tower(i, h)); // 현재 탑의 정보를 스택에 저장
                        break; // 반복문 종료
                    }
                    Tower t = stack.peek(); // 스택의 맨 위에 있는 탑의 정보를 가져옴
                    if(t.height > h) { // 현재 탑의 높이보다 높은 탑이 스택에 있을 경우
                        sb.append(t.index + " "); // 스택에서 꺼낸 탑의 번호를 추가하고 StringBuilder에 저장
                        stack.push(new Tower(i, h)); // 현재 탑의 정보를 스택에 저장
                        break; // 반복문 종료
                    } else { // 현재 탑의 높이보다 낮은 탑이 스택에 있을 경우
                        stack.pop(); // 스택에서 탑을 제거하고 다음 탑을 확인하기 위해 반복문을 계속 진행
                    }
                }
            }
        }
        System.out.print(sb); // 결과 출력
    }

    class Tower { // 탑을 표현하는 클래스
        int index; // 탑의 번호
        int height; // 탑의 높이

        public Tower(int index, int height) { // 생성자
            this.index = index; // 탑의 번호 초기화
            this.height = height; // 탑의 높이 초기화
        }
    }
}