[프로그래머스] 알고리즘 연습 문제 : 탑 (자바/JAVA)

https://programmers.co.kr/learn/courses/30/lessons/42588

알고리즘 연습 - 탑 | 프로그래머스

 

import java.util.Stack;
class Solution {
    public int[] solution(int[] heights) {
        int[] answer = new int[heights.length];
        Stack<Integer> stack = new Stack<>();
 
        for(int height : heights) {
            stack.push(height);
        }
 
        int heightsIndex = heights.length;
 
        while(heightsIndex >= 0) {
            heightsIndex--;
            for(int i = heightsIndex; i >= 0; i--) {
                if(heights[i] > stack.peek()) {
                    answer[heightsIndex] = i+1;
                    stack.pop();
                    break;
                }
                if(i == 0) {
                    answer[heightsIndex] = 0;
                    stack.pop();
                }
            }
        }
        return answer;
    }
}

 

stack을 굳이 안써도 될거 같은데, 문제의 카테고리가 스텍/큐 아래 있어서 굳이 써서 해결했다.