Q32 LeetCode15 三数之和

难点在于不能重复

1.将数组进行排序

2.找到合适组合后将三个指针都要进行向后去重操作

 

 

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> ans = new ArrayList<>();
        int n = nums.length;
        for (int i = 0; i < n - 2; ++i) {
            int x = nums[i];
            if (i > 0 && x == nums[i - 1]) continue; // 跳过重复数字
            int j = i + 1, k = n - 1;
            while (j < k) {
                int s = x + nums[j] + nums[k];
                if (s > 0) k--;
                else if (s < 0) j++;
                else {
                    ans.add(List.of(x, nums[j], nums[k]));
                    j=j+1;
                    while(j < k&&nums[j] == nums[j - 1]){
                        j++;
                    }
                    //for (j=j+1; j < k && nums[j] == nums[j - 1]; ++j); // 跳过重复数字
                    k=k-1;
                     while(j < k&&nums[k] == nums[k + 1]){
                        k--;
                    }
                    //for (k=k-1; k > j && nums[k] == nums[k + 1]; --k); // 跳过重复数字
                }
            }
        }
        return ans;
    }
}