leetCode:三数之和
题目:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重
复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
思路:
求三数之和为 0, nums[i] + nums[j] + nums[k] = 0 可以转换为求两个数之和,也即是 nums[i] + nums[j] = -nums[k]; 那么就可以通过指针移动去查找满足公式的数字,而为了方便控制指针的移动,前提是一组数字是有序的。
那么大概的解题思路如下:
对数组排序
- 定义三个指针 i, left, right; 指针 i 指向的是求和项,指针 left, right 分别指向数组的头部和尾部
- 当 -nums[i] = nums[left] + nums[right] 时满足公式,指针 left, right 继续向中部移动
- 当 -nums[i] > nums[left] + nums[right] 时指针 left 向中部移动
- 当 -nums[i] < nums[left] + nums[right] 时指针 righr 向中部移动
- 当指针 left,right 交叉时停止移动,指针 i 向下移动,指针 left, right 重置
代码:
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
Set<List<Integer>> list = new HashSet<>();
for (int i = 0; i < nums.length; i++) {
int left = i + 1;
int right = nums.length - 1;
while (left < right) {
if (nums[left] + nums[right] == -nums[i]) {
List<Integer> data = new ArrayList<>();
data.add(nums[i]);
data.add(nums[left]);
data.add(nums[right]);
list.add(data);
left++;
right--;
} else if (nums[left] + nums[right] > -nums[i]) {
right--;
} else {
left++;
}
}
}
return new ArrayList<>(list);
}
}