[JAVA] 백준 2493 탑 | Stack

 

~목차~

문제

문제 해결 포인트

작성 코드

---

 

문제

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

2493번: 탑

 

 

문제 해결 포인트

- Stack을 이용해 먼저 쌓인 값 중 (가장 가까운) 본인보다 큰 값 찾기

 

1. Stack이 비어 있는지 아닌지 판별

2. 비어있다면 0 출력 & 본인 push VS 비어있지 않다면 Stack에서 본인보다 큰 값을 찾을 때까지 pop

3. 본인보다 큰 값을 찾았다면 출력하고 본인 push

4. 1 ~ 3 반복

 

 

작성 코드

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

public class 탑2493 {

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		int n = Integer.parseInt(br.readLine());
		Stack<int[]> stack = new Stack<>();
		
		StringTokenizer st = new StringTokenizer(br.readLine());
		for(int i=0; i<n; i++) {
			int num = Integer.parseInt(st.nextToken());
			if(stack.isEmpty()) {
				sb.append("0 ");
				stack.push(new int[] {i+1, num});
			} else {
				while(true) {
					if(stack.isEmpty()) {
						sb.append("0 ");
						stack.push(new int[] {i+1, num});
						break;
					} else {
						if(stack.peek()[1] > num) {
							sb.append(stack.peek()[0] + " ");
							stack.push(new int[] {i+1, num});
							break;
						} else{
							stack.pop();
						}
					}
				}
			}
			
		}
		System.out.println(sb);

	}

}

 

 

 

도움이 되셨다면 좋아요 눌러주세요💚