B-寻寻觅觅寻不到

牛客练习赛88-B-寻寻觅觅寻不到
循环哈希匹配,匹配成功就退出。
比赛时一直wa,原因在于,代码第33行,循环的终止条件应该是 i <= lens - k + 1,而不是 i <= lens。
思路是,枚举每一个空位,将长度为k的子串插到当前的空,判断得到的新串的哈希值是否等于母串的哈希值。
wa的原因就是,末尾的子串前是不存在空位的了,不能拿去枚举。
在这里插入图片描述

#include <bits/stdc++.h>
using namespace std;
#define int unsigned long long
const int N = 3e5 + 5, mod = 998244353;
unsigned long long h1[N], h2[N], P[N];
signed main()
{
    int n;
    cin >> n;
    P[0] = 1;
    for (int i = 1; i <= N - 1; i++) P[i] = P[i - 1] * 131;
    while(n--) {
        string s, t;
        int k;
        cin >> s >> t >> k;
        int lens = s.length(), lent = t.length();
        if (lens != lent || k > lens) {
            printf("NO\n");
            continue;
        }
        s = " " + s;
        t = " " + t;
        for (int i = 1; i <= lens; i++) {
            h1[i] = h1[i - 1] * 131 + s[i];
            h2[i] = h2[i - 1] * 131 + t[i];
        }
        if (h1[lens] == h2[lens]) {
            printf("YES\n");
            continue;
        }
        int hk = h2[lens] - h2[lens - k] * P[k];
        bool ck = 0;
        for (int i = 1; i <= lens - k + 1; i++) {
            int l = h2[i - 1];
            int mid = h2[lens - k] - h2[i - 1] * P[lens - k - i + 1];
            int pp = l * P[lens - (i - 1)] + hk * P[lens - k - (i - 1)] + mid;
            if (pp == h1[lens]) {
                ck = 1;
                break;
            }
        }
        if (ck) printf("YES\n");
        else printf("NO\n");
        for (int i = 1; i <= lens; i++) h1[i] = h2[i] = 0;
    }
    return 0;
}

PS:几个特判删掉好像也不会WA

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值