代码随想录 Day16 哈希表 | LC18 四数之和 && 哈希表总结

最新推荐文章于 2024-07-19 15:45:46 发布

妖怪森林

最新推荐文章于 2024-07-19 15:45:46 发布

阅读量650

点赞数 11

分类专栏：代码随想录文章标签：散列表算法 leetcode java

本文链接：https://blog.csdn.net/hello779/article/details/137589279

版权

代码随想录专栏收录该内容

21 篇文章 0 订阅

订阅专栏

本文讲述了在给定整数数组中寻找满足四数之和等于目标值的不重复四元组的解题方法，强调了使用哈希表进行去重和剪枝操作的重要性，以及与三数之和问题的区别。

摘要由CSDN通过智能技术生成

九、四数之和

题目：

力扣18：四数之和

给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[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 = [2,2,2,2,2], target = 8
输出：[[2,2,2,2]]

提示：

1 <= nums.length <= 200
-10⁹ <= nums[i] <= 10⁹
-10⁹ <= target <= 10⁹

解题思路分析：

难点在于去重和剪枝。

参考上节课的三数之和的思路：四数之和其实就是在三数之和的外面再套一层for循环；

具体操作如下图所示：

四数之和的双指针用法

第一层循环遍历k，第二层循环遍历i，然后k、i固定，找left和right；

找到对应的nums[k]+nums[i]+nums[left]+nums[right]=target

剪枝操作：

在三数之和中，如果nums[i]>0，我们就直接进行剪枝操作；

在四数之和中，大家可能会延续上面的想法，觉得nums[k]>target，就能直接进行剪枝操作；

这样做是不行的！

因为在三数之和中，要求的三个数加起来为0，排序后可以这样剪枝；

但是在四数之和中，要求的是四个数加起来等于target，这个target可正可负；

如果target是负数，那么存在负数和负数加起来会变小的情况，不能直接用nums[k]>target来剪枝；

假设target是-7，那么-3、-2、-1、-1都大于target，但是加起来等于-7。

因为大家的惯性思维会觉得两个数加起来应该是变大的，所以会忽略负数相加的情况。

那么如何进行剪枝操作呢？

需要增加一个前提条件，让nums[k]>target能做剪枝；

前提条件就是nums[k]>0 && target>0，这样当nums[k]>target时，就能做剪枝；

去重操作：

这里需要注意的点和三数之和中一样，nums[i]要和nums[i-1]进行比较才行；

四数之和中：

第一层for循环中判断nums[k]==nums[k-1]，就对k进行了去重；

接下来的i、left、right的去重逻辑就与三数之和中一样了。

代码讲解：

//定义一个List，用来存放找到的四元组
List<List<Integer> > result = new List<>();
//数组排序
Arrays.sort(nums);
//循环遍历k
for(int k = 0; k < nums.length; k++) {
  //剪枝操作
  if(nums[k] > target && nums[k] > 0 && target > 0) return result;
  //对nums\[k\]进行去重
  if(k > 0 && nums[k] == nums[k-1]) {
    continue;
  }
  //循环遍历i
  for(int i = 0; i < nums.length; i++) {
    //二级剪枝
    if(nums[k] + nums[i] > target && nums[k] + nums[i] > 0 && target > 0) return result;
      break;
    }
    //对nums[i]去重
    if(i > k + 1 && nums[i] == nums[i-1]) {
      continue;
    }
    int left = i + 1;
    int right = nums.length - 1;
    while(left < right) {
      // nums[k] + nums[i] + nums[left] + nums[right] > target int会溢出
      long sum = (long) nums[i] + nums[j] + nums[left] + nums[right];
      if(sum > target) right--;
      else if(sum < target) left++;
      else {
        result.add(Arrays.asList(nums[k], nums[i], nums[left], nums[right]));
        while(right > left && nums[right] == nums[right - 1]) right--;
        while(right > left && nums[left] == nums[left + 1]) left++;
        right--;
        left++;
      }
    }
  }
}
return result;