一、题目描述
给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请
你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意,输出的顺序和三元组的顺序并不重要。
示例 2:
输入:nums = [0,1,1]
输出:[]
解释:唯一可能的三元组和不为 0 。
示例 3:
输入:nums = [0,0,0]
输出:[[0,0,0]]
解释:唯一可能的三元组和为 0 。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、运行结果
三、解题思路
首先对原数组按非降序的方式进行排序。逐个将从第1个至第n-2个元素作为三个元素中的第一个(i),另外两个元素只考虑i后面的元素,采用双指针的方式,两个指针分别从两端向中间靠拢,如果出现重复的值就跳过(只考虑第一次出现的时候),避免出现重复的组合。
这里效率比较低的原因可能是在循环里的判断太多,可以尝试一下改进循环内结构,减少判断,如将while循环也改为for循环,即以第二个元素的下标作为内存循环。
四、AC代码
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
int len = nums.length;
List<List<Integer>> ans = new ArrayList<>();
Arrays.sort(nums); //对原数组排序
for(int i=0; i<len-2; i++){ //第一个元素下标
if(i>0 && nums[i]==nums[i-1]) continue;
int j = i+1, k = len -1; //j是第二个元素的下标,k是第三个元素的下标
while(j < k){
List<Integer> tmplist= new ArrayList<>();
if(j > i+1){ //去除重复的组合
while(nums[j] == nums[j-1] && j<k) j++;
}
if(k < len-1){
while(nums[k] == nums[k+1] && k>j) k--;
}
if(j >= k) break;
if(nums[i] + nums[j] + nums[k] == 0){
tmplist.add(nums[i]);
tmplist.add(nums[j]);
tmplist.add(nums[k]);
ans.add(tmplist);
j++; k--;
}
else if(nums[i] + nums[j] + nums[k] > 0) k--;
else j++;
}
}
return ans;
}
}