초 단위로 기록된 주식가격이 담긴 배열 prices가 매개변수로 주어질 때, 가격이 떨어지지 않은 기간은 몇 초인지를 return 하도록 solution 함수를 완성하세요.
제한사항
prices의 각 가격은 1 이상 10,000 이하인 자연수입니다.
prices의 길이는 2 이상 100,000 이하입니다.
풀이
입출력 예시
1초 시점의 ₩1은 끝까지 가격이 떨어지지 않았습니다.
2초 시점의 ₩2은 끝까지 가격이 떨어지지 않았습니다.
3초 시점의 ₩3은 1초뒤에 가격이 떨어집니다. 따라서 1초간 가격이 떨어지지 않은 것으로 봅니다.
4초 시점의 ₩2은 1초간 가격이 떨어지지 않았습니다.
5초 시점의 ₩3은 0초간 가격이 떨어지지 않았습니다.
풀이
예시에도 잘 나와있듯, 가격이 떨어지지 않은 기간을 계산해야한다. 그 외에는 별다른 어려운 사항이 없다.
#include <string>
#include <vector>
using namespace std;
vector<int> solution(vector<int> prices) {
vector<int> answer;
for (int i = 0; i < prices.size() - 1; i++) {
int time = 0;
// 유지된 시간 계산
for (int j = i + 1; j < prices.size(); j++) {
time++;
if (prices[i] > prices[j]) { // 가격이 하락하면 루프 멈춤
break;
}
}
answer.push_back(time);
}
answer.push_back(0); // 마지막 가격은 지속시간 0초
return answer;
}
다음 가격이랑 비교하면서 1초씩 늘리다가, 가격이 낮아지는 순간 for 루프를 멈춰줬다. 마지막에 0을 추가한 이유는 마지막 가격은 유지시간이 항상 0초이기 때문이다.