[프로그래머스 42584번] 주식가격

문제

번호

주식가격(프로그래머스 42584번)

내용

초 단위로 기록된 주식가격이 담긴 배열 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초간 가격이 떨어지지 않았습니다.

언어

JAVA

 

---

풀이

이 문제가 스택이나 큐와 크게 관계가 있는가...에 대해서는 솔직히 잘 모르겠다.

 

문제는 단순하다.

계속해서 배열을 비교해나가면서 떨어진 순간까지의 count를 세는 것이다.

즉, <pre>prices</pre>에서 $a_i > a_j$ (단, $i < j$)인 순간까지의 count를 기록하는 것이다.

이 때문에 이중 for문을 사용하였다.

 

특이케이스 : 마지막에서 첫번째 & 두번째

특이한 케이스는 마지막에서 첫번째 것과 두번째 것이다.

우선, 마지막에서 첫번째 것은 무조건 0이다.

문제 내용 중 "입출력 예 설명"의 5초 시점에 의거한다.

마지막에서 두번째 것은 가격이 바로 떨어지면 1이고 떨어지지않는다 하더라도 더이상의 원소가 없기 때문에 항상 1이다.

 

이로인해 <pre>answer[total-2] = 1</pre>과 <pre>answer[total-1] = 0</pre> 라인이 추가되었다.

 

첫번째 케이스는 맞지만 코드 제출 후 나머지는 전부 틀릴 경우

본 케이스는 이주은님의 의견을 참고하였다.

1번 케이스만 통과할 경우에는 아래 케이스를 추가하면 무엇이 문제인지 알 수 있다.

INPUT : [5, 8, 6, 2, 4, 1]
OUTPUT : [3, 1, 1, 2, 1, 0]

---

코드

class Solution {
    public int[] solution(int[] prices) {
        int total = prices.length;
        int[] answer = new int[total];

        for (int i = 0 ; i < total-1 ; i++) {
            for(int j = i+1 ; j < total ; j++) {
                ++answer[i];
                if(prices[i] > prices[j]) break;
            }
        }

        answer[total-2] = 1;
        answer[total-1] = 0;

        return answer;
    }
}