XMUOJ-校外实训一 【三数之和】

题目：

给定一个目标值 target，请在整数数组 a中，找出三个元素(x,y,z) 使x+y+z==target。

请找到所有满足条件的三元组，并且请按从小到大的顺序输出所有合法的三元组。

注意：三元组中不允许包含重复数字，且输出的三元组中要求 x<y<z.

例如:给定target = 17，n=7, 数组a= [0, 2, 7, 10, 15,18,25]

结果返回两个三元组：(0,2, 15), (2,7,10)

暴力搜索超时严重

同样用双指针算法优化

注意：首先对数组进行排序，并且清除重复项

排序：sort

清除重复项：v.erase(unique(v.begin(),v.end()),v.end());

unique函数将重复项放在最后，erase将其清除

最后结果同样需要排序：sort（xx,xx,comp)

代码实现：

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
vector<vector<int> >threesum(vector<int>v,int target)
{
    vector<vector<int>> res;
    sort(v.begin(),v.end());
    v.erase(unique(v.begin(),v.end()),v.end());
    for(int i=0;i<v.size();i++)
    {
        for(int j=i+1,k=v.size()-1;k>j;j++)
        {
            while(j<k&&v[i]+v[k]+v[j]>target)k--;
            if(v[i]+v[k]+v[j]==target)
            {
                res.push_back({v[i],v[j],v[k]});
            }
        }
    }
    return res;
}
bool comp(vector<int>&a,vector<int>&b)
{
    if(a[0]<b[0])return true;
    else if(a[0]==b[0]&&a[1]<b[1])return true;
    else if(a[0]==b[0]&&a[1]==b[1]&&a[2]<b[2])return true;
    else return false;
}
int main(){
    int target,n;
    cin>>target>>n;
    vector<int>v(n);
    for(int i=0;i<n;i++)
        cin>>v[i];
    vector<vector<int> >res;
    res=threesum(v,target);
    sort(res.begin(),res.end(),comp);
    for(auto line : res)
        cout<<line[0]<<" "<<line[1]<<" "<<line[2]<<endl;
    return 0;
}