[LeetCode] 502번 IPO 문제를 풀어보았다. (ft. java)

2023년 02월 23일 목요일 - 대황T1!

오늘 올려볼 문제는 502번 IPO 이라는 문제이다.

사진을 클릭하면 해당 문제로 이동합니다.

오늘도 LeetCode 사이트 오늘의 문제를 가지고 왔다.

크으 티원 경기력 미쳤다리

입력

사진에서도 볼 수 있듯이 int 값 2개와 int 배열 2개가 입력으로 들어온다.

---

풀이 및 코드

문제 설명하기에는 조금 어렵다... 사진 참고해주세요...

오늘은 처음부터 정답을 생각해냈다.

TreeMap으로 key-value 형식으로 저장해준다.

이 후 floorKey를 사용해서 최대 이익을 뽑아내고 그리디를 사용하는 방식으로 문제를 풀었다.

이제 코드를 봐보자!

풀이코드

class Solution {
    public int findMaximizedCapital(int k, int w, int[] profits, int[] capital) {
        var map = new TreeMap<Integer, ArrayList<Integer>>();
        
        for(int i = 0; i < profits.length; i++) {
            if(!map.containsKey(capital[i])) map.put(capital[i], new ArrayList<>());
            map.get(capital[i]).add(profits[i]);
        }
        
        var pq = new PriorityQueue<Integer>(Collections.reverseOrder());
        int result = w;
        
        for(int i = 0; i < k; i++) {
            while(map.floorKey(result) != null) {
                pq.addAll(map.get(map.floorKey(result)));
                map.remove(map.floorKey(result));
            }
            
            if(!pq.isEmpty()) result += pq.poll();
        }
        
        return result;
    }
}

---

제출 화면

pq 2개로 문제를 풀 수 있었다... ㄷㄷ

아 그리고 어제 감성 글은 안썼다.

왜냐고? 까먹었거든 ㅋ

오늘 쓸 듯

내일도 문제를 풀어서 블로그에 글을 쓸 수 있으면 좋겠다.