[프로그래머스 알고리즘] 2단계 주식가격 문제

프로그래머스 2단계 주식가격 문제

 url : https://programmers.co.kr/learn/courses/30/lessons/42584?language=java

문제설명

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

제한사항

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

prices의 길이는 2 이상 100,000 이하입니다.

입출력 예

prices	return
[498,501,470,489]	[2,1,1,0]

입출력 예 설명

1초 시점의 ₩498은 2초간 가격을 유지하고, 3초 시점에 ₩470으로 떨어졌습니다.

2초 시점의 ₩501은 1초간 가격을 유지하고, 3초 시점에 ₩470으로 떨어졌습니다.

3초 시점의 ₩470은 최종 시점까지 총 1초간 가격이 떨어지지 않았습니다.

4초 시점의 ₩489은 최종 시점까지 총 0초간 가격이 떨어지지 않았습니다.

자바코드

class Solution {
    public static int[] solution(int[] prices) {
        int[] answer = new int[prices.length];
        for(int i = 0; i < answer.length; i++) {
            answer[i] = answer.length - 1 - i;
        }

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

        return answer;
    }
}

문제설명

처음 반복문에서는 각 시점에서 가질 수 있는 최대값을 초기 셋팅한다. (주식이 절대 떨어지지 않았을 때 가질 수 있는 시간(초))

다음에는 반복문을 돌면서 각 각 시점에서 주식의 가격이 떨어지는 순간을 배열에 넣는다. 두번째 반복문은 가격이 떨어지는 시점에만 

기록이 되기 때문에 첫번째 반복문으로 셋팅하는 작업이 반드시 필요! 

Pain Point

스택, 큐 문제로 나와있었지만, 반복문과 배열을 이용해 해결했다. 스택과 큐를 사용하였다면... 왠지 이중 for문을 사용하지 않아도 될거같다!

이 방법은 차후 더 고민해봐야겠다!