Codeforces Round #486 (Div. 3) C "Equal Sums" （map+pair<>）

传送门

•题意

给k个数列，从中k个数列中找出任意2个数列 i ,j

使得数列i删除第x个数，和数列j删除第y个数的和相等

若存在，输出 i ,x 和 j,y

•思路

每个数列之间的联系为数列的和之间的差det

如果开二维数组记录每个数列之间的det的话，显然是不可行的_(:з」∠)_

这里用map<x ,pair<i ,j > >mp表示序列 i 删除第 j 个数后的总和为 x；

如果某两个序列各删除一个数，得到的总和相等，

也就是后一个序列得到的总和已存在(被前一个所记录)的话，就找到了

•代码

 #include<bits/stdc++.h>
 using namespace std;
 #define ll long long
 const int maxn=2e5+;
 map<int,pair<int,int> > mp;
 int a[maxn];
 ll sum;
 int main()
 {
     int k;
     cin>>k;
     for(int i=;i<=k;i++)
     {
         int n;
         cin>>n;
         sum=;
         for(int j=;j<=n;j++)
         {
             cin>>a[j];
             sum+=a[j];
         }
         for(int j=;j<=n;j++)
         {
             int x=sum-a[j];
             if(mp.count(x)&&mp[x].first!=i)
             {
                 cout<<"YES"<<endl;
                 cout<<i<<' '<<j<<endl;
                 cout<<mp[x].first<<' '<<mp[x].second<<endl;
                 return ;
             }
             mp[x]=make_pair(i,j);
         }
     }
     cout<<"NO"<<endl;
 }