[프로그래머스/Lv. 2] 주식 가격

🔺 문제

프로그래머스

 

 

🔺 코드

(틀림)

반복문을 썼는데 실패

import java.util.*;
import java.io.*;
 
class Solution {    
    public int[] solution(int[] prices) {
        int len = prices.length;
        int[] answer = new int[len];
       
        for(int i = 0 ; i < len ; i++) {
            int num = prices[i];
            int cnt = 0;
            
            if(i == len-1) {
                answer[i] = 0;
                break;
            }
            
            for(int j = i + 1 ; j < len ; j++) {
                if(prices[j] >= prices[i]) {
                    cnt++;
                }
            }
            answer[i] = cnt;
        }
     
        return answer;
    }
}

 

 

(정답)

import java.util.*;
import java.io.*;
 
public class Solution {
    public static int[] solution(int[] prices) {
        int len =  prices.length;
        int[] answer = new int[len];
        Stack<Integer> stack = new Stack<>();
        
        for(int i = 0 ; i < len ; i++) {
            while(!stack.isEmpty() && prices[stack.peek()] > prices[i]) { 
                answer[stack.peek()] = i - stack.peek();
                stack.pop();    // 답을 구했으므로 스택에서 제거
            }
            stack.push(i);
        }
        
        // 값을 구하지 못한 idx = 끝까지 가격이 떨어지지 않은 주식
        while(!stack.isEmpty()) {
            answer[stack.peek()] = prices.length - stack.peek() - 1; // 최대 기간
            stack.pop();
        }
        
        return answer;
    }
    
    // 실행
    public static void main(String[] args) {
        int[] prices = {1, 2, 3, 2, 3};
        
        int[] answer = solution(prices);
        for(int i : answer) {
            System.out.print(i + " ");
        }
    }
}
 

✅ 해결 아이디어
✔ 스택
- 스택이 비어있지 않고, 주식 가격이 떨어졌을 때 answer에 값 넣어주기 → 해당 인덱스는 pop
(예 ; i가 3이고 stack.peek()이 2인 경우, answer[stack.peek()]에 i - stack.peek()의 값인 1 (= 3 - 2)을 넣는다.
- 그 외 경우는 해당 인덱스를 push
- 끝까지 가격이 떨어지지 않은 주식 (=스택에서 끝까지 남아있는 수). 전체 길이에서 해당 인덱스를 빼고 pop

 

 

---

🔺 다른 풀이들

- 와 설명 짱이시다... 글로 설명을 어떻게 이렇게 풀어서 하시지.. 최고...

[프로그래머스] 스택/큐 주식가격 JAVA 자바풀이

 

 

- 으아악 설명 너무 친절하시고 최고신데 내 머리가 따라가질 못함.. (그림 설명)

[프로그래머스] 주식가격(Stack) - JAVA

---

💬 느낀 점

스택을 이렇게 쓰라고 하면.. 나는 구현 연습을 더 해야할 것 같다.

 

 

1회독	2회독	3회독	4회독	5회독
V

---

(참고)

✔ 풀이 감사합니다...

[프로그래머스] 주식가격(자바)