diff --git a/src/main/java/com/thealgorithms/misc/FourSumProblem.java b/src/main/java/com/thealgorithms/misc/FourSumProblem.java new file mode 100644 index 000000000000..2d18e0a4d3b5 --- /dev/null +++ b/src/main/java/com/thealgorithms/misc/FourSumProblem.java @@ -0,0 +1,43 @@ +package com.thealgorithms.misc; + +import java.util.ArrayList; +import java.util.Arrays; +import java.util.List; + +public class FourSumProblem { + public List> fourSum(int[] nums, int target) { + + int n = nums.length; + List> ans = new ArrayList<>(); + Arrays.sort(nums); + + for (int i = 0; i < n; i++) { + if (i > 0 && nums[i] == nums[i - 1]) continue; + + for (int j = i + 1; j < n; j++) { + if (j > i + 1 && nums[j] == nums[j - 1]) continue; + + int k = j + 1; + int l = n - 1; + + while (k < l) { + long sum = (long) nums[i] + nums[j] + nums[k] + nums[l]; + + if (sum == target) { + ans.add(Arrays.asList(nums[i], nums[j], nums[k], nums[l])); + k++; + l--; + + while (k < l && nums[k] == nums[k - 1]) k++; + while (k < l && nums[l] == nums[l + 1]) l--; + } else if (sum < target) { + k++; + } else { + l--; + } + } + } + } + return ans; + } +}