C. Equal Sums

题目链接
You are given k sequences of integers. The length of the i-th sequence equals to ni.

You have to choose exactly two sequences i and j (i≠j) such that you can remove exactly one element in each of them in such a way that the sum of the changed sequence i (its length will be equal to ni−1) equals to the sum of the changed sequence j (its length will be equal to nj−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 0) sequence is 0.

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

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

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

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

The sum of lengths of all given sequences don’t exceed 2⋅105, i.e. n1+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 i, x (1≤i≤k,1≤x≤ni), in the third line — two integers j, y (1≤j≤k,1≤y≤nj). It means that the sum of the elements of the i-th sequence without the element with index x equals to the sum of the elements of the j-th sequence without the element with index y.

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

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

input
2
5
2 3 1 3 2
6
1 1 2 2 2 1

output
YES
2 6
1 2

input
3
1
5
5
1 1 1 1 1
2
2 3

output
NO

input
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
YES
2 2
4 1

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

题解:
我们可以使用map<long long, pair<long long, long long> > ,map的第一个关键字用来存储序列删除掉 j (j= 1; j<= n; j++)后的sum; 第二个关键字的两个long long分别代表序列编号和删除的值的下标,先将每组数列的和sum算出来,然后for循环用map记录sum-a[i]的值关于 数列编号和减去的数的编号 的映射关系即可。(a[i]记录的是当前输入的数列)

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 2e5 + 10;
LL k,n,sum,flag,a[N];
map <LL,pair<LL,LL>> mp;
int main(){
	ios :: sync_with_stdio(false);
	cin.tie(0);
	cin >> k;
	flag = false;
	for(int i = 1;i <= k;i++){
		cin >> n;
		sum = 0;
		for(int j = 1;j <= n;j++){
			cin >> a[j];
			sum += a[j];
		}
		if(flag) continue;
		for(int j = 1;j <= n;j++){
			if(mp.count(sum - a[j]) != 0){
				flag = true;
				cout << "YES" << endl << i << " " << j << endl << mp[sum - a[j]].first << " " << mp[sum -
				 a[j]].second << endl;
				break;
			}
		}
		for(int j = 1;j <= n;j++) mp[sum - a[j]] = {i,j};
	}
	if(!flag) cout << "NO" << endl;
	return 0;
} 
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值