题目:
给定一个目标值 target,请在整数数组 A中,找出四个元素(a,b,c,d) 使a+b+c+d==target。
请找到所有满足条件的四元组,并且请按从小到大的顺序输出所有合法的四元组。
注意:四元组中不允许包含重复数字,且输出的四元组中要求 a<b<c<d
例如:给定target = 17,n=7, 数组a= [0, 2, 5, 10, 15,18,25]
结果返回两个四元组:(0,2,5,10)
暴力搜索超时
双指针算法优化
具体见两数之和,三数之和
代码:
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
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 if(a[0]==b[0]&&a[1]==b[1]&&a[2]==b[2]&&a[3]<b[3])return true;
else return false;
}
vector<vector<int>> foursum(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;j<v.size();j++)
{
for(int k=j+1,l=v.size()-1;l>k;k++)
{
while(v[i]+v[j]+v[k]+v[l]>target&&l>k)l--;
if(v[i]+v[j]+v[k]+v[l]==target)
{
res.push_back({v[i],v[j],v[k],v[l]});
}
}
}
}
return res;
}
int main(){
int target,n;
cin>>target>>n;
vector<int>v(n);
vector<vector<int>>res;
for(int i=0;i<n;i++)cin>>v[i];
res=foursum(v,target);
sort(res.begin(),res.end(),comp);
for(auto line : res)
cout<<line[0]<<" "<<line[1]<<" "<<line[2]<<" "<<line[3]<<endl;
return 0;
}