Codeforces Round #486 (Div. 3)C. Equal Sums【模拟,map】

C. Equal Sums
time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output

You are given kk sequences of integers. The length of the ii-th sequence equals to nini.

You have to choose exactly two sequences ii and jj (iji≠j) such that you can remove exactly one element in each of them in such a way that the sum of the changed sequence ii (its length will be equal to ni1ni−1) equals to the sum of the changed sequence jj (its length will be equal to nj1nj−1).

Note that it's required to remove exactly one element in each of the two chosen sequences.

Assume that the sum of the empty (of the length equals 00) sequence is 00.

Input

The first line contains an integer kk (2k21052≤k≤2⋅105) — the number of sequences.

Then kk pairs of lines follow, each pair containing a sequence.

The first line in the ii-th pair contains one integer nini (1ni<21051≤ni<2⋅105) — the length of the ii-th sequence. The second line of the ii-th pair contains a sequence of nini integers ai,1,ai,2,,ai,niai,1,ai,2,…,ai,ni.

The elements of sequences are integer numbers from 104−104 to 104104.

The sum of lengths of all given sequences don't exceed 21052⋅105, i.e. n1+n2++nk2105n1+n2+⋯+nk≤2⋅105.

Output

If it is impossible to choose two sequences such that they satisfy given conditions, print "NO" (without quotes). Otherwise in the first line print "YES" (without quotes), in the second line — two integers iixx (1ik,1xni1≤i≤k,1≤x≤ni), in the third line — two integers jjyy (1jk,1ynj1≤j≤k,1≤y≤nj). It means that the sum of the elements of the ii-th sequence without the element with index xx equals to the sum of the elements of the jj-th sequence without the element with index yy.

Two chosen sequences must be distinct, i.e. iji≠j. You can print them in any order.

If there are multiple possible answers, print any of them.

Examples
input
Copy
2
5
2 3 1 3 2
6
1 1 2 2 2 1
output
Copy
YES
2 6
1 2
input
Copy
3
1
5
5
1 1 1 1 1
2
2 3
output
Copy
NO
input
Copy
4
6
2 2 2 2 2 2
5
2 2 2 2 2
3
2 2 2
5
2 2 2 2 2
output
Copy
YES
2 2
4 1
Note

In the first example there are two sequences [2,3,1,3,2][2,3,1,3,2] and [1,1,2,2,2,1][1,1,2,2,2,1]. You can remove the second element from the first sequence to get [2,1,3,2][2,1,3,2] and you can remove the sixth element from the second sequence to get [1,1,2,2,2][1,1,2,2,2]. The sums of the both resulting sequences equal to 88, i.e. the sums are equal.

Codeforces (c) Copyright 2010-2018 Mike Mirzayanov
The only programming contests Web 2.0 platform
Server time: Jun/06/2018 14:18:58 UTC+8 (d3).
Desktop version, switch to  mobile version.

Privacy Policy

题目大意:给定t组数据,每组数据有n个,问能否找到两组数据,是两组数据个删除掉一个数并且两组数据的和相等,如果有则找出两组数据是第几组,并且删除的数据在改组数据中的位置。

题目上反映的一一对应关系说明要用到map,然后就接着道题学了一下map的基本用法,后来看大牛的博客发现还可以用map和pair的组合,就有些了一遍。附上两种写法,以及运行时间。

第一种写法:

38984137 2018-06-06 08:26:06 d-xiang-ha C - Equal Sums GNU C++17 Accepted 327 ms 16600 KB
#include<bits/stdc++.h>
using namespace std;
map<int,int> ma1;//用来记录数据的组数
map<int,int> ma2;//用来记录数据的位数
int a[1000010];
int main(){
    int t,r1,r2,r3,r4;
    cin>>t;
    bool flag=false;
    for(int i=1;i<=t;i++){
        int n;
        cin>>n;
        int sum=0;
        if(flag) continue;
        for(int j=1;j<=n;j++){
            cin>>a[j];
            sum=sum+a[j];
        }
        for(int j=1;j<=n;j++){
            map<int,int>::iterator ite=ma1.find(sum-a[j]);
            if(ite!=ma1.end()){
                if(ma1[sum-a[j]]!=i){
                    r1=ma1[sum-a[j]];
                    r2=ma2[sum-a[j]];
                    r3=i;
                    r4=j;
                    flag=true;
                    break;
                }
            }
            else{
                ma1[sum-a[j]]=i;
                ma2[sum-a[j]]=j;
            }
        }
    }
    if(!flag) cout<<"NO"<<endl;
    else{
        cout<<"YES"<<endl;
        cout<<r1<<" "<<r2<<endl;
        cout<<r3<<" "<<r4<<endl;
    }
    return 0;
}第二种写法:
38985054 2018-06-06 09:17:59 d-xiang-ha C - Equal Sums GNU C++17 Accepted 296 ms 11900 KB
#include<bits/stdc++.h>
using namespace std;
map<int,pair<int,int> > ma;
int a[1000010];
int main(){
    int t;
    cin>>t;
    for(int i=1;i<=t;i++){
        int n;
        cin>>n;
        int  sum=0;
        for(int j=1;j<=n;j++){
            cin>>a[j];
            sum=sum+a[j];
        }
        for(int j=1;j<=n;j++){
            if(ma.count(sum-a[j])){
                cout<<"YES"<<endl;
                cout<<ma[sum-a[j]].first<<" "<<ma[sum-a[j]].second<<endl;
                cout<<i<<" "<<j<<endl;
                return 0;
            }
        }
        for(int j=1;j<=n;j++){
            ma[sum-a[j]]={i,j};
        }
    }
    cout<<"NO"<<endl;

    return 0;
}


猜你喜欢

转载自blog.csdn.net/xiang_hehe/article/details/80594285