题目:
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路
先对数组进行排序,然后开始遍历固定一个值(i)。
定义两个变量left和right分别从左和右向中间逼夹。
right为数组最后一个元素,left为i+1的元素。
既然数组已经有序 如果如果三数和大于0 需要对数值进行减小 right --,
如果小于0 需要对数值进行增大 left ++
如果等于0则加入result
另外满足返回的结果不能是重复的,最先想到的是使用Set来记录结果,但是会需要更多的内存空间。
既然已经有序则i left right不能和前面遍历过得元素相同。
在取i值时判断 i值是否和i-1相同,如果相同取下一个
在计算满足0 后 加入result后 检查下一个要遍历的left和right是否和当前数相同,如果相同需要用下一个,直到取到第一个不和当前数想等的值。
优化:判断如果i>0则 三数和必大于0 需要break.
题解:
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> result = new ArrayList<List<Integer>>();
if (nums == null || nums.length < 3) {
return result;
}
Arrays.sort(nums);
for (int i = 0; i < nums.length - 2; i++) {
//要找个第一个数已经大于0了 所以需要停止查找了
if (nums[i] > 0) break;
//防止第一个数重复
if (i > 0 && nums[i]==nums[i - 1]) {
continue;
}
int left = i + 1;
int right = nums.length - 1;
while (left < right) {
if (nums[i] + nums[left] + nums[right] > 0) {
right--;
} else if (nums[i] + nums[left] + nums[right] < 0) {
left++;
} else {
List<Integer> oneResult = new ArrayList<>(3);
oneResult.add(nums[i]);
oneResult.add(nums[left]);
oneResult.add(nums[right]);
result.add(oneResult);
//避免第二个数重复
while (left < right && nums[left]==nums[left + 1]) {
left++;
}
//避免第三个数重复
while (left < right && nums[right]==nums[right - 1]) {
right--;
}
right--;
left++;
}
}
}
return result;
}
耗时及排名