输入样例1:
2
5
2 3 1 3 2
6
1 1 2 2 2 1
输出样例1:
YES
2 6
1 2
输入样例2:
3
1
5
5
1 1 1 1 1
2
2 3
输出样例2:
NO
输入样例3:
4
6
2 2 2 2 2 2
5
2 2 2 2 2
3
2 2 2
5
2 2 2 2 2
输出样例3:
YES
2 2
4 1
算法:
只需要在k个序列中任意取两个序列,分别取出一个数,然后看看取完数后这两个序列的和是否相等,设一个序列的和为sum,取出a1,剩下的和就为sum-a1,同理第二次数为sum-a2…
那么我们可以用一个哈希表来存储每次操作完后的和,和记录它操作的{行,列},每次操作完后查表,如果存在和一样,且对应的行数是不一样的,那么就是我们要找的结果,如果没有,那么把当前操作存到哈希表里;
C++代码:
#include <iostream>
#include <unordered_map>
#define x first
#define y second
using namespace std;
typedef pair<int,int> PII;
const int N = 200010;
int w[N];
int main(){
int k,n;
unordered_map<int,PII> S;
scanf("%d",&k);
for(int r = 1; r <= k; r++){
scanf("%d",&n);
int sum = 0;
for(int i = 1; i <= n; i++){
scanf("%d",&w[i]);
sum += w[i];
}
for(int i = 1; i <= n; i++){
int t = sum - w[i];
if(S.count(t) && S[t].x != r){
puts("YES");
printf("%d %d\n",S[t].x,S[t].y);
printf("%d %d\n",r,i);
return 0;
}
S[t] = {r,i};
}
}
puts("NO");
return 0;
}