프로그래머스[Java] - (Level2)주식가격

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

프로그래머스

코드

class Solution {
    public int[] solution(int[] prices) {
        int[] answer = new int[prices.length];
        answer[prices.length-1] = 0;    // 마지막 가격은 떨어질 수 없으므로 0
        for(int i=0; i<prices.length-1; i++){
             boolean flag = true;
            if(prices[i] == 1){ // 주식 가격이 1원이면 가격은 끝까지 떨어지지 않는다.
                answer[i] = prices.length - i - 1;
            }

            else{
                // 현재 주식의 가격이 떨어지는 지점 찾기
                for(int j=i+1; j<prices.length; j++){
                    if(prices[i] > prices[j]){
                        answer[i] = j-i;    // j-i초간 가격이 떨어지지 않는것으로 본다.
                        flag = false;
                        break;
                    }
                }
                // 만약 현재 주식이 끝까지 떨어지지 않는 경우
                if(flag){
                 answer[i] = prices.length - i - 1;
                 }
            }
           
        }
        return answer;
    }
}

풀이

문제를 이해하는것이 약간 어려웠다.

주식가격이 담긴 배열 prices = [1, 2, 3, 2, 3]이 매개변수로 주어지는데, 가격이 떨어지지 않은 기간이 몇초인지 

answer배열에 담아서 리턴하는 문제이다.

 

위 prices = [1, 2, 3, 2, 3] 을 예로 들어보면,

첫 번째 가격 1의 경우 이후의 가격[2, 3, 2, 3] 보다 작다. 따라서 주식이 끝날때까지 가격은 떨어지지 않으며,

그기간은 4초가 된다.

두 번째 가격 2의 경우도 이후의 가격[3, 2, 3] 보다 작거나 같으므로. 주식이 끝날때까지 가격은 떨어지지 않으며,

그 기간은 3초가 된다.

세 번째 가격 3의 경우, 바로 뒤[2, 3]의 가격보다 작으므로, 1초뒤에 떨어진다. 따라서 기간은 1초

네 번째 가격 2의 경우, 이후의 가격[3] 보다 작으므로, 주식이 끝날때까지 가격은 떨어지지 않으며,

그 기간은 1초가 된다.

마지막 가격 3의 경우, 이후에는 아무 가격이 없으므로 0초다.

 

여기서 생각할 수 있는점은 다음과 같다.

prices의 가격은 1이상 10,000 이하인 자연수이므로,

1) 배열의 값이 1일경우

 - 그 배열의 최솟값이 되므로, 주식이 끝날때까지 가격은 떨어지지 않는다.

 - 따라서 그 경우 기간은 배열의길이 - i - 1이 된다.

 - 위에서 첫 번째 가격의 경우, 배열의길이(5) - i(0) - 1 = 4 이다.

 

2) 배열의 마지막 값은 항상 0초다.

 

3) 그 외 가격이 떨어지는 시간을 찾는다.

 - 현재 주식 가격(i)번째를 기준으로 그 다음 주식 가격(j = i+1)부터 배열의 끝까지 탐색한다.

 - 만약 가격이 떨어지는 곳이 있을경우(prices[i] > prices[j])

 - 그 시간은 j - i 가 된다. 위의 예에서도 [1, 2, 3, 2, 3] 2번째값 3과 3번째값 2의 차이는 1이다.

 - 그 후 체크를 해주기위해 flag변수를 false로 설정하고, break를 통해 탈출한다.

 - 만약 for문을 빠져나와 flag가 true인 경우(현재 가격이 끝날때까지 떨어지지 않는 경우)

 - 위의 경우는 현재 인덱스를 기준으로 시간을 구한다.(prices.length - i - 1)

 

 

 

다른 코드

class Solution {
    public int[] solution(int[] prices) {
        int[] answer = new int[prices.length];
        
        for(int i=0; i<answer.length; i++){
            int sec = 0;
            for(int j=i+1; j<answer.length; j++){
                sec ++;
                if(prices[i] > prices[j])
                    break;
            }
            answer[i] = sec;
        }
        return answer;
    }
}

간단하다...

주식 가격이 유지되는 시간 sec를 설정하고, 배열을 탐색하면서 1초씩 증가시켜준다.

만약 가격이 커지면 for문을 탈출하고, sec의 값이 시간이 된다.