bzoj4199: [Noi2015]品酒大会

http://www.lydsy.com/JudgeOnline/problem.php?id=4199

　　给定一个字符串S，设Si表示S从i开始的的后缀。若Si, Sj存在长度为r的公共前缀，则称i和j是r相似的。每个后缀有一个价值vi，一对后缀Si与Sj的价值是vi*vj。对于每个r=0~n-1，统计有多少对r相似的后缀，以及r相似的后缀的最大价值(n≤300000)。

　　我们先后缀排序，并得到height值。我们考虑动态维护集合，保证集合内的答案已经统计过了，在合并的时候更新集合与集合间的答案即可。初始每个后缀自己构成一个大小为1的集合。按height从大到小枚举每个后缀，将rank为i的后缀与rank为i-1的后缀合并，他们更新的r即为当前的height。更新答案并用并查集维护集合即可。

#include<bits/stdc++.h>
using namespace std;
const int maxn=300015;
typedef long long int64;
typedef pair<int,int> PII;
int n,v[maxn];char s[maxn];
struct Tsuffix_array{
    int sum[maxn],sa[maxn],rank[maxn],tsa[maxn],trank[maxn];
    bool cmp(int i,int j,int l){
        if (i+l>n||j+l>n) return 0;
        return rank[i]==rank[j]&&rank[i+l]==rank[j+l];
    }
    void suffix_sort(){
        int m=255,p,i,j;
        for (i=0;i<=m;++i) sum[i]=0;
        for (i=1;i<=n;++i) ++sum[rank[i]=s[i]];
        for (i=1;i<=m;++i) sum[i]+=sum[i-1];
        for (i=n;i>=1;--i) sa[sum[rank[i]]--]=i;
        for (j=1,p=0;p<n;j<<=1,m=p){
            for (p=0,i=n-j+1;i<=n;++i) tsa[++p]=i;
            for (i=1;i<=n;++i) if (sa[i]>j) tsa[++p]=sa[i]-j;
            for (i=0;i<=m;++i) sum[i]=0;
            for (i=1;i<=n;++i) ++sum[rank[tsa[i]]];
            for (i=1;i<=m;++i) sum[i]+=sum[i-1];
            for (i=n;i>=1;--i) sa[sum[rank[tsa[i]]]--]=tsa[i];
            for (p=trank[sa[1]]=1,i=2;i<=n;++i) trank[sa[i]]=cmp(sa[i],sa[i-1],j)?p:++p;
            memcpy(rank,trank,sizeof(int)*(n+1));
        }
    }
    int height[maxn];
    void get_height(){
        for (int h=0,i=1;i<=n;++i){
            if (rank[i]==1) continue;
            for (h?--h:0;s[i+h]==s[sa[rank[i]-1]+h];++h);
            height[rank[i]]=h; 
        }
    }
    int fa[maxn],siz[maxn],fmx[maxn],fmn[maxn];
    int find(int u){return u==fa[u]?u:fa[u]=find(fa[u]);}
    void merge(int x,int y){
        if (siz[x]<siz[y]) swap(x,y);
        fa[y]=x;siz[x]+=siz[y];
        fmx[x]=max(fmx[x],fmx[y]);
        fmn[x]=min(fmn[x],fmn[y]);
    }
    PII t[maxn];
    int64 ans1[maxn],ans2[maxn];
    void get_ans(){
        for (int i=2;i<=n;++i) t[i-1]=make_pair(height[i],i);
        for (int i=1;i<=n;++i) fmx[rank[i]]=fmn[rank[i]]=v[i];
        sort(t+1,t+n,greater<PII>());memset(ans2,200,sizeof(ans2));
        for (int i=1;i<=n;++i){fa[i]=i;siz[i]=1;}
        for (int i=1;i<=n-1;++i){
            int x=find(t[i].second-1),y=find(t[i].second);
            ans1[t[i].first]+=1ll*siz[x]*siz[y];
            ans2[t[i].first]=max(ans2[t[i].first],1ll*fmx[x]*fmx[y]);
            ans2[t[i].first]=max(ans2[t[i].first],1ll*fmn[x]*fmx[y]);
            ans2[t[i].first]=max(ans2[t[i].first],1ll*fmx[x]*fmn[y]);
            ans2[t[i].first]=max(ans2[t[i].first],1ll*fmn[x]*fmn[y]);
            merge(x,y);
        }
        for (int i=n-1;i>=0;--i){
            ans1[i]+=ans1[i+1];
            ans2[i]=max(ans2[i],ans2[i+1]);
        }
        for (int i=0;i<=n-1;++i)
            printf("%lld %lld\n",ans1[i],!ans1[i]?0:ans2[i]);
    }
}SA;
void init(){
    scanf("%d%s",&n,s+1);
    for (int i=1;i<=n;++i) scanf("%d",&v[i]);
}
void work(){
    SA.suffix_sort();
    SA.get_height();
    SA.get_ans();
}
int main(){
    init();
    work(); 
    return 0;
}