Leetcode---Three sum三数之和---双指针法（C++）

Three sum三数之和

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

解法思路：

双指针法（先对数组排序，再使用左右指针）

C++代码，可直接在IDE中运行的完整主函数，包括了输入获取，类函数调用，以及结果的打印输出。

//three sum三数之和-双指针法

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;                             //保存结果
        sort(nums.begin(),nums.end());                       //先排序，以便后续使用双指针法
        for(int k=0;k<nums.size();k++)
        {
            if(nums[k]>0) break;
            if(k>0 && nums[k]==nums[k-1]) continue;          //结果去重处理
            int i=k+1;            //左指针
            int j=nums.size()-1;  //右指针
            int target = -nums[k];
            while(i<j)
            {
                if(nums[i]+nums[j]==target)
                {
                    res.push_back({nums[k],nums[i],nums[j]}); //保存一组结果
                    while(i<j && nums[i]==nums[i+1]) i++;     //结果去重处理
                    while(i<j && nums[j]==nums[j-1]) j--;     //结果去重处理
                    i++;
                    j--;
                }
                else if(nums[i]+nums[j]<target)
                    i++;                                     //小于target,移动左指针
                else
                    j--;                                     //大于target,移动右指针
            }
        }
        return res;
    }
};
//主函数
int main(){
    Solution s;
    vector<int> nums;
    cout<<"Please input the size of nums:"<<endl;
    int size,temp;
    cin>>size;                                         //数组大小设置
    for(int t=0;t<size;t++)                            //输入获取待处理的nums
    {
        cin>>temp;
        nums.push_back(temp);
    }
    cout<<"The size of nums:"<<nums.size()<<endl;
    vector<vector<int>> re;
    re=s.threeSum(nums);
    vector<vector<int>> *p=&re;
    vector<vector<int>>::iterator it;                  //第1层迭代器
    vector<int>::iterator it2;                         //第2层迭代器
    int index0=-1;
    for(it= p->begin(); it != p->end(); it++)          //迭代读取每组答案
    {
        index0++;
        vector<int> *p2=&(*it);
        int index1=0;
        for(it2= p2->begin(); it2 != p2->end(); it2++) //迭代读取并打印，每组答案的每个元素
            cout<<"answer"<<index0<<" "<<(index1++)<<" is :"<<*it2<<endl;
    }
    cout<<"总共有"<<index0+1<<"组答案"<<endl;          //输出答案组数

    return 0;
}