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

传送门

 

•题意

　　给你 k 个序列，第 i 个序列含有 n_i 个元素；

　　问是否存在两个不同的序列 i,j；

　　满足序列 i 删除某个数后的总和等于序列 j 删除某个数后的总和；

•题解

　　定义 map<int , pair<int ,int > >f;

　　为了方便表述，将上述定义改为 map<x ,pair<y ,z > >f;

　　表示序列 y 删除第 z 个数后的总和为 x；

　　对于第 i 个序列，在删除第 j 个数后，求出删除后的总和 cur；

　　判断 f 中是否存在 cur，如果存在，直接输出，反之继续找；

•Code

#include<bits/stdc++.h>
using namespace std;
#define pii pair<int ,int >
const int maxn=2e5+50;

int k;
int a[maxn];
map<int ,pii >f;

int main()
{
    scanf("%d",&k);
    for(int i=1;i <= k;++i)
    {
        int n;
        scanf("%d",&n);

        int sum=0;
        for(int j=1;j <= n;++j)
        {
            scanf("%d",a+j);
            sum += a[j];
        }

        for(int j=1;j <= n;++j)
        {
            int cur=sum-a[j];

            ///判断之前的序列是否存在cur
            ///f[cur].first != i 是必要的
            ///测试如下样例便可明白
            /**
                1
                2
                1 1
            */
            if(f.count(cur) && f[cur].first != i)
            {
                puts("YES");
                pii tmp=f[cur];
                printf("%d %d\n",i,j);
                printf("%d %d\n",tmp.first,tmp.second);
                return 0;
            }
            f[cur]=pii(i,j);
        }
    }
    puts("NO");

    return 0;
}