[JS] 2Level / 스택/큐 / 주식가격

문제 설명

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

프로그래머스

 

문제 풀이 방법

• stack을 이용해 풀이를 진행했다.
  문제는 각 가격이 언제까지 떨어지지 않는가를 측정하는 문제인데,
  떨어지는 기준은 해당 가격보다 아래로 내려가는 것이다.
• 스택에 떨어지지 않는 가격의 인덱스를 담고, 이후에 떨어지게 되면 stack의 마지막 요소부터 빼줘서 기간을 계산한다.
• 위의 과정을 거치고도 stack에 남아있는 수들은 한번도 떨어지지 않은 가격이므로
  가격 배열의 길이에서 인덱스값을 빼주고 -1을 계산해주면 기간이 계산된다.

 

코드 

const solution = (prices) => {
    let answer = Array.from({length: prices.length}, () => 0);
    let stack = [];
    let length = prices.length;
    
    for(let i=0; i<length; i++){
        while(stack.length && prices[i] < prices[stack[stack.length - 1]]){
            let temp = stack.pop();
            answer[temp] = i - temp;
        }
        stack.push(i);
    }
    // stack에 남아있는 경우는 끝까지 가격이 떨어지지 않는 경우
    while(stack.length){
        let temp = stack.pop();
        answer[temp] = length - temp - 1;
    }
    return answer;
}