15. 三数之和
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路:
- 1 首先进行数组排序,时间复杂度 O(nlogn)
- 2 在数组 nums 中,进行遍历,每遍历一个值利用其下标i,形成一个固定值 nums[i]
- 3 再使用前指针指向 start = i + 1 处,后指针指向 end = nums.length - 1 处,也就是结尾处
- 4 如果 nums[i]大于 0,则三数之和必然无法等于 0,结束循环
- 5 几种情况去重
- 5 比较sum和 0 的大小,start-- 或 end++
上代码:
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
int len = nums.length;
List<List<Integer>> res = new ArrayList<List<Integer>>();
if(nums == null || len<3){
return res;
}
Arrays.sort(nums);
for(int i =0 ;i<len ;i++){
if(nums[i]>0)//如果当前数字大于0,则三数之和一定大于0,所以结束循环
{
break;//结束for循环
}
if( i>0 && nums[i]==nums[i-1]){
continue;//结束当前此次循环,后面的语句不执行
}
int start=i+1,end=len-1;
while(start<end){
int sum = nums[i]+nums[start]+nums[end];
if(sum==0){
res.add(Arrays.asList(nums[i],nums[start],nums[end]));
while (start<end && nums[start] == nums[start+1]) start++; // 去重
while (start<end && nums[end] == nums[end-1]) end--; // 去重
start++;
end--;
}else if(sum<0){
start++;
}else if(sum>0){
end--;
}
}
}
return res;
}
}
知识点:res.add(Arrays.asList(nums[i],nums[start],nums[end]));
- 1 Arrays.asList的作用是将数组转化为list,一般是用于在初始化的时候,设置几个值进去,简化代码,省去add的部分。
- 2 有以下几点需要注意:
(1)该方法不适用于基本数据类型(byte,short,int,long,float,double,boolean)
(2)该方法将数组与列表链接起来,当更新其中之一时,另一个自动更新
(3)不支持add和remove方法