[JAVA][스택/큐][L2] 주식가격

문제 설명

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

제한사항

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

입출력 예

pricesreturn

[1, 2, 3, 2, 3]

[4, 3, 1, 1, 0]

입출력 예 설명

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

 

코드

코드설명: 해당 문제를 스택으로 풀었을때에는 스택에 prices를 넣고, 빼면서 price 값들을 비교해야 하는 그 시간이 오래걸리기 때문에 stack 또는 Queue로 풀기에는 적합하지 않은것 같다.

import java.util.ArrayList;

class Solution {
    public int[] solution(int[] prices) {
        int[] answer = new int[prices.length];
        //기준점에서 시작하여 떨어지는 시점까지 for문 반복
         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;
                }
                    
            }
                if(answer[i] == 0)
                    answer[i] = prices.length-i-1;
        }   
        return answer;
        }
    }