[프로그래머스] 탑(stack, java)

탑 문제

 

 

stack을 이용해서 문제를 풀어보았다.

풀이는 아래 블로그를 참고하였다.

 

https://jhnyang.tistory.com/125

[프로그래머스 알고리즘 Level 2] 탑 문제 풀이 및 해설- 스택 큐

 

나의코드 :

 

import java.util.*;
 
class Solution {
    public int[] solution(int[] heights) {
        
        Stack<Integer> st = new Stack<Integer>();
        int t_len = heights.length;//number of tower;
        int[] answer = new int[t_len];
        //array to stack
        for(int i=0; i<t_len; i++){
            st.push(heights[i]);
        }        
        
        for(int i=0 ;i<t_len; i++){
            int target = st.pop();
            for(int j = t_len-2-i ;j>=0; j--){
                if(heights[j]>target){
                    answer[t_len-i-1] = j+1;
                    break;
                }
            }
        }
        return answer;
    }
}