house robber solution

Author: jinvicky

house-robber-ii/jinvicky.java

@@ -0,0 +1,38 @@
+class Solution {
+    /**
+     * 기존 house-robber 1 문제에서 로직을 가져오되, 접근법을 달리하는 문제
+     */
+    public int rob(int[] nums) {
+        if (nums.length == 1) return nums[0];
+        else if (nums.length == 2)
+            return Math.max(nums[0], nums[1]);
+        else if (nums.length == 3) {
+            return Math.max(Math.max(nums[0], nums[1]), nums[2]);
+        }
+
+        int n = nums.length;
+        // 첫째 요소만 포함한 dp (마지막 요소 포함 X)
+        int[] firstDp = new int[n - 1];
+        firstDp[0] = nums[0];
+        for (int i = 1; i < n - 1; i++) {
+            int prev2AndNowRob = (i - 2 < 0 ? 0 : firstDp[i - 2]) + nums[i];
+            int prev1Rob = firstDp[i - 1];
+
+            firstDp[i] = Math.max(prev2AndNowRob, prev1Rob);
+        }
+        // System.out.println(firstDp[n-2]); // ok
+
+        // 마지막 요소만 포함한 dp (첫번째 요소 포함 X)
+        int[] lastDp = new int[n];
+        lastDp[1] = nums[1];
+        for (int i = 2; i < n; i++) {
+            int prev2AndNowRob = (i - 2 < 1 ? 0 : lastDp[i - 2]) + nums[i];
+            int prev1Rob = lastDp[i - 1];
+
+            lastDp[i] = Math.max(prev2AndNowRob, prev1Rob);
+        }
+        // System.out.println(lastDp[n-1]); // ok
+
+        return Math.max(firstDp[n - 2], lastDp[n - 1]);
+    }
+}

top-k-frequent-elements/jinvicky.java

@@ -1,6 +1,4 @@
-import java.util.HashMap;
-import java.util.Map;
-import java.util.PriorityQueue;
+import java.util.*;
 
 class Solution {
     public int[] topKFrequent(int[] nums, int k) {
@@ -16,9 +14,7 @@ public int[] topKFrequent(int[] nums, int k) {
          * 결과배열에 할당할 때는 그냥 앞에서부터 하면 된다.
          * 이미 k개를 만족했고, 빈도수가 낮 -> 높 순서대로 그대로 쌓으면 된다.
          *
-         * 우선순위큐에 할당할 때 [숫자, 빈도수]냐 [빈도수, 숫자] 냐는 별 상관이 없다.
-         * 1. 우선순위 큐 초기화
-         * 2. 결과배열에 값 할당할 때 참조하는 [인덱스]
+         * 첫번째부터 꺼내서 결과배열에 할당할 거면 [빈도수, 숫자]로 큐에 저장하는 게 맞다
          */
         int[] answer = new int[k];