[C++/PGS] Lv.2 : 기능개발

 

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

프로그래머스

---

프로그래머스 레벨 2.

스택을 활용하여 해결해야 한다

 

처음에 제출했을 때 몇몇 개의 테스트케이스에서 실패가 떴는데,

아래 테케를 활용해서 통과할 수 있었다 ㅎ.ㅎ

스택 내부의 max값을 기억해두어야 한다!

 

 

 

나의 풀이

#include <string>
#include <vector>
#include <stack>
#include <iostream>
using namespace std;

vector<int> solution(vector<int> progresses, vector<int> speeds) {
    vector<int> answer;
    int n = progresses.size();
    int nowMax = 0;
    stack<int> st;
    
    for(int i=0; i<n; ++i){
        int left = 100 - progresses[i];
        int date = left / speeds[i];
        int before = false;
        if(left % speeds[i] != 0) date += 1;
        // cout << date <<"\n";
        if(st.empty()) {
            st.push(date);
            nowMax = date;
        }
        else{
            int size = st.size();
            // top이 더 크면 스택에 push
            if(nowMax >= date){
                st.push(date);
                continue;
            }
            // 맨 위의 숫자보다 크면 기존 내용을 모두 pop, count
            while(!st.empty()){
                st.pop();
            }
            answer.push_back(size);
            st.push(date);
            nowMax = date;
        }
    }
    if(!st.empty()){
        answer.push_back(st.size());
    }
    
    return answer;
}