주식가격 Programmers 프로그래머스 C++

문제 설명

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

제한사항

• prices의 각 가격은 1 이상 10,000 이하인 자연수입니다.
• prices의 길이는 2 이상 100,000 이하입니다.

입출력 예

prices	return
[1, 2, 3, 2, 3]	[4, 3, 1, 1, 0]

입출력 예 설명

• 1초 시점의 ₩1은 끝까지 가격이 떨어지지 않았습니다.
• 2초 시점의 ₩2은 끝까지 가격이 떨어지지 않았습니다.
• 3초 시점의 ₩3은 1초뒤에 가격이 떨어집니다. 따라서 1초간 가격이 떨어지지 않은 것으로 봅니다.
• 4초 시점의 ₩2은 1초간 가격이 떨어지지 않았습니다.
• 5초 시점의 ₩3은 0초간 가격이 떨어지지 않았습니다.

구현

출력 값이 vector이므로 prices를 돌면서 각 값들이 몇 초 뒤까지 가격이 안떨어지는지 정해서 answer에 저장해준다.

1. i번째 값이 자신을 제외한 prices의 값들 보다 작거나 같다면 카운트 해준다.
2. prices의 값이 자신 보다 크다면 카운트를 더해주고 멈춘다.
3. 해당 카운트를 answer에 저장해준다.

위와 같은 방법으로 코드를 구성하면 다음과 같이 나온다.
중요한 점은 prices의 값이 자신 보다 클 때 카운트를 더해주어야 한다는 점이다.

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> prices) {
    vector<int> answer;
    int len = prices.size();
    int cnt;
    for (int i = 0; i < len; i++) {
        cnt = 0;
        for (int j = i+1; j < len; j++) {
            if (prices[i] <= prices[j]) {
                cnt++;
            }
            else {
                cnt++;
                break;
            }
        }
        answer.push_back(cnt);
    }
    return answer;
}