相等的和(哈希表)

在这里插入图片描述
在这里插入图片描述

输入样例12
5
2 3 1 3 2
6
1 1 2 2 2 1
输出样例1:
YES
2 6
1 2
输入样例23
1
5
5
1 1 1 1 1
2
2 3
输出样例2:
NO
输入样例34
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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值