[Java] 뒤에 있는 큰 수 찾기 - Lv2 프로그래머스

https://school.programmers.co.kr/learn/courses/30/lessons/154539

프로그래머스

 

풀이

---

Stack을 활용 ( 정리 링크 )

Stack

 

최근에 Stack에 관한 문제도 풀고 자료도 찾아보고 그래서 그런가 바로 생각이 났다.

import java.util.Stack;

class Solution {

    public int[] solution(int[] numbers) {
    	Stack<Integer> stack = new Stack<>();
    	for(int i=0; i<numbers.length; i++) {		
        	//stack에 idx값을 넣어 numbers배열에서 비교
    		while(!stack.empty() && numbers[stack.peek()] < numbers[i]) {
            		//가까운 큰 값으로 변경
    			numbers[stack.pop()] = numbers[i];   			
    		}
    		stack.push(i); //numbers idx값 넣기
    	}
        //남은 idx값은 더 큰녀석이 없음 -> -1
    	for(int i : stack) {
    		numbers[i] = -1;
    	}
    	
        return numbers;
    }
}