Stock Price / 심화

Author: oh-chaeyeon

oh-chaeyeon/2주차_스택/Stock_Price.js

@@ -0,0 +1,18 @@
+function solution(prices) {
+  const answer = Array(prices.length).fill(0);
+  const stack = [];
+
+  prices.forEach((price, i) => {
+    while (stack.length && prices[stack[stack.length - 1]] > price) {
+      const idx = stack.pop();
+      answer[idx] = i - idx;
+    }
+    stack.push(i);
+  });
+
+  stack.forEach((idx) => {
+    answer[idx] = prices.length - 1 - idx;
+  });
+
+  return answer;
+}