标题 | 15. 三数之和 |
难度 | 中等 |
天数 | 第1天,第3/3题 |
数据结构 | 数组 |
给你一个包含
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
以上内容来源:力扣(LeetCode)
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
- 3个数和等于0,那么只要保证一个数不是重复的,就满足的条件
- 那就先排序,从左到右遍历,保证这个正在遍历的数,不重复就可以了
//排序
Arrays.sort(nums);
- 创建返回的集合
List<List<Integer>> res = new ArrayList();
- 从0开始遍历数组,如果当前元素 > 0,就停止循环
int itemI = nums[i];
//排序后 如果当前数>0就没有符合规则的数了,直接返回记录的数组
if(itemI > 0){
return res;
}
- 采用双指针思路解这个题
i
是当前元素,保证i
不重复,那么满足的就不是重复数组
//去重
if(i > 0 && itemI == nums[i-1]){
continue;
}
i+1
是左指针leftnums.length-1
是右指针right
int left = i + 1;
int right = m - 1;
- 情景1,三个数相加 > 0 ,右指针左移,right–;
- 情景2,三个数相加 < 0 ,左指针右移,left++;
int sum = itemI + nums[left] + nums[right];
if(sum > 0){
right--;
}else if(sum < 0){
left++;
}
- 情景3,相加 == 0 ,添加进数组
- 去重右指针
- 去重左指针
- 去重后,左指针右移,右指针左移
- 继续寻找符合条件
List<Integer> list = new ArrayList();
list.add(itemI);
list.add(nums[left]);
list.add(nums[right]);
res.add(list);
while(right > left && nums[right] == nums[right-1]){
right--;
}
while(right > left && nums[left] == nums[left+1]){
left++;
}
right--;
left++;
- 终止条件,左指针 >= 右指针
while(right > left){}
完整代码:
class Solution {
//数据结构基础 第 1 天 数组 3/3
public List<List<Integer>> threeSum(int[] nums) {
//排序
Arrays.sort(nums);
List<List<Integer>> res = new ArrayList();
int m = nums.length;
for(int i = 0 ; i < m;i++){
int itemI = nums[i];
//排序后 如果当前数>0就没有符合规则的数了,直接返回记录的数组
if(itemI > 0){
return res;
}
//去重
if(i > 0 && itemI == nums[i-1]){
continue;
}
int left = i + 1;
int right = m - 1;
while(right > left){
int sum = itemI + nums[left] + nums[right];
if(sum > 0){
right--;
}else if(sum < 0){
left++;
}else{
List<Integer> list = new ArrayList();
list.add(itemI);
list.add(nums[left]);
list.add(nums[right]);
res.add(list);
while(right > left && nums[right] == nums[right-1]){
right--;
}
while(right > left && nums[left] == nums[left+1]){
left++;
}
right--;
left++;
}
}
}
return res;
}
}