15 三数之和

此题要求在一维数组中寻找三个数的和为0,首先我们想到的最直接的方法就是 蛮力法 ,直接使用三层循环进行嵌套,查找和为0的三个数,这种方案的是复杂度是15 三数之和 - 图1,空间复杂度是15 三数之和 - 图2, 很显然对于这道题这样的时间复杂度是无法接受的。同时考虑到这道题中还要求 不重复 ,因此我们还需要一些额外的检查机制才保证不重复,或者达到去重的效果。

联想到之前做过的两数之和中,我们采用空间换时间的方案,使用哈希表的方法将两数之和的时间复杂度降为15 三数之和 - 图3。此题中,我们同样能够使用这个方法进行复杂度的降低,但是此时依然存在需要检查重复的问题,同时使用 hashTable 之后的时间复杂度依然为15 三数之和 - 图4。因此考虑到此,我们可以发现虽然这道题之后将之前的两数之和变为三数之和,但是在求解的方面还是存在着很大的差异,因此我们需要寻找额外的方法进行求解。

对于这道题采用的一个优化办法就是 双指针法 进行优化。由于这道题中的一个要求是 不重复 ,因此我们首先将原数组按从小到大进行 排序 ,排序之后我们就能很容易的通过相邻的元素是否相等这个条件来进行去重。同时,排序之后的元素序列也便于我们使用双指针法进行求解。

排序之后我们为什么能够使用双指针法进行求解?可以发现,如果我们固定了前两重循环枚举到的元素15 三数之和 - 图515 三数之和 - 图6,那么只有唯一的15 三数之和 - 图7满足15 三数之和 - 图8,当我们向后枚举15 三数之和 - 图9时有15 三数之和 - 图10,我们需要15 三数之和 - 图11,那么一定有15 三数之和 - 图12,也就是说,我们可以从小到大枚举 bb,同时从大到小枚举 cc,即第二重循环和第三重循环实际上是并列的关系。因此保持大的两层循环框架不变,第三层循环我们从右侧开始,向左移动指针。

按照上面的分析,我们可以编写下面的代码,在编写的过程中我们还需要注意一些细节的问题:由于此题中是要求三个数的和为15 三数之和 - 图13,因此我们在循环中可以使用一些额外的条件,确保在和无法为15 三数之和 - 图14时,及时的退出当前的求解, 加速整个求解过程。

算法的时间消耗和空间消耗主要来自两个部分,一个是排序,另一个两层循环中的求解。排序过程中的时间复杂度为15 三数之和 - 图15,空间复杂度为15 三数之和 - 图16,两层循环求解时时间复杂度为15 三数之和 - 图17,空间复杂度为15 三数之和 - 图18。因此算法整体的时间复杂度为15 三数之和 - 图19,空间复杂度为15 三数之和 - 图20

与此题类似的还有四数之和我们也采用同样的办法进行求解。

  1. class Solution {
  2. public List<List<Integer>> threeSum(int[] nums) {
  3. List<List<Integer>> ans = new ArrayList<List<Integer>>();
  4. Arrays.sort(nums);
  5. for (int first = 0; first < nums.length && nums[first] <= 0; ++first) {
  6. if (first != 0 && nums[first] == nums[first - 1])
  7. continue;
  8. int target = -nums[first];
  9. int third = nums.length - 1;
  10. if (nums[third] < 0)
  11. break;
  12. for (int second = first + 1; second < nums.length; ++second) {
  13. if (second != first + 1 && nums[second] == nums[second - 1])
  14. continue;
  15. while (second < third && nums[second] + nums[third] > target)
  16. --third;
  17. if (second == third)
  18. break;
  19. if (nums[second] + nums[third] == target) {
  20. List list = new ArrayList<Integer>();
  21. list.add(nums[first]);
  22. list.add(nums[second]);
  23. list.add(nums[third]);
  24. ans.add(list);
  25. }
  26. }
  27. }
  28. return ans;
  29. }
  30. }