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)
共有两行,第一行一个整数n代表珠子的长度,(n<=200000),第二行是由空格分开的颜色ai(1<=ai<=n)。
也有两行,第一行两个整数,第一个整数代表能获得的最大不同的子串个数,第二个整数代表能获得最大值的k的个数,第二行输出所有的k(中间有空格)。
1 #include<bits/stdc++.h>
2 #define N 200005
3 #define mod 998532737
4 #define ll long long
5 using namespace std;
6 const ll x=12138;
7 int n,Max,tot,num;
8 int a[N],t[N];
9 ll b[N],h1[N],h2[N];
10 map<ll,bool> mp;
11 int main(){
12 scanf("%d",&n);
13 for (int i=1;i<=n;i++) scanf("%d",&a[i]);
14 b[0]=1;
15 for (int i=1;i<=n;i++) b[i]=(b[i-1]*x)%mod;
16 for (int i=1;i<=n;i++) h1[i]=(h1[i-1]*x+a[i])%mod;
17 for (int i=n;i>=1;i--) h2[i]=(h2[i+1]*x+a[i])%mod;
18 for (int k=1;k<=n;k++){
19 mp.clear();
20 if (n<Max*k) break;
21 num=0;
22 for (int i=1;i<=n/k;i++){
23 ll s1=((h1[i*k]-h1[(i-1)*k]*b[k])%mod+mod)%mod,s2=((h2[(i-1)*k+1]-h2[i*k+1]*b[k])%mod+mod)%mod;
24 if (!mp[s1*s2]){
25 num++;
26 mp[s1*s2]=true;
27 }
28 }
29 if (num>Max) Max=num,tot=1,t[1]=k; else
30 if (num==Max) t[++tot]=k;
31 }
32 printf("%d %d\n",Max,tot);
33 for (int i=1;i<=tot-1;i++)
34 printf("%d ",t[i]);
35 printf("%d\n",t[tot]);
36 return 0;
37 }
View Code