题目
- 给定一个包含n个整数的数组nums,判断nums中是否存在三个元素a, b, c 使得a + b +c =0 ?找出所有满足条件且不重复的三元组。
- 注意:答案中不可以包含重复的三元组。
- 例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路
- 先把数组排序,设定两个指针left和right
代码
public List<List<Integer>> threeSum(int[] nums) {
// 先排序,提高效率
Arrays.sort(nums);
List<List<Integer>> result = new LinkedList<>();
for (int i = 0; i < nums.length; i++) {
int left = i + 1, right = nums.length - 1;
while(left < right){
int sum = nums[i] + nums[left] + nums[right];
if(sum == 0){
List<Integer> tmp = new LinkedList<>();
tmp.add(nums[i]);
tmp.add(nums[left]);
tmp.add(nums[right]);
// 去重
if(!tmp.contains(tmp)){
result.add(tmp);
}
}
// 防止死循环
if(sum >= 0){
left ++;
}else {
right--;
}
}
}
return result;
}