【C++算法】8.双指针_三数之和

文章目录

• • 题目链接：
  • 题目描述：
  • 解法
  • C++ 算法代码：
  • 图解

题目链接：

15.三数之和

题目描述：

解法

C++ 算法代码：

class Solution 
{
    public:
    vector<vector<int>> threeSum(vector<int>& nums) 
    {
        vector<vector<int>> ret;//用ret记录结果
        // 1. 排序
        sort(nums.begin(), nums.end());
        // 2. 利用双指针解决问题
        int n = nums.size();
        for(int i = 0; i < n; ) // 固定数 a
        {
            if(nums[i] > 0){
                break; // 小优化
            }
            int left = i + 1, right = n - 1, target = -nums[i];
            while(left < right)
            {
                int sum = nums[left] + nums[right];
                if(sum > target){
                    right--;
                }
                else if(sum < target){
                    left++;
                }
                else{//说明找到最终结果
                    ret.push_back({nums[i], nums[left], nums[right]});//把三个数放到ret里面，{}会形成一个vector int的数组放到ret里面
                    left++, right--;
                    // 去重操作 left 和 right
                    while(left < right && nums[left] == nums[left - 1]){
                        left++;
                    }
                    while(left < right && nums[right] == nums[right + 1]){
                        right--;
                    }
                }
            }
             
            i++;
            // 去重 i
            while(i < n && nums[i] == nums[i - 1]){
                i++;
            }
        }
        return ret;
    }
};

图解

nums=[-4，-4，-1，0，0，0，1，1，4，4，5，6]

1. n=12，进入for循环，i=0，left=1，right=11，target=4，left < right进入while循环，sum=2<target，left++

2. sum=-1+6=5>target，right--

3. sum=-1+5=4=targe，把[-4，-1，5]放到ret里面，left++, right--

4. sum=0+4=4=targe，把[-4，0，4]放到ret里面，left++, right--，执行去重操作

5. sum=1+1=2<targe，left++，跳出while循环，i++，执行去重，然后开始第二轮双指针算法。

6. 后面的步骤类似，就不多赘述了。