Codeforces Round #486 (Div. 3) CEqual Sums

C.相等和 给你k个整数序列。每个序列的长度为ni。 你需要选择两个序列i和j（i！=j）使得从i中删除一个元素之后的元素和等于从j中删除一个元素之后的元素和相等。 注意要求是从每两个你选的序列中删除一个元素。 假设空集合的和为0。 输入：第一行是一个k（2<=k<=2*1e5）——序列的和。 接下来有k对，没对包括一个序列。 第一行是第i对序列的长度ni（1<=ni<=1*1e5），第二行是第i对序列ai1，ai2，……aini。 每个序列的元素是（-1e4~1e4） 所有给你的序列的长度不超过2*1e5（i.e<=n1+n2+…+nk<=2*1e5） 输出： 如果不可能选择出两个序列满足上述条件。则输出“NO”（不带引号）。否则第一行输出“YES”（不带引号），第二行输出两个整数i，x，在第三行输出两个整数j，y。代表第i个序列去掉第x个位置的元素后剩下的所有元素的和等于第j个序列去掉第y个位置的元素剩下的元素和。 选择的两个序列的必须是不同的序列，你可以以任意顺序输出答案。 如果有多组答案，输出它们中的任意一个。 提示： 在第一个按例中有两个序列：【2,3,1,3,2,】和【1,1,2,2,2,1】。你可以在第一个序列中删除第二个元素得到【2,1,3,2】并且你可以从第二个序列中删除第六个元素得到【1,1，2,2,2,】。然后这两个序列的结果都是8，它们的和相等。

题目链接：http://codeforces.com/contest/988/problem/C

通过的代码：

#include<iostream> #include<map> #define mp(a,b) make_pair(a,b) #define fr first #define sc second using namespace std; const int N=2e5+6; int a[N],n,k; typedef pair<int,int>pa; map<int,pa>e; int main(){ ios::sync_with_stdio(false);cin.tie(0); cin>>k;int a1=-1,a2,b1=-1,b2; for(int i=1;i<=k;i++){ if(a1!=-1)continue; int sum=0;cin>>n; for(int j=1;j<=n;j++)cin>>a[j],sum+=a[j]; for(int j=1;j<=n;j++)if(e.count(sum-a[j])){ a1=e[sum-a[j]].fr;a2=e[sum-a[j]].sc; b1=i,b2=j;break; } for(int j=1;j<=n;j++)e[sum-a[j]]=mp(i,j); } if(a1==-1)puts("NO"); else cout<<"YES"<<endl<<a1<<" "<<a2<<endl<<b1<<" "<<b2<<endl; }