Codeforces Round #486 (Div. 3) C. Equal Sums(stl应用)

题意：给你几个数列，问是否存在这样两个数列使得他们删去各自的某一个成员后二者的和相等。

因为题目限制了n_i<=2e5,所以这题的解法很多。

感觉自己最初的解法有点复杂了，在这里我讲一个STL应用的解法。

我们定义sumi为第i行的和，id_i==i，pos表示第i行的第pos个.

map<sumi,pair<id_i,pos> > 这样便简单的存下了我们需要的所有信息。

对于每一个新的行，我们找前面是否出现过sumi等于我现在的某一个sumj，

只需mp.lower(sumj)返回迭代器即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

#define inf int(0x3f3f3f3f)
#define mod int(1e9+7)
#define eps double(1e-6)
#define pi acos(-1.0)
#define lson  root << 1
#define rson  root << 1 | 1



map< int,pair<int ,int> >mmp;

map< int ,pair<int,int> >::iterator it;

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int zu;
    mmp.clear();
    cin>>zu;
    for(int t=1;t<=zu;t++)
    {
        int n;
        cin>>n;
        vector<int> v;
        int sum=0;
        int x;
        for(int i=0;i<n;i++) cin>>x,v.push_back(x),sum+=x;
        for(int i=0;i<n;i++)
        {
            int ans=sum-v[i];
            if(t!=1)
            {
                it=mmp.lower_bound(ans);
                if(it!=mmp.end()&&it->first==ans&&it->second.first!=t)
                {
                    cout<<"YES"<<endl;
                    cout<<it->second.first<<" "<<it->second.second<<endl;
                    cout<<t<<" "<<i+1<<endl;
                    return 0;
                }

            }
            mmp[ans]=make_pair(t,i+1);
        }
    }
    cout<<"NO"<<endl;
}