洛谷P2178 [NOI2015]品酒大会——题解

题目传送门
题目大意：
有一个长为n的字符串，每一个字符有一个权值，称相同的两个长度为r的子串的开头是“1相似”、“2相似”、$\cdots$、“（r-1）相似”、“r相似”的，试问有多少k相似的字符（1<=k<=n）以及k相似的两个字符的权值积最大为多少。

---

思考过程：
我是在洛谷的后缀数组分类里找到这道题的，那么就先码一遍后缀数组的模板2333。
看过题目之后，我们发现，i相似字符的询问会包含j相似字符的询问（i < j）于是我们想到将所有的height按从大到小排序，这样如果求出了每个height的答案做一遍前缀和就能得到所有的结果
回想height的定义，是排名相邻的两个后缀的最长公共前缀的长度，我们使用并查集，每次计算了答案就将他们合并。这样每次计算时就相当于计算了一个并查集中的所有后缀的贡献了

---

具体做法：
1.以height为第一关键字排序
2.首先每个点创建并查集，size设为1，maxx和minn设为自身的权值（需要记录minn是为了防止负负得正）
3.扫排序后的数组，如果两个后缀的开头不在同一个并查集中，ans[height][0]就加上两个并查集各自大小的积，ans[height][1]就变成max(maxx*maxx,minn*minn,ans[height][1])，然后将他们合并即可
4。统计答案时要考虑LCP大的对小的的贡献，所以从大的开始做一遍前缀和

---

代码：

#include <bits/stdc++.h>
using namespace std;

const int maxn=3e5+100;
const long long inf=1e18;
char s[maxn];
int a[maxn];
int c[maxn],sa[maxn],y[maxn],x[maxn],height[maxn],rk[maxn];
int id[maxn],fa[maxn];
long long ans[maxn][2],minn[maxn],maxx[maxn],size[maxn];
int n,m,num;

void get_sa()
{
    for(int i=1;i<=n;i++) ++c[x[i]=s[i]];
    for(int i=2;i<=m;i++) c[i]+=c[i-1];
    for(int i=n;i>=1;i--) sa[c[x[i]]--]=i;
    for(int k=1;k<=n;k<<=1)
    {
        num=0;
        for(int i=n-k+1;i<=n;i++) y[++num]=i;
        for(int i=1;i<=n;i++) if(sa[i]>k) y[++num]=sa[i]-k;
        for(int i=1;i<=m;i++) c[i]=0;
        for(int i=1;i<=n;i++) ++c[x[i]];
        for(int i=2;i<=m;i++) c[i]+=c[i-1];
        for(int i=n;i>=1;i--) sa[c[x[y[i]]]--]=y[i],y[i]=0;
        swap(x,y);
        x[sa[1]]=1,num=1;
        for(int i=2;i<=n;i++) x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?num:++num;
        if(num==n) break;
        m=num;
    }
}

void get_height()
{
    int k=0;
    for(int i=1;i<=n;i++) rk[sa[i]]=i;
    for(int i=1;i<=n;i++)
    {
        if(rk[i]==1) continue;
        if(k) k--;
        int j=sa[rk[i]-1];
        while(j+k<=n&&i+k<=n&&s[j+k]==s[i+k]) k++;
        height[rk[i]]=k;
    }
}

bool cmp1(int x,int y)
{
    return height[x]>height[y]; 
}

int find(int x)
{
    if(x==fa[x]) return x;
    return fa[x]=find(fa[x]);   
}

void stick(int x,int y)
{
    int p1=find(x),p2=find(y);
    int k=height[x];

    ans[k][0]+=size[p1]*size[p2];
    ans[k][1]=max(max(maxx[p1]*maxx[p2],minn[p1]*minn[p2]),ans[k][1]);

    maxx[p1]=max(maxx[p1],maxx[p2]);
    minn[p1]=min(minn[p1],minn[p2]);
    fa[p2]=p1;
    size[p1]+=size[p2];

}

void work()
{
    for(int i=1;i<=n;i++)
    {
        fa[i]=i;
        size[i]=1;
        minn[i]=maxx[i]=a[sa[i]];
    }
    fa[0]=1;
    for(int i=1;i<=n;i++)
    {
        if(find(id[i])!=find(id[i]-1))
            stick(id[i],id[i]-1);   
    }
}

int main()
{
    m=122;
    scanf("%d",&n);
    scanf("%s",s+1);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    get_sa();
    get_height();
    for(int i=1;i<=n;i++)
    {
        id[i]=i;
        ans[i][0]=0;
        ans[i][1]=-inf; 
    }
    sort(id+1,id+1+n,cmp1);
    work();
    for(int i=n-2;i>=0;i--)
    {
        ans[i][0]+=ans[i+1][0];
        ans[i][1]=max(ans[i][1],ans[i+1][1]);   
    }
    for(int i=0;i<=n-1;i++) printf("%lld %lld\n",ans[i][0],ans[i][0]?ans[i][1]:0);
    return 0;   
}