面试经典题---15.三数之和

15.三数之和

我的解法:

  • 预处理
    • 当nums大小小于3时,直接返回空的res
    • 对nums排序后,若首元素小于0或末尾元素大于0,也直接返回空的res
  • 双指针法找出三元组(nums[i]、nums[left]和nums[right])
    • i从0开始取值,对于每个i,判断是否存在三元组和为0的left(初值为i+1)和right(初值为n-1)
      • nums[i]+nums[left]+nums[right]<0时,左移left
      • nums[i]+nums[left]+nums[right]>0时,右移right
      • nums[i]+nums[left]+nums[right]=0时,将三元组下标插入res中
    • 注意排除重复的三元组,因此找到和为0的三元组后,要while循环判断下一步的left和right值是否重复,重复则直接跳过
  • 外部的while循环:每轮都是找出当前i是否存在满足题意的三元组,每轮结束后也要判断用while循环跳过重复的i
class Solution {
public:
    vector> threeSum(vector& nums) {
        int n = nums.size();
        vector> res;
        if(n < 3){
            return res;
        }
        sort(nums.begin(), nums.end());
        if(nums[0] > 0 || nums[n - 1] < 0){
            return res;
        } 
        int i = 0, left, right = n - 1;
        while(i < n){
            if(nums[i] > 0){
                break;
            }
            left = i + 1;
            right = n - 1;
            while(left < right){
                if(nums[i] + nums[left] < -nums[right]){
                    left++;
                }
                else if(nums[i] + nums[left] > -nums[right]){
                    right--;
                }
                else{
                    res.push_back({nums[i], nums[left], nums[right]});
                    left++;
                    right--;
                    while(left < right && nums[left - 1] == nums[left]){
                        left++;
                    }
                    while(left < right && nums[right + 1] == nums[right]){
                        right--;
                    }
                }
            }
            i++;
            while(i < n && nums[i] == nums[i - 1]){
                i++;
            }
        }
        return res;
    }
};

你可能感兴趣的:(面试,算法,leetcode,c++)