HDU 4821 字符串HASH

题目链接

题意：

给两个整数m，l和一个字符串s。求字符串中有多少子串符合长为m*l，且将子串均分成m串任意两串皆不相同。

思路：

字符串HASH。先对字符串s进行HASH处理。

遍历初始下标[1,l]，每次取长为l的串，取其HASH值用map保存维护。当要取第m+1个串时将第1串踢出map存入第m+1个串，每次操作过后看看是不是map.size()==m，是就+1。

C++代码：

#include<bits/stdc++.h>
using namespace std;
typedef unsigned long long ull;
const int maxn = 100010;
const int mod1 = 1e9+7;
const int mod2 = 1e9+9;
const int base1 = 131;
const int base2 = 233;

struct My_Hash
{
    ull Hash1[maxn],p1[maxn];
    ull Hash2[maxn],p2[maxn];

    void Insert( char s[] )
    {
        int len = strlen(s+1);
        Hash1[0] = 0,p1[0] = 1;
        Hash2[0] = 0,p2[0] = 1;
        for ( int i=1 ; i<=len ; i++ )
        {
            p1[i] = p1[i-1]*base1%mod1;
            p2[i] = p2[i-1]*base2%mod2;
            Hash1[i] = (Hash1[i-1]*base1%mod1+(ull)s[i])%mod1;
            Hash2[i] = (Hash2[i-1]*base2%mod2+(ull)s[i])%mod2;
        }
    }
    pair<ull,ull> GetHash( int l , int r )
    {
        ull tmp1 = ( Hash1[r]-p1[r-l+1]*Hash1[l-1]%mod1+mod1 )%mod1;
        ull tmp2 = ( Hash2[r]-p2[r-l+1]*Hash2[l-1]%mod2+mod2 )%mod2;
        return make_pair( tmp1 , tmp2 );
    }
}S;

int m,l,ans;
char s[maxn];
map< pair<ull,ull> , int >mp;

int main()
{
    while( scanf ( "%d%d" , &m , &l )==2 )
    {
        scanf ( "%s" , s+1 );
        S.Insert(s);
        ans = 0;
        int len = strlen(s+1);
        for ( int i=1 ; i<=l&&i+m*l-1<=len ; i++ )
        {
            mp.clear();
            for ( int j=i,k=1 ; k<=m ; j+=l,k++ )
            {
                pair<ull,ull>t = S.GetHash( j , j+l-1 );
                mp[t]++;
            }
            if  ( mp.size()==m ) ans++;
            for ( int j=i+m*l ; j+l-1<=len ; j+=l )
            {
                pair<ull,ull>t1 = S.GetHash( j-m*l , j-m*l+l-1 );
                mp[t1]--;
                if ( mp[t1]==0 ) mp.erase(t1);
                pair<ull,ull>t2 = S.GetHash( j , j+l-1 );
                mp[t2]++;
                if ( mp.size()==m ) ans++;
            }
        }
        printf ( "%d\n" , ans );
    }
    return 0;
}