【c++】【leetcode15】 三数之和，双指针实现

解题思路

双指针解题，三个数分别为nums[i]，nums[left]，nums[right]，表示从第i个数出发，往后找到nums[left]和nums[right]

解题步骤：

1、将数组进行升序排序

2、sum = nums[i] + nums[left] + nums[right];

• sum<0,左指针向后移动
• sum>0,右指针向前移动
• sum=0，保存结果，左右指针同时向中间移动

注：当某一个指针指向的下一个值和当前值相同时，指针继续移动，避免出现重复解
在nums数组中，从第1个数出发一直找到从第n-3个数出发，得到所有解。

时间复杂度分析
排序的时间复杂度为O(nlogn),遍历查找的时间复杂度为O(n²),最终时间复杂度为O(n²)。

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int n = nums.size();
        if (n < 3)return {};//特例，n<3时必然不可能会有结果
        sort(nums.begin(), nums.end());
        int left, right;
        vector<vector<int>>ans;
        for (int i = 0; i < n; i++)
        {
            if (nums[i] > 0)break;//当找到第一个大于0的数时，后面不可能再有和为0的，直接返回
            left = i + 1;
            right = n - 1;
            while (left < right)
            {
                int sum = nums[i] + nums[left] + nums[right];
                if (sum == 0) {
                    ans.push_back({ nums[i],nums[left],nums[right] });
                    //左指针和右指针如果有重复的数跳过
                    //注意left和right用到了+1和-1，要添加限制条件以防数组越界
                    while (left<n-1&&nums[left] == nums[left + 1]) { left++; }
                    while (right>1&&nums[right] == nums[right - 1]) { right--; }
                    left++;
                    right--;
                }
                else if (sum < 0) {//小于0，左指针向后移动
                    while (left < n - 1 && nums[left] == nums[left + 1]) { left++;}
                    left++;
                }
                else {//大于0，右指针向前移动
                    while (right > 1 && nums[right] == nums[right - 1]) { right--; }
                    right--;
                }
            }
            //从某一个数出发时如果有重复的数字跳过
            while (i<n - 1 && nums[i] == nums[i + 1]) { i++; }
        }
        return ans;
    }
};