我理解的算法 - 四数之和

LeetCode 18.四数之和

这道题的题目大家自行查看：链接在这
其解法和思路与两数之和和三数之和都是一样的，还不熟悉的小朋友们可以看这里：两数之和，三数之和

那么这道题目的难点在哪里呢？其实这道题目的难点就在于要求的和变成了四个数，比三个数还多一个数，所以我们要想办法尽量把多余的一些处理给它优化掉，这样才能提高其效率，我们来看一下吧

四数之和的思路和三数之和是一样的，我们还是可以使用双指针来做，只不过这边我们需要固定2个元素即固定a和b来移动c和d，来看他们的和是比目标值大还是小，这样就会比三数之和多一次循环。代码如下

public List<List<Integer>> fourSum(int[] nums, int target) {
    List<List<Integer>> retResult = new ArrayList<>();
    int length = nums.length;
    if (nums == null || length < 4) {
        return retResult;
    }
    Arrays.sort(nums);//排序，为了下面去重

    for(int i = 0; i < length - 3; i++){
        if(i > 0 && nums[i] == nums[i - 1]) continue;//去重
        if((long) nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target) break;//剪枝
        if((long) nums[i] + nums[length - 3] + nums[length - 2] + nums[length - 1] < target) continue;//剪枝

        for(int j = i + 1; j < length - 2; j++){
            if(j > i + 1 && nums[j] == nums[j - 1]) continue;//去重
            if((long) nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target) break;//剪枝
            if ((long) nums[i] + nums[j] + nums[length - 2] + nums[length - 1] < target) continue;//剪枝

            int left = j + 1;
            int right = length - 1;
            while(left < right){
                int diff = nums[i] + nums[j] + nums[left] + nums[right] - target;
                if(diff > 0){
                    while(left < right && nums[right] == nums[--right]);//去重
                }else if(diff < 0){
                    while(left < right && nums[left] == nums[++left]);//去重
                }else{
                    retResult.add(new ArrayList<>(Arrays.asList(nums[i], nums[j], nums[left], nums[right])));
                    while(left < right && nums[right] == nums[--right]);//去重
                    while(left < right && nums[left] == nums[++left]);//去重
                }
            }
        }
    }
    return retResult;
}

这边需要说明一下那些剪枝的操作，剪枝就是说，如果我们可以判断出来后面的数没必要再进行计算了，那我们就可以跳过这些步骤，从而降低执行时间，就像把没有用的树枝剪掉一样。

第一段剪枝操作

if((long) nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target) break;//剪枝

这边就是$a+b+c+d$如果大于0了，那说么在d+1往后肯定都是大于0的了，所以我们就没必要再进行后续的计算了，直接break掉。

第二段

if((long) nums[i] + nums[length - 3] + nums[length - 2] + nums[length - 1] < target) continue;//剪枝

如果我们假设用最大的三个数加上了此次固定的a还比目标数小，那么就说明我们的a肯定小了，所以需要继续遍历而不需要再执行移动双指针来尝试了。

下面2段和上面的逻辑是一样的，这样做后，可以大幅度的缩减执行时间，提高执行效率，精华在剪枝处理上，这样的算法你理解了吗