BZOJ2081[POI2010] Beads
Description
Zxl有一次决定制造一条项链,她以非常便宜的价格买了一长条鲜艳的珊瑚珠子,她现在也有一个机器,能把这条珠子切成很多块(子串),每块有k(k>0)个珠子,如果这条珠子的长度不是k的倍数,最后一块小于k的就不要拉(nc真浪费),保证珠子的长度为正整数。 Zxl喜欢多样的项链,为她应该怎样选择数字k来尽可能得到更多的不同的子串感到好奇,子串都是可以反转的,换句话说,子串(1,2,3)和(3,2,1)是一样的。写一个程序,为Zxl决定最适合的k从而获得最多不同的子串。 例如:这一串珠子是: (1,1,1,2,2,2,3,3,3,1,2,3,3,1,2,2,1,3,3,2,1), k=1的时候,我们得到3个不同的子串: (1),(2),(3) k=2的时候,我们得到6个不同的子串: (1,1),(1,2),(2,2),(3,3),(3,1),(2,3) k=3的时候,我们得到5个不同的子串: (1,1,1),(2,2,2),(3,3,3),(1,2,3),(3,1,2) k=4的时候,我们得到5个不同的子串: (1,1,1,2),(2,2,3,3),(3,1,2,3),(3,1,2,2),(1,3,3,2)
Input
共有两行,第一行一个整数n代表珠子的长度,(n<=200000),第二行是由空格分开的颜色ai(1<=ai<=n)。
Output
也有两行,第一行两个整数,第一个整数代表能获得的最大不同的子串个数,第二个整数代表能获得最大值的k的个数,第二行输出所有的k(中间有空格)。
Sample Input
21
1 1 1 2 2 2 3 3 3 1 2 3 3 1 2 2 1 3 3 2 1
Sample Output
6 1
2
Solution:
有一个神奇的东西(调和级数):
知道了这个以后,这道题就是简单的字符串哈希了。
原本想用Hash判重,结果交了一面的WA…于是就用set了。但是网上看题解,貌似有一个Hash用的B特别强,在此先记下: B=200019
#include<stdio.h>
#include<iostream>
#include<set>
#define ll unsigned long long
#define B 200019
#define M 200005
using namespace std;
ll Base[M],H1[M],H2[M];
int A[M],Ans[M];
set<ll>S;
void Rd(int &res){
char c;res=0;
while(c=getchar(),!isdigit(c));
do{
res=(res<<1)+(res<<3)+(c^48);
}while(c=getchar(),isdigit(c));
}
int main(){
int n,ans=0,sz=0;
Rd(n);
for(int i=1;i<=n;i++)
Rd(A[i]);
Base[0]=1;
for(int i=1;i<=n;i++)
Base[i]=Base[i-1]*B;
for(int i=1;i<=n;i++)
H1[i]=H1[i-1]*B+A[i];
for(int i=n;i>=1;i--)
H2[i]=H2[i+1]*B+A[i];
for(int k=1;k<=n;k++){
if(n/k<ans)break;
S.clear();
int cnt=0;
for(int j=k;j<=n;j+=k){
ll t1=H1[j]-H1[j-k]*Base[k];
ll t2=H2[j-k+1]-H2[j+1]*Base[k];
ll t3=t1*t2;
if(S.count(t3))continue;
S.insert(t3);
cnt++;
}
if(cnt>ans){
ans=cnt;
sz=0;
Ans[++sz]=k;
}else if(cnt==ans)Ans[++sz]=k;
}
printf("%d %d\n",ans,sz);
for(int i=1;i<=sz;i++)
printf("%d%c",Ans[i],i==sz?'\n':' ');
return 0;
}