【LeetCode】5994. 查找给定哈希值的子串

class Solution {
public:
    string subStrHash(string s, int p, int m, int k, int hash) {
        int n = s.size();
        unsigned long long pow[20010];  //long long防止溢出
        pow[0] = 1;
        for (int i = 1; i <= k; ++i) {
            pow[i] = (pow[i - 1] * p) % m;  //提前计算power的k次幂
        }
        
        for(int i=0;i<n;++i)   //初始化字母代表的数值，方便代码书写
        {
            s[i]-='a'-1;
        }

        int ans;
        unsigned long long sum=0;

        for(int j=0; j<k ; ++j)   //对第一个窗口n-k~n-1进行特殊判断
        {
            sum=(sum+(s[n-k+j])*pow[j]) % m;
        }
        if(sum==hash)
        {
            ans=n-k;
        }

        for (int i = n-k-1 ; i >=0 ; i--)  //滑动窗口
        {
            sum=(sum*p-((s[i+k])*pow[k])%m+s[i]+m)%m;   //加m以防止变为负数
            if(sum==hash)
            {
                ans=i;
            }
        }

        string b;
        for(int j=0;j<k;++j)
        {
            b+=(s[ans+j]+('a'-1)); //还原结果，加上'a'-1
        }

        return b;
    }
};

思路基本写在注释里了。

观察题意，因为解的结构的相似性，所以很容易联想到滑动窗口算法；但是由于除法不满足取模公式，所以这个滑动窗口需要倒着滑。

首先构造初始窗口，再用循环依次滑到最开始，因为要求最先出现的子串，所以得遍历一遍字符串。

代码中做了存储幂结果的优化，以及使用long long防止溢出。