2081: [Poi2010]Beads
Time Limit: 10 Sec Memory Limit: 259 MB
Submit: 1111 Solved: 369
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
解析:
直接暴力枚举长度,用字符串哈希计算个数,然后是一些小优化。。。
代码:
#include <bits/stdc++.h>
#include <tr1/unordered_map>
using namespace std;
using namespace std::tr1;
const int Max=200005;
int n,m,len,ans,sum,maxx;
int id[Max],a[Max];
unsigned long long hash1[Max],hash2[Max],Pow[Max];
unordered_map<unsigned long long,bool>v;
vector<int>q;
inline int get_int()
{
int x=0,f=1;
char c;
for(c=getchar();(!isdigit(c))&&(c!='-');c=getchar());
if(c=='-'){f=-1;c=getchar();}
for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+c-'0';
return x*f;
}
inline void pre()
{
Pow[0]=1;
for(int i=1;i<=n;i++)
{
hash1[i] = hash1[i-1] * 13331 + a[i];
Pow[i] = Pow[i-1] * 13331;
}
for(int i=n;i>=1;i--)
hash2[i] = hash2[i+1] * 13331 + a[i];
}
//inline unsigned long long calc(int i){return (hash1[i] - hash1[i-len] * Pow[len]) * (hash2[i-len+1] - hash2[i+1] * Pow[len]);}
inline void solve(int len)
{
if(n/len < maxx) return; //如果最大数量都小于当前最大数,则直接返回
v.clear();
for(int i=len;i<=n;i+=len)
if(!v[(hash1[i] - hash1[i-len] * Pow[len]) * (hash2[i-len+1] - hash2[i+1] * Pow[len])])
{
id[len]++;
v[(hash1[i] - hash1[i-len] * Pow[len]) * (hash2[i-len+1] - hash2[i+1] * Pow[len])]=1;
}
if(id[len] > maxx) q.clear(),maxx=id[len];
if(id[len] == maxx) q.push_back(len);
}
inline int mx(int x,int y){return x > y ? x : y;}
inline void print(int x)
{
if(x > 9) print(x/10);
putchar(x%10 + '0');
}
int main()
{
n=get_int();
for(int i=1;i<=n;i++) a[i]=get_int();
pre();
for(int k=1;k<=n;k++) solve(k);
print(maxx),putchar(' '),print(q.size()),putchar('\n');
for(int i=0;i<q.size();i++) print(q[i]),putchar(' ');
return 0;
}