JZOJ5462. 【NOIP2017提高A组冲刺11.8】好文章

Description

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

然而，nodgd懒得老老实实计算这个评分了，就把任务丢给了你。

Input

第一行包含两个整数n,m，表示文章的长度和需要统计的子串长度。

第二行包含一个长度为n的只包含小写字母的字符串。

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≤m≤n≤200；

对于50%的数据，1≤m≤n≤2000；

对于另外20%的数据，1≤m≤50≤n≤200000；

对于100%的数据，1≤m≤n≤200000。

题解

字符串判断是否重复，

哈希判重。

将字符串转化成为一个数，

然后就可以了。

但是要用多几个质数，否则容易出错。

code

#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#define N 200003
#define ll long long
#define M1 2333333
#define M2 998244353
using namespace std;
char ch;
void read(int& n)
{
n=0;
for(ch=getchar();ch<'0'||ch>'9';ch=getchar());
for(;'0'<=ch && ch<='9';n=(n<<3)+(n<<1)+ch-48,ch=getchar());
}

void G(char &ch)
{
for(ch=getchar();ch<'a'||ch>'z';ch=getchar());
}

int n,m,ans;
ll z1
,z2
,p
,s1,s2,h[M1][2];

bool ins()
{
int x=s1%M1;
while((h[x][0]!=0 || h[x][1]!=0)&&(h[x][0]!=s1 || h[x][1]!=s2))x=(x+1)%M1;
if(h[x][0]==s1 && h[x][1]==s2)return 0;
h[x][0]=s1;
h[x][1]=s2;
return 1;
}

int main()
{
freopen("article.in","r",stdin);
freopen("article.out","w",stdout);
read(n);read(m);
z1[0]=z2[0]=1;
for(int i=1;i<=m;i++)
z1[i]=z1[i-1]*m%M1,z2[i]=z2[i-1]*m%M2,G(ch),p[i]=ch-48;
s1=s2=0;
for(int i=1;i<=m;i++)
s1=(z1[m-i]*p[i]+s1)%M1,s2=(z2[m-i]*p[i]+s2)%M2;
ans=1;
ins();
for(int i=m+1;i<=n;i++)
{
G(ch),p[i]=ch-48;
s1=(s1-z1[m-1]*p[i-m]%M1+M1)%M1;
s2=(s2-z2[m-1]*p[i-m]%M2+M2)%M2;
s1=(s1*m+p[i])%M1;
s2=(s2*m+p[i])%M2;
if(ins())ans++;
}
printf("%d\n",ans);
return 0;
}