Update Combination Sum IV.java

varunu28 · web-flow · commit 42c41ac80063 · 2023-09-09T07:44:36.000-07:00
diff --git a/Medium/Combination Sum IV.java b/Medium/Combination Sum IV.java
@@ -1,25 +1,22 @@
 class Solution {
-  public int combinationSum4(int[] nums, int target) {
-    Integer[] dp = new Integer[target + 1];
-    return getCount(target, nums, dp);
-  }
-  
-  public int getCount(int target, int[] nums, Integer[] dp) {
-    if (dp[target] != null) {
-      return dp[target];
+    public int combinationSum4(int[] nums, int target) {
+        Integer[] dp = new Integer[target + 1];
+        return helper(nums, target, dp);
     }
-    if (target == 0) {
-      return 1;
+
+    private int helper(int[] nums, int target, Integer[] dp) {
+        if (dp[target] != null) {
+            return dp[target];
+        }
+        if (target == 0) {
+            return 1;
+        }
+        int count = 0;
+        for (int num : nums) {
+            if (target >= num) {
+                count += helper(nums, target - num, dp);
+            }
+        }
+        return dp[target] = count;
     }
-    if (target < 0) {
-      return 0;
-    }
-    int total = 0;
-    for (int num : nums) {
-      if (target >= num) {
-        total += getCount(target - num, nums, dp);
-      }
-    }
-    return dp[target] = total;
-  }
 }
