[프로그래머스]스택/큐 : 주식가격 (level 2) (c++)

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

프로그래머스

 

 

방법 1: O(N^2)

#include <string>
#include <vector>
 
using namespace std;
 
vector<int> solution(vector<int> prices) {
 
    // 주어진 벡터 길이의 answer 벡터 생성
    vector<int> answer(prices.size());
 
    // prices 뒤에서부터 비교
    for (int i = prices.size() - 1; i >= 0; i--)
    {
        answer[i] = prices.size() - 1 - i;
        
        // 현재 요소보다 뒷 요소가 더 작으면 answer벡터 갱신
        for (int j = i + 1; j < prices.size(); j++)
        {
            if (prices[j] < prices[i])
            {
                answer[i] = j - i;
                break;
            }
        }
    }
 
    return answer;
}

 

방법 2: stack 사용 O(N)

#include <string>
#include <vector>
#include <stack>
 
using namespace std;
 
vector<int> solution(vector<int> prices) {
 
    // 주어진 벡터 길이의 answer 벡터 생성
    vector<int> answer(prices.size());
 
    // 아직 더 작은 값을 찾지 못한 idx의 스택
    stack<int> s;
 
    for (int i = 0; i < prices.size(); i++)
    {
        // 더 작은 값을 찾은 s.top은 answer벡터 입력 후 pop
        while (!s.empty() && prices[s.top()] > prices[i])
        {
            answer[s.top()] = i - s.top();
            s.pop();
        }
        s.push(i);
    }
 
    // 스택에 남아있는 인덱스들 처리 = 끝까지 더 작은 값을 찾지 못함
    while (!s.empty())
    {
        answer[s.top()] = prices.size() - 1 - s.top();
        s.pop();
    }
 
    return answer;
}

 

 

TIP

스택의 기준을 잡자

 

스택에 들어가야할 숫자들은 어떤 숫자들일까?

1. 아직 자기보다 더 큰 값을 못찾은 숫자?

2. 아직 자기보다 더 작은 값을 못찾은 숫자?

 

초 단위로 기록된 주식가격이 담긴 배열 prices가 매개변수로 주어질 때,
가격이 떨어지지 않은 기간은 몇 초인지를 return 하도록 solution 함수를 완성하세요.

 

문제에서 가격이 떨어지지 않은 기간을 구하라고 했으니,

자신보다 더 작은 값을 찾는 것이 관건인 문제다.

 

따라서 스택에 들어가야할 숫자는 

2. 아직 자기보다 더 작은 값을 못찾은 숫자 들이다.

그중에서도 인덱스를 넣는게 여러모로 편리할 것이다.

 

 

방법 2는 BOJ에서 풀었던 오등큰수 문제가 생각나는 풀이이다. 

 

 

 

사실 이 문제는 분류가 스택/큐로 되어있었지만,

직관적인 방법1 O(N^2) 풀이로 구현하는게 쉬웠다.

 

다른분의 스택 풀이를 보는 순간 아.. 하고 깨달음.