[프로그래머스] 기능개발 JAVA

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

코딩테스트 연습 - 기능개발

 

스택을 사용한 문제

 

포인트는

입력받은 progresses를 stack에 거꾸로 넣어줘야한다.

peak가 앞에 문제가 되도록

 

풀이 

import java.util.*;

public class Solution {
    public int[] solution(int[] progresses, int[] speeds) {
        ArrayList<Integer> answerTemp = new ArrayList<>();
        Stack<Integer> progress = new Stack<>();
        Stack<Integer> speed = new Stack<>();

        for(int i = progresses.length-1; i>=0 ; i--){
            progress.push(progresses[i]);
            speed.push(speeds[i]);
        }
        while(!progress.isEmpty()){
            int cnt = 0;
            if(progress.peek() == 100){
                progress.pop();
                speed.pop();
                cnt++;
                while (!progress.isEmpty() && progress.peek() == 100) {
                    progress.pop();
                    speed.pop();
                    cnt++;
                }
            }else{
//                진도율 올려준
                for(int i=0; i<progress.size(); i++){
                        progress.set(i, Math.min(100, progress.get(i) + speed.get(i)));
                }
            }
            if(cnt > 0){
                answerTemp.add(cnt);
            }
        }
        int[] answer = new int[answerTemp.size()];
        for(int i = 0; i< answerTemp.size(); i++){
            answer[i] = answerTemp.get(i);
        }
        return answer;
    }
}