leetcode 18.四数之和 java

题目

思路

整体在三数之和的基础上进行修改。（所有需要修改的地方，我在代码里加了//改 的注释）

大的一个思路就是，在三数之和的外面再套一层循环。相当于固定前两个数。然后这道题目标值变成一个参数了，不是三数之和的0了。所以我们把a+b+c+d=target的放到res列表里。注意这里求sum 用了long，因为测试案例中有一些数非常大。

需要注意的细节就是a,b,c,d的去重。

a的去重和我们三数之和a的去重一样。j>0 && nums[j] == nums[j - 1]。

b的去重我们修改成了i>j+1 && nums[i] == nums[i - 1]。为啥写成这，而不是i>1呢？看个例子 [-2，-1，-1，1，1，2，2] target=0。如果i>1 就剪枝的话，当j=1,i=2的时候 这时候nums[2]==nums[3]直接就被剪掉了，而我们还没有搜索呢。

c的去重很简单，就是把原来sum>0,改成了sum>target。

d的去重类推。

此外还有一个细节就是，不要判断a > target 就break，因为这次我们的target可能是负数，这和三数之和不一样了。

举个例子：[-4,-3,-2,-1],target=-10。虽然a>target，但是仍然有sum能等于target。如果我们的数组所有的数都是正数，就可以按原来的剪枝。

代码

import java.sql.Array;
import java.util.ArrayList;

//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> res = new ArrayList<>();
        //对数组从小到大进行排序
        Arrays.sort(nums);

        for (int j = 0; j < nums.length; j++) {
            if (nums[j] > 0 && nums[j] > target) //改
                break;
            //对a去重
            if (j > 0 && nums[j] == nums[j - 1]) {
                continue;
            }
            for (int i = j + 1; i < nums.length; i++) {  //改
                //对b去重
                if (i > j + 1 && nums[i] == nums[i - 1]) {
                    continue;
                }
                int left = i + 1;
                int right = nums.length - 1;

                while (right > left) {
                    long sum = (long) nums[j] + nums[i] + nums[left] + nums[right]; //改
                    if (sum > target) {  //改
                        right--;
                    } else if (sum < target) { //改
                        left++;
                    } else {
                        res.add(Arrays.asList(nums[j], nums[i], nums[left], nums[right]));  //改
                        //对c和d去重
                        while (right > left && nums[left] == nums[left + 1]) left++;
                        while (right > left && nums[right] == nums[right - 1]) right--;
                        //找到一个四元组后，继续寻找下一个四元组
                        left++;
                        right--;
                    }
                }
            }
        }
        return res;
    }
}
//leetcode submit region end(Prohibit modification and deletion)