P3498 [POI2010]KOR-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代表珠子的長度，（），第二行是由空格分開的顏色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

題解：

枚舉長度+哈希

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/hash_policy.hpp>
using namespace __gnu_pbds;
using namespace std;
const int maxn=200000+5;
const int B=200191;
typedef unsigned long long ull;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1; ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
ull mul[maxn],s1[maxn],s2[maxn];
int n,mx,num;
int a[maxn],b[maxn];
gp_hash_table<ull,bool>mp;
ull gethash(int l,int r)
{
    ull t;
    if(l<=r) t=s1[r]-s1[l-1]*mul[r-l+1];
    else
    t=s2[r]-s2[l+1]*mul[l-r+1];
    return t;
}
void solve(int x)
{
    if(mx*x>n) return;
    int ans=0;
    mp.clear();
    for(int i=1;i<=n;i+=x)
    if(i+x-1<=n)
    {
        ull t=gethash(i,i+x-1)*gethash(i+x-1,i);
        if(mp[t]) continue;
        else ans++;
        mp[t]=1;
    }
    if(ans>mx) mx=ans,num=0;
    if(ans==mx) b[++num]=x;
}
int main()
{
    n=read();
    for(int i=1;i<=n;i++) a[i]=read();
    mul[0]=1;
    for(int i=1;i<=n;i++) mul[i]=mul[i-1]*B;
    for(int i=1;i<=n;i++)
        s1[i]=s1[i-1]*B+a[i];
    for(int i=n;i>=1;i--)
        s2[i]=s2[i+1]*B+a[i];
    for(int i=1;i<=n;i++) solve(i);
    printf("%d %d\n",mx,num);
    for(int i=1;i<=num;i++)
    {
        printf("%d",b[i]);
        if(i!=num) printf(" ");
    }
    return 0;
}