JZOJ5462 好文章

Description

　　nodgd写了一篇文章，自认为这是一篇好文章。nodgd的文章由𝑛个小写英文字母组成。文章的一个子串指的是文章中的一段连续的字母，子串的长度就是这一段的字母个数。nodgd在文章中用了排比、对偶、前后照应之类的手法，所以就有很多个子串是相同或者相近的。为了向大家证明这是一篇好文章，nodgd决定给自己的文章进行评分。nodgd首先确定了一个整数𝑚，然后统计出文章中有多少个不相同的长度为𝑚的子串，这个数量就是文章的评分。
　　然而，nodgd懒得老老实实计算这个评分了，就把任务丢给了你。

Input

　　第一行包含两个整数𝑛,𝑚，表示文章的长度和需要统计的子串长度。
　　第二行包含一个长度为𝑛的只包含小写字母的字符串。

Output

　　输出一行一个整数，表示文章的评分。

Sample Input

输入1：
　　5 3
　　aaaab

输入2：
　　9 3
　　abcabacba

Sample Output

输出1：
　　2
　　样例解释1：
　　长度为3的子串有3个，分别是aaa,aaa,aab，其中不同的只有2个。
输出2：
　　7
　　样例解释2
　　共有7个长度为3的子串，每个长度为3的子串都不同。

Data Constraint

　　对于30%的数据，1≤𝑚≤𝑛≤200；
　　对于50%的数据，1≤𝑚≤𝑛≤2000；
　　对于另外20%的数据，1≤𝑚≤50≤𝑛≤200000；
　　对于100%的数据，1≤𝑚≤𝑛≤200000。

Solution

　　字符串hash，为了提高容错率所以用双hash。

#include<cstdio>
#include<algorithm>
#define N 1000000007
#define M 1000000009
using namespace std;
struct arr
{
    long long x,y;
}h[2000000],s[200000];
int n,m,a[3000000];
long long p1,p2;
bool cmp(arr a,arr b)
{
    return a.x<b.x||a.x==b.x&&a.y<b.y;
}
int main()
{
    scanf("%d%d",&n,&m);
    int i=0;
    while (i<n)
    {
        char ch;
        scanf("%c",&ch);
        int t=(int)ch;
        if (t<=122&&t>=97)
        {
            i++;
            a[i]=t-96;
        }
    }
    p1=53;p2=97;
    for (int i=2;i<=m;i++)
    {
        p1=(p1*53)%N;
        p2=(p2*97)%M;
    }
    for (int i=1;i<=n;i++)
    {
        h[i].x=(h[i-1].x*53+a[i])%N;
        h[i].y=(h[i-1].y*97+a[i])%M;
    }
    for (int i=1;i<=n-m+1;i++)
    {
        s[i].x=(h[i+m-1].x-h[i-1].x*p1)%N;
        s[i].y=(h[i+m-1].y-h[i-1].y*p2)%M;
        while (s[i].x<0) s[i].x=s[i].x+N;
        while (s[i].y<0) s[i].y=s[i].y+M;
    }
    sort(s+1,s+n-m+2,cmp);
    int ans=0;
    for (int i=1;i<=n-m+1;i++)
        if (s[i].x!=s[i-1].x||s[i].y!=s[i-1].y)
            ans++;
    printf("%d\n",ans);
}