Codeforces Round #486 (Div. 3) C. Equal Sums(stl应用)-优快云博客

本文介绍了一种使用STL解决EqualSums问题的方法。通过定义sum_i为第i行的和，利用map存储每行删除一个元素后的和，快速判断是否存在两行在删除一个元素后和相等的情况。

Equal Sums

这里写图片描述

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

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

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

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

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

对于每一个新的行，我们找前面是否出现过 $sum_i$ 等于我现在的某一个 $sum_j$ ，
只需 $mp.lower$ _ $bound(sum_j)$ 返回迭代器即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<vector>
#include<set>
#include<queue>
#include<limits.h>
#include<string.h>
#include<map>
#include<list>
#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;
}