码迷,mamicode.com
首页 > 其他好文 > 详细

POJ 1200 字符串HASH

时间:2016-07-04 11:27:02      阅读:139      评论:0      收藏:0      [点我收藏+]

标签:

 

题目链接:http://poj.org/problem?id=1200

题意:给定一个字符串,字符串只有NC个不同的字符,问这个字符串所有长度为N的子串有多少个不相同。

思路:字符串HASH,因为只有NC个不同的字符,所以我们可以把字符串看成是一个NC进制的串,然后计算出字符串的前缀HASH。然后枚举起点判断子串的HASH值是否已经存在。因为有了前缀HASH值,所以转移是O(1)的。

#define _CRT_SECURE_NO_DEPRECATE
#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<set>
using namespace std;
typedef long long int LL;
typedef unsigned int uint;
const int MAXN=16000000+5;
const int mod=16000000+5;
int Hash[MAXN],vis[256];
char str[MAXN];
bool cnt[MAXN];
int pow_mod(int a,int b){
    int ans=1;
    while (b)
    {
          if (b & 1)
              ans = (1LL*ans*a)%mod;
          b >>= 1 ;
          a = (1LL*a*a)%mod;
    }
    return ans;
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m)){
        int hashNum=0,ans=0;
        scanf("%s",str+1); 
        memset(vis,-1,sizeof(vis));
        memset(cnt,false,sizeof(cnt));
        int len=strlen(str+1);
        for(int i=1;i<=len;i++){
            if(vis[str[i]]==-1){
                vis[str[i]]=hashNum;
                Hash[i]=hashNum++;
            }
            else{
                Hash[i]=vis[str[i]];
            }
        }
        int P=pow_mod(m,n-1),pNum=0;
        for(int i=1;i<=len;i++){
            if(i>=n){
                pNum=(((pNum-Hash[i-n]*P)*m+Hash[i])%mod+mod)%mod;
                if(!cnt[pNum]){
                    cnt[pNum]=true;
                    ans++;
                }
            }
            else{
                pNum=(pNum*m+Hash[i])%mod;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}

 

POJ 1200 字符串HASH

标签:

原文地址:http://www.cnblogs.com/kirito520/p/5639691.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!