LeetCode 15:三数之和(Java实现)
题目
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],满足要求的三元组集合为:[ [-1, 0, 1] , [-1, -1, 2] ]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路1:暴力法
几点不熟悉的知识:
java.lang.Integer
java.util.Arrays.asList(Integer… a)
java.util.List
java.util.List<List>
java.util.ArrayList.ArrayList<List>()
void java.util.Arrays.sort(int[] a)
boolean java.util.List.contains(Object o)
boolean java.util.List.add(List e)
代码
public List<List<Integer>> threeSum (int[] nums) {
List<List<Integer>> ans = new ArrayList<>();
Arrays.sort(nums);
for(int i=0;i<nums.length-2;i++){
for(int j=i+1;j<nums.length-1;j++){
for(int k=j+1;k<nums.length;k++){
if(nums[i]+nums[j]+nums[k]==0){
if(!ans.contains(Arrays.asList(nums[i],nums[j],nums[k])))
ans.add(Arrays.asList(nums[i],nums[j],nums[k]));
}
}
}
}
return ans;
}
结果
orz超出时间限制
思路2:双指针
- 将数组排序;i从数组首端开始遍历数组,同时设L、R两个指针,根据sum的正负判断L、R往左或者往右移动。
- 要注意防止重复:a.当i/R/L与即将遍历到的下一个相等时,直接跳过。
- “根据sum、nums[i]的正负判断是否结束循环、L/R往左或往右移”这个思想不错。
代码
public List<List<Integer>> threeSum (int[] nums){
List<List<Integer>> ans = new ArrayList<>();
Arrays.sort(nums);//排序
int len = nums.length;
if(len<3) return ans;//长度小于3
for(int i=0;i<len-2;i++){
if(nums[i]>0) return ans;
if(i>0&&nums[i]==nums[i-1]) continue;//防重,并且i>0防止越界
int L = i+1;
int R = len-1;
while(L<R){
int sum = nums[i]+nums[L]+nums[R];
if(sum==0){
while (L<R&&nums[L]==nums[L+1]) L++;//防重,注意要是写L<R&&nums[L]==nums[L-1]的话会有越界的情况
while (L<R&&nums[R]==nums[R-1]) R--;//防重,同上理
ans.add(Arrays.asList(nums[i],nums[L],nums[R]));
L++;
R--;
}
else if(sum>0) R--;
else if(sum<0) L++;
}
}
return ans;
}