标签:
题意:给定五个集合,每个集合有n个数,从每个集合各取一个数使和为0。如果能输出Yes,不能则输出No。
看了网上的题解有很多解法,个人感觉这题暴力就能过。
首先将一二两个集合合并为数组a,再将三四两个集合合并数组b,然后进行从小到大排序,
设第五个数组为数组c,依次遍历c中的每一个数,看在a,b中是否存在两个数的和与c中的数的和为0,
将数组a,b进行sort排序,然后一个正序即从小到大遍历,一个逆序从大到小遍历,
如果三个数的和小于0,a数组后移一位,否则b后移一位。
这题WA了两次,第一次WA后发现else if用的是三个i,忘了改成i,j,k了。
第二次WA后发现Yes输出成了YES,No输出成了NO。
#include <iostream> #include <stdio.h> #include <string.h> #include <algorithm> using namespace std; long long a[40005],b[40005],c[205],d[205],e[205],f[205],g[205]; long long i,j,x,k,l,n,t,flag=0,p; int main() { scanf("%lld",&t); while(t--) { flag=0; //输入 scanf("%lld",&n); for(i=0;i<n;i++) scanf("%lld",&d[i]); for(i=0;i<n;i++) scanf("%lld",&e[i]); for(i=0;i<n;i++) scanf("%lld",&f[i]); for(i=0;i<n;i++) scanf("%lld",&g[i]); for(i=0;i<n;i++) scanf("%lld",&c[i]); //合并 int cnt1=0; for(i=0;i<n;i++) for(j=0;j<n;j++) a[cnt1++]=d[i]+e[j]; int cnt2=0; for(i=0;i<n;i++) for(j=0;j<n;j++) b[cnt2++]=f[i]+g[j]; //三四合并 sort(a,a+cnt1); sort(b,b+cnt2); //计算 for(i=0;i<n;i++) //第五个数组 c[] { j=0; k=cnt2-1; while(j<cnt1 && k>=0) { if(a[j]+b[k]+c[i]==0LL) { flag=1; break; } else if(a[j]+b[k]+c[i]<0) j++; else k--; } } if(flag) cout<<"Yes"<<endl; else cout<<"No"<<endl; } return 0; }
标签:
原文地址:http://www.cnblogs.com/Ritchie/p/5352268.html