Equal Sums CodeForces - 988C(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.


题意:给出n组数列,编号1-n,每组数列的长为k,数字编号1-k,问,是否存在这样的两组数列,将这两组数列减去任意一个值,使得这两组数列相等。如果有,输出”YES“,并且输出组号,以及删除的数字的编号,如果不存在这样的两个数列,则输出“NO”。(如果有多个答案,输出任意一个)

思路:先将每组数列的和sum算出来,然后for循环用map记录sum-a[i]的值关于  数列编号和减去的数的编号  的映射关系即可。(a[i]记录的是当前输入的数列)

#include "iostream"
#include "map"
using namespace std;
const int Max=2e5+10;
int a[Max];
typedef pair<int,int> P;//first记录数列编号,second记录数字的编号
int main()
{
    ios::sync_with_stdio(false);
    int t,n,sum,flag=0;
    cin>>t;
    map<int,P> mp;
    mp.clear();
    for(int i=0;i<t;i++){
        cin>>n;
        sum=0;
        for(int j=0;j<n;j++){
            cin>>a[j];
            sum+=a[j];//记录和
        }
        for(int j=0;j<n;j++){
            int k=sum-a[j];//枚举所有减去一个数可能的情况
            if(mp.count(k)&&!flag&&mp[k].first!=i+1){//如果mp[k]有映射关系并且不是自己,那么说明找到了这样的两个数组,直接输出。之后就不用进入这个判断了
                cout<<"YES"<<endl;
                cout<<i+1<<" "<<j+1<<endl;
                cout<<mp[k].first<<" "<<mp[k].second<<endl;
                flag=1;
            }
            P p;
            p.first=i+1;
            p.second=j+1;
            mp[k]=p;
        }
    }
    if(!flag)
        cout<<"NO"<<endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_41874469/article/details/80551802