1461:Beads
【题目描述】
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(中间有空格)。
【输入样例】
21
1 1 1 2 2 2 3 3 3 1 2 3 3 1 2 2 1 3 3 2 1
【输出样例】
6 1
2
虽然数据是整数,但还是把他们看成是字符串来处理。 因为子串(1,2,3)和(3,2,1)是一样的,所以必须正向反向都进行字符串哈希处理。然后对长度1~n的字串分别进行暴力计算。 对每一个字串计算正向哈希值h1,反向哈希值h2,假如两个字串是一样的(不管是正向还是反向),那么h1 * h2的值必然相同。
用map存储h1 * h2的值用来判重,注意如果用map分别存储h1和h2会超时。
#include <bits/stdc++.h>
typedef unsigned long long ll;
using namespace std;
int n,a[200005],ans[200005],total=1;
ll p[200005],suml[200005],sumr[200005],b=1000000007;
map<ll,int>mp;
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
int i,j,c=0,maxn=1;
cin>>n;
p[0]=1;
for(i=1;i<=n;i++)
{
cin>>a[i];
p[i]=p[i-1]*b,suml[i]=suml[i-1]*b+a[i];
}
for(i=n;i>=1;i--)
{
sumr[i]=sumr[i+1]*b+a[i];
}
for(i=1;i<=n;i++) /**< i用来表示字串长度,最大为n,不可以用n/2*/
{
mp.clear();
c=0;
for(j=i;j<=n;j+=i)
{
ll h1=suml[j]-suml[j-i]*p[i];
ll h2=sumr[j-i+1]-sumr[j+1]*p[i];
ll h3=h1*h2; /**< 正反认为同一个串,所以如果两串相同,
正向hash值h1和反向hash值h2的乘积的值必然一样 */
if(!mp[h3])
{
c++;
mp[h3]=1;
}
}
if(c>maxn)
{
maxn=c;
total=1;
ans[total++]=i;
}
else if(c==maxn)
{
ans[total++]=i;
}
}
cout<<maxn<<' '<<total-1<<endl;
for(i=1;i<total;i++)
cout<<ans[i]<<' ';
return 0;
}