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

  1. C.相等和
    给你k个整数序列。每个序列的长度为ni。
    你需要选择两个序列i和j(i!=j)使得从i中删除一个元素之后的元素和等于从j中删除一个元素之后的元素和相等。
    注意要求是从每两个你选的序列中删除一个元素。
    假设空集合的和为0。
    输入:第一行是一个k(2<=k<=2*1e5)——序列的和。
    接下来有k对,没对包括一个序列。
    第一行是第i对序列的长度ni(1<=ni<=1*1e5),第二行是第i对序列ai1,ai2,……aini。
    每个序列的元素是(-1e4~1e4)
    所有给你的序列的长度不超过2*1e5(i.e<=n1+n2+…+nk<=2*1e5)
    输出:
    如果不可能选择出两个序列满足上述条件。则输出“NO”(不带引号)。否则第一行输出“YES”(不带引号),第二行输出两个整数i,x,在第三行输出两个整数j,y。代表第i个序列去掉第x个位置的元素后剩下的所有元素的和等于第j个序列去掉第y个位置的元素剩下的元素和。
    选择的两个序列的必须是不同的序列,你可以以任意顺序输出答案。
    如果有多组答案,输出它们中的任意一个。
    提示:
    在第一个按例中有两个序列:【2,3,1,3,2,】和【1,1,2,2,2,1】。你可以在第一个序列中删除第二个元素得到【2,1,3,2】并且你可以从第二个序列中删除第六个元素得到【1,1,2,2,2,】。然后这两个序列的结果都是8,它们的和相等。
  2. 题目链接:http://codeforces.com/contest/988/problem/C

  3. 通过的代码:

#include<iostream>
#include<map>
#define mp(a,b) make_pair(a,b)
#define fr first
#define sc second
using namespace std;
const int N=2e5+6;
int a[N],n,k;
typedef pair<int,int>pa;
map<int,pa>e;
int main(){
    ios::sync_with_stdio(false);cin.tie(0);
    cin>>k;int a1=-1,a2,b1=-1,b2;
    for(int i=1;i<=k;i++){
        if(a1!=-1)continue;
        int sum=0;cin>>n;
        for(int j=1;j<=n;j++)cin>>a[j],sum+=a[j];
        for(int j=1;j<=n;j++)if(e.count(sum-a[j])){
            a1=e[sum-a[j]].fr;a2=e[sum-a[j]].sc;
            b1=i,b2=j;break;
        }
        for(int j=1;j<=n;j++)e[sum-a[j]]=mp(i,j);
    }
    if(a1==-1)puts("NO");
    else cout<<"YES"<<endl<<a1<<" "<<a2<<endl<<b1<<" "<<b2<<endl;
}

猜你喜欢

转载自blog.csdn.net/light2chasers/article/details/80631894
今日推荐