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+5;
map<int,pair<int,int> > mp;
int a[maxn];
ll sum;
int main()
{
    int k;
    cin>>k;
    for(int i=1;i<=k;i++)
    {
        int n;
        cin>>n;
        sum=0;
        for(int j=1;j<=n;j++)
        {
            cin>>a[j];
            sum+=a[j];
        }
        for(int j=1;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 0;
            }
            mp[x]=make_pair(i,j);
        }
    }
    cout<<"NO"<<endl;
}