G2493. 탑

문제

링크

제출 코드

• 나는 이렇게 탑 높이와 번호를 스택을 따로 만들어서 구현했지만,

  Stack<Integer> stack_len = new Stack<Integer>();
  Stack<Integer> stack_idx = new Stack<Integer>();
  

• 교수님처럼 번호와 높이를 하나의 객체로 생성하여 하나의 스택을 사용하는 것이 더 깔끔한 듯 (top정보를 담는 클래스 만들고 이걸 <>안에 넣기)

  Stack<Tower> towers = new Stack<>();
  
  static class Tower{
  		int num, height;
  
  		Tower(int n, int h){
  			num = n;
  			height = h;
  		}
  	}
  

  
  

package gold;

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

public class G2493_top {

	public static void main(String[] args) throws NumberFormatException, IOException {

		Stack<Integer> stack_len = new Stack<Integer>();
		Stack<Integer> stack_idx = new Stack<Integer>();
		StringBuilder sb = new StringBuilder();

		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

		int N = Integer.parseInt(br.readLine());
		int[] num = Arrays.stream(br.readLine().split(" ")).mapToInt(Integer::parseInt).toArray();

		for(int n=0; n<N; n++) {
			while(!stack_len.empty() && stack_len.peek()<=num[n]) {
				stack_len.pop();
				stack_idx.pop();
			}
			if (stack_len.empty()) {
				sb.append(0);
			}else {
				sb.append(stack_idx.peek()+1);
			}
			stack_len.push(num[n]);
			stack_idx.push(n);
			if(n != N-1) sb.append(" ");
		}

		System.out.println(sb);
	}

}