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

最新推荐文章于 2024-04-17 15:06:31 发布

原创最新推荐文章于 2024-04-17 15:06:31 发布 · 643 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++

本文介绍如何使用双指针技巧优化暴力搜索，解决在给定整数数组中找到四个元素之和等于目标值的问题，同时确保四元组无重复且按升序排列。通过实例和代码实现，提升查找效率避免超时。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

给定一个目标值 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;
}