Skip to main content

15.三数之和

caution

原题链接：

思路​

  1. 特判，对于数组长度 nn，如果数组为 nullnull 或者数组长度小于 33，返回 [][]。
  2. 对数组进行排序。
  3. 遍历排序后数组：
    • 若 nums[i]>0nums[i]>0：因为已经排序好，所以后面不可能有三个数加和等于 0，直接返回结果。
    • 对于重复元素：跳过，避免出现重复解
    • 令左指针 L=i+1L=i+1，右指针 R=n−1R=n-1，当 L<RL<R 时，执行循环：
      • 当 nums[i]+nums[L]+nums[R]==0nums[i]+nums[L]+nums[R]==0，执行循环，判断左界和右界是否和下一位置重复，去除重复解。并同时将 L,RL,R 移到下一位置，寻找新的解
      • 若和大于 0，说明 nums[R]nums[R] 太大，RR 左移
      • 若和小于 0，说明 nums[L]nums[L] 太小，LL 右移

代码​

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;

        if (nums.size() < 3) return res;

        sort(nums.begin(), nums.end());
        for(int k = 0; k < nums.size() - 2; k++){
            if(nums[k] > 0) break;
            if(k > 0 && nums[k] == nums[k - 1]) continue;
            int i = k + 1, j = nums.size() - 1;
            while(i < j){
                int sum = nums[k] + nums[i] + nums[j];
                if(sum < 0){
                    while(i < j && nums[i] == nums[++i]);
                } else if (sum > 0) {
                    while(i < j && nums[j] == nums[--j]);
                } else {
                    res.push_back({nums[k], nums[i], nums[j]});
                    while(i < j && nums[i] == nums[++i]);
                    while(i < j && nums[j] == nums[--j]);
                }
            }
        }
        return res;
    }
};