LeedCode算法题——四数之和（Java）

题目

给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。

注意：

答案中不可以包含重复的四元组。

示例：

给定数组 nums = [1, 0, -1, 0, -2, 2]，和 target = 0。

满足要求的四元组集合为：
[
[-1, 0, 0, 1],
[-2, -1, 1, 2],
[-2, 0, 0, 2]
]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/4sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

跟我之前写的三数之和的思路很像，之前是先固定一个数，然后用双指针遍历剩下的数，这里我用的是固定两个数，然后用双指针前后遍历剩下的数，这样子其实是不太完善的，缺点很明显，如果这时候是五个数，六个数，那就得重新写，复用不好，但是为了快点完成，所以还是这么写了。我觉得后面改进的话是可以写一个通用的函数，如果求四个数的和，那就先固定一个数，然后求三个数的和，如果是三个数的和，就先固定一个数，然后求两个数的和，这样子的递归可能会通用性更强。先附上不太完善的代码：

代码

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        int i,j,k,d,temp=0;;
        Arrays.sort(nums);
       List end = new ArrayList();
       Set<List<Integer>>  set=new HashSet<>();
        for(i=0;i<nums.length-3;i++)
            for(j=i+1;j<nums.length-2;j++)
            {temp=nums[i]+nums[j];
                k=j+1;
                d=nums.length-1;
                while(k<d){
                    if(nums[k]+nums[d]==target-temp)
                    {  List<Integer> result = new ArrayList<>();
                        result.add(nums[i]);
                        result.add(nums[j]);
                        result.add(nums[k]);
                        result.add(nums[d]);
                        set.add(result);
                 
                        k++;
                        d--;
                    }
                    else
                        if(nums[k]+nums[d]<target-temp)
                            k++;
                    else
                         if(nums[k]+nums[d]>target-temp)
                             d--;
                }
            }
        for(List<Integer> list:set)
            end.add(list);
        return end;
    }
}