题目:
给定一个包含n个整数的数组nums,判断nums中是否存在三个元素a,b,c,使得a + b+c=0?请找出所有满足此要求且不重复的三元组。
假设给定的数组为{1, 7, 2, -1, -1, 0, -2, -5}
解答思路
-
暴力解法
使用三重循环,依次遍历所有的可能,但是这个方法并不推荐使用,因为此方法的时间复杂度为O(N^3)。 -
排序+双指针
先将给定的数组按照从小到大的顺序排序,然后通过控制指针的移动去选择元素。一个指针L执行当前元素的右相邻元素,另外一个指针R指向数组的最后一个元素。排完序后的数组:
1.什么时候结束循环?
当nums[i]大于0的时候,此时L,R指针指向的数肯定也是大于0,所以nums[i] + nums[L] + nums[R] = 0 肯定就不成立了,此时就可以结束了。
2.什么时候L,R指针移动?
nums[i] + nums[L] + nums[R] > 0; R- -
nums[i] + nums[L] + nums[R] < 0; L++
nums[i] + nums[L] + nums[R] == 0; L++, R- -
3.怎么保证结果集不重复?即去重?
首先数组是排好序递增的
对a去重:当nums[i] 和 nums[i -1]相等,即a和前一个元素相等时,跳过当前循环。因为nums[i-1]的结果集肯定是包含nums[i]的结果集的。
当a的值确定时,要做的就是对b和c去重
对b去重:当nums[L]和nums[L+1]相等时,左指针向右移动一步
对c去重:当nums[R]和nums[R-1]相等时,右指针向左移动一步
代码实现:
private static List<List<Integer>> calculation(int[] nums) {
int len;
List<List<Integer>> result = new ArrayList<>();
if (nums != null && (len = nums.length) >= 3) {
Arrays.sort(nums);
System.out.println("nums = " + Arrays.toString(nums));
for (int i = 0; i < len; i++) {
// L指针指向当前指针右侧
int L = i + 1;
// R指针指向末尾
int R = len - 1;
if (i > 0 && nums[i] == nums[i-1]) continue;
// 当当前指针的数大于0时,那么L和R指针肯定也是大于0的,即不成立
if (nums[i] <= 0) {
while (L < R ) {
int sum = nums[i] + nums[L] + nums[R];
if (sum < 0) {
L++;
} else if (sum > 0) {
R--;
} else {
result.add(Arrays.asList(nums[i], nums[L], nums[R]));
// 在这里循环判断,直到把b,c重复的值排除掉
while(R > L && nums[L] == nums[L+1])
L++;
while(R > L && nums[R] == nums[R-1])
R--;
// 最后左右指针再次移动
L++;
R--;
}
}
}
}
}
return result;
}
最后得到的结果为: [[-5, -2, 7], [-2, 0, 2], [-1, -1, 2], [-1, 0, 1]]