22蔚来9 mancher 字符串哈希 概率计算 前缀和优化

· · 个人记录

first E magic spell

自己的ac代码:

#include <bits\stdc++.h>
using namespace std;
typedef unsigned long long ULL;
typedef pair<ULL, ULL> llp;
const ULL X = 131;
string str[6];
vector<ULL> p, h, pp, hh;
string s;
set<llp> mmset[6];
ULL ans;
ULL mod = 1e9 + 7, mmod = 1e9 + 3;
void BKDR_hash() //初始化,这里没有使用双哈希。
{
    hh[0] = s[0];
    h[0] = s[0];
    p[0] = 1;
    pp[0] = 1;
    for (int i = 1, size = s.size(); i < size; i++)
    {
        h[i] = (h[i - 1] * X + s[i]) % mod;
        p[i] = (p[i - 1] * X) % mod;
        hh[i] = (hh[i - 1] * X + s[i]) % mmod;
        pp[i] = (pp[i - 1] * X) % mmod;
    }
}
llp get_hash(int l, int r)
{
    ULL a, b;
    a = l ? (h[r] - h[l - 1] * p[r - l + 1]%mod+mod) % mod : h[r];
    b = l ? (hh[r] - hh[l - 1] * pp[r - l + 1]%mmod+mmod) % mmod : hh[r];
    return llp(a, b);
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    int kk;
    cin >> kk;
    for (int i = 1; i <= kk; i++)
        cin >> str[i];
    for (int j = 1; j <= kk; j++)
    {

        //完成初始化。
        s.clear();
        h.clear();
        hh.clear();
        pp.clear();
        p.clear();
        s = str[j];
        //先
        int n = s.length();
        p.resize(n);
        pp.resize(n);
        hh.resize(n);
        h.resize(n);
        vector<int> d1(n);
        BKDR_hash();
        for (int i = 0, l = 0, r = -1, k; i < n; i++)
        {
            k = (i > r) ? 1 : min(d1[l + r - i], r - i + 1);
            mmset[j].insert({ s[i], s[i] });
            while (i - k >= 0 && i + k < n && s[i - k] == s[i + k])
            {
                mmset[j].insert(get_hash(i - k, i + k));
                k++;
            }
            d1[i] = k--;
            if (i + k > r)
                l = i - k, r = i + k;
        }
        vector<int> d2(n);
        for (int i = 0, l = 0, r = -1, k; i < n; i++)
        {
            k = (i > r) ? 0 : min(d2[l + r - i + 1], r - i + 1);
            while (i - k - 1 >= 0 && i + k < n && s[i - k - 1] == s[i + k])
            {
                mmset[j].insert(get_hash(i - k - 1, i + k));
                k++;
            }
            d2[i] = k--;
            if (i + k > r)
            {
                l = i - k - 1;
                r = i + k;
            }
        }
    }
    //所有的哈希值已经记录完毕
    for (auto i : mmset[kk])
    {
        bool found = true;
        for (int j = 1; j < kk; j++)

        {
            if (mmset[j].count(i) == 0)
            {
                found = false;
                break;
            }
        }
        if (found)
            ans++;
    }
    cout << ans << '\n';
}

不够简洁。优化方向如下

记得补!!!

Second Two Frogs

新知识补充,由于提高check的精度,对结果采取求模运算作为答案。分数求模运算中引入逆元。

A/B\%P=A\cdot B^{-1}\%P

逆元myself