Leetcode--Java--18. 四数之和

题目描述

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

样例描述

示例 1：

输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
示例 2：

输入：nums = [], target = 0
输出：[]

思路

1. 双指针。与下面这题思路类似
   Leetcode–Java–15. 三数之和
   在此基础上，在多固定一个指针，同时去重即可。
2. 双指针的题一定不要忘了先保证有序

代码

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
           List<List<Integer>> res = new ArrayList<>();
           int len = nums.length;
           Arrays.sort(nums);
           for (int i = 0; i < len - 3; i ++ ) {
               //去重
               if (i > 0 && nums[i] == nums[i - 1]) continue;
               for (int j = i + 1; j < len - 2; j ++ ) {
                   //去重
                   if (j > i + 1 && nums[j] == nums[j - 1]) continue;
                  int l = j + 1, r = len - 1;
                  while (l < r) {
                      int s = nums[i] + nums[j] + nums[l] + nums[r];
                      if (s < target) l ++;
                      else if (s > target) r --;
                      else {
                          res.add(new ArrayList(Arrays.asList(nums[i], nums[j],
                           nums[l], nums[r])));
                           //去重
                           while (l < r && nums[l] == nums[l + 1]) l ++;
                           while (l < r && nums[r] == nums[r - 1]) r --;
                           l ++;
                           r --;
                      }
                  }
               }
           }
           return res;
    }
}