Codeforces Round #486 (Div. 3) C

传送

题意：找出两个序列，各删除一个数后序列和相等，问删除哪两个序列的哪两个数
题解：预处理出删除每个数后的序列和，set存一下。

#include<bits/stdc++.h>
using namespace std;
int k,a,sum[200005];
vector<int> b[200005];
set<pair<int ,int> >s;
map<int,pair<int,int> >mp;
int main()
{
    cin>>k;
    for(int i=0;i<k;i++)
    {
        cin>>a;
        for(int j=0;j<a;j++)
        {
            int d;
            cin>>d;
            b[i].push_back(d);
            sum[i]+=d;
        }

    }
    for(int i=0;i<k;i++)
    {
        for(int j=0;j<a;j++)
        {
            int t=sum[i]-b[i][j];
            if(mp.find(t)!=mp.end()&&mp[sum[i]-b[i][j]].first!=i+1)
            {
                cout<<"YES"<<endl;
                cout<<i+1<<' '<<j+1<<endl;
                cout<<mp[sum[i]-b[i][j]].first<<' '<<mp[sum[i]-b[i][j]].second<<endl;
                return 0;
            }
            mp[sum[i]-b[i][j]]=make_pair(i+1,j+1);
        }
    }
    cout<<"NO"<<endl;
}