力扣HOT100-三数之和

思路

先固定一个数，在寻找两数之和。

优化:这样会导致重复遍历，例如对于A元素，找到了B和C。

那么对于B元素时，同样还会再找一遍A和C，造成了重复。

因此，我们可以考虑对数组从大到小进行排序，这样我们仅需遍历固定元素的右边元素即可。

Python版本

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        # 先固定一个数，在寻找两数之和
        # 时间复杂度为O(n*n)
        # 优化:这样会导致重复遍历，例如对于A元素，找到了B和C
        # 那么对于B元素时，同样还会再找一遍A和C，造成了重复
        # 因此，我们可以考虑对数组从小到大进行排序，这样我们仅需遍历固定元素的右边元素即可
        n = len(nums)
        nums.sort()
        res = []
        for i in range(n-2):
            if i > 0 and nums[i] == nums[i-1]:
                # 因为答案不可以包含重复的三元组，所以需要跳过
                continue
            l = i + 1
            r = n - 1
            while l < r:
                s = nums[i] + nums[l] + nums[r]
                if s > 0:
                    # 调低nums[l] + nums[r]
                    r -= 1
                    # while l < r and nums[r] == nums[r+1]:
                    #     r -= 1
                elif s < 0:
                    l += 1
                    # while l < r and nums[l] == nums[l-1]:
                    #     l += 1
                else:
                    res.append([nums[i],nums[l],nums[r]])
                    r -= 1
                    while l < r and nums[r] == nums[r+1]:
                        r -= 1
                    l += 1
                    while l < r and nums[l] == nums[l-1]:
                        l += 1
                    
        return res

Java版本

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        int n = nums.length;
        Arrays.sort(nums);
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        for (int i = 0;i < n - 2;i++){
            if (i > 0 && nums[i] == nums[i -1]){
                continue;
            }
            int l = i + 1;
            int r = n - 1;
            while (l < r){
                int s = nums[i] + nums[l] + nums[r];
                if (s > 0){
                    r -= 1;
                }else if (s < 0){
                    l += 1;
                }else{
                    List<Integer> list = new ArrayList<Integer>();
                    list.add(nums[i]);
                    list.add(nums[l]);
                    list.add(nums[r]);
                    res.add(list);
                    r -= 1;
                    while (l < r && nums[r] == nums[r+1]){
                        r -= 1;
                    }
                    l += 1;
                    while (l < r && nums[l] == nums[l -1]){
                        l += 1;
                    }
                }
            }
        }
        return res;
    }
}