题目描述
给你一个包含 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)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析及函数部分代码
我的想法是排序之后进行三重循环,然后只要保证最后的ans里面没有重复元素即可。在看过官方题解之后发现我的想法不咋滴。只要用两种循环即可,再加上双指针就能实现。我的想法无论是再空间复杂度还是时间复杂度上面都是不可取的。所以在这里展示一下我参考官方题解之后的代码:
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> ans=new ArrayList<List<Integer>>();
for(int i=0;i<nums.length;i++){
int target=-nums[i];
int second=nums.length-1;
if(i>0&&nums[i]==nums[i-1]){
continue;
}
for(int j=i+1;j<nums.length;j++){
if(j>i+1&&nums[j]==nums[j-1]){
continue;
}
while(second>j&&nums[second]>(target-nums[j])){
second--;
}
if(second==j){
break;
}
if(nums[second]==target-nums[j]){
List<Integer> list=new ArrayList<Integer>();
list.add(nums[i]);
list.add(nums[j]);
list.add(nums[second]);
ans.add(list);
}
}
}
return ans;
}
}