LeetCode系列(二)-3Sum
给定一个包含 n 个整数的数组 nums,
判断 nums 中是否存在三个元素 a,b,c ,
使得 a + b + c = 0 ?
找出所有满足条件且不重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
解法:
1. 时间复杂度O(n^2)
思路:双指针求指定和。优化:去重判断;剪枝优化。
List<List<Integer>> list = new ArrayList<>(); Arrays.sort(nums); for (int i = 0; i < nums.length-2; i++) { // 第一个数去重 如:-4, -1, -1, 0, 1, 2 数组。当第二个数 -1=第三个数 -1。那么第三个数-1得到的情况<=第二个数-1的情况。会有重复。 if (i==0||(i>0&&!(nums[i]==nums[i-1]))) { int target = -nums[i]; // 剪枝优化,因为排序后。如果第一个数为正数,则第二第三个数也会为正数。就不会有相加等于0的情况。 if (nums[i] > 0) { return list; } // 采用双指针法 int lo = i + 1; int hi = nums.length - 1; while (lo < hi) { if (nums[lo] + nums[hi] == target) { list.add(Arrays.asList(nums[i], nums[lo], nums[hi])); // 二三数 去重,如果之后的元素和当前元素一样的话,就跳过他。因为得到的结果一样。 while (lo < hi && nums[lo] == nums[lo + 1]) lo++; while (lo < hi && nums[hi] == nums[hi - 1]) hi--; lo++; hi--; } else if (nums[lo] + nums[hi] > target) { hi--; } else { lo++; } } } } return list;