题目链接:
力扣https://leetcode-cn.com/problems/3sum/
【分析】这道题用到了排序和双指针,首先对数组进行排序,然后先固定一个i,-i的值定位target,接着j从i+1开始,k从n-1开始,计算j和k的和sum:
(1)如果sum==target,加入答案。
(2)如果sum>target,让k--,直到下一个不同的元素。
(3)如果sum<target,让j++,直到下一个不同的元素。
(4)对这个i找完之后,让i++,直到下一个不同的i元素。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> ans = new ArrayList<>();
int n = nums.length, i, j, k, target, sum;
if(n < 3) return ans;
Arrays.sort(nums);
for(i = 0; i < n; i++){
if(nums[i] > 0) break;
if(i != 0 && nums[i] == nums[i - 1]) continue;
j = i + 1; k = n - 1; target = -nums[i];
while(j < k){
sum = nums[j] + nums[k];
if(sum == target){
List<Integer> list = new ArrayList<>();
list.add(nums[i]); list.add(nums[j]); list.add(nums[k]);
ans.add(list);
while(j < k && nums[k] == nums[k - 1]) k--;
while(j < k && nums[j] == nums[j + 1]) j++;
k--; j++;
}else if(sum > target){
while(k != 0 && nums[k] == nums[k - 1]) k--; k--;
}else{
while(j != n - 1 && nums[j] == nums[j + 1]) j++; j++;
}
}
}
return ans;
}
}