【NOI2015】bzoj4199 品酒大会【解法一】

来源:互联网 发布:淘宝店铺怎么免费推广 编辑:程序博客网 时间:2024/05/30 23:53

解法二【后缀数组+并查集】见【这里】

建出后缀树以后,因为任意两个后缀的LCP都是他们的LCA,所以在树上dp,维护大小和最值就可以了。

#include<cstdio>#include<cstring>#include<vector>#include<algorithm>using namespace std;#define LL long longconst int maxn=600010;const LL oo=5e18;char s[maxn];vector<int> son[maxn];int trans[maxn][27],fa[maxn],val[maxn],a[maxn],n,tot=1;LL mn1[maxn],mx1[maxn],mn2[maxn],mx2[maxn],size[maxn],num[maxn],ans[maxn];void dfs(int u){    int v;    for (vector<int>::iterator it=son[u].begin();it!=son[u].end();++it)    {        v=*it;        dfs(v);        num[val[u]]+=size[u]*size[v];        size[u]+=size[v];        if (mn1[v]<=mn1[u])        {            mn2[u]=mn1[u];            mn1[u]=mn1[v];            mn2[u]=min(mn2[u],mn2[v]);        }        else mn2[u]=min(mn2[u],mn1[v]);        if (mx1[v]>=mx1[u])        {            mx2[u]=mx1[u];            mx1[u]=mx1[v];            mx2[u]=max(mx2[u],mx2[v]);        }        else mx2[u]=max(mx2[u],mx1[v]);    }    if (size[u]>=2) ans[val[u]]=max(ans[val[u]],max(mn1[u]*mn2[u],mx1[u]*mx2[u]));}int main(){    int last=1,p,np,q,nq,x;    scanf("%d",&n);    scanf("%s",s+1);    for (int i=1;i<=n;i++) scanf("%d",&a[i]);    for (int i=1;i<n-i+1;i++) swap(s[i],s[n-i+1]);    for (int i=1;i<=n;i++)    {        x=s[i]-'a'+1;        val[np=++tot]=val[p=last]+1;        last=np;        while (p&&!trans[p][x])        {            trans[p][x]=np;            p=fa[p];        }        if (!p) fa[np]=1;        else        {            q=trans[p][x];            if (val[q]==val[p]+1) fa[np]=q;            else            {                val[nq=++tot]=val[p]+1;                fa[nq]=fa[q];                fa[q]=fa[np]=nq;                for (int j=1;j<=26;j++) trans[nq][j]=trans[q][j];                while (p&&trans[p][x]==q)                {                    trans[p][x]=nq;                    p=fa[p];                }            }        }    }    for (int i=1;i<=tot;i++)    {        mn1[i]=mn2[i]=oo;        mx1[i]=mx2[i]=-oo;    }    p=1;    for (int i=1;i<=n;i++)    {        p=trans[p][s[i]-'a'+1];        mn1[p]=mx1[p]=a[n-i+1];        size[p]=1;    }    for (int i=0;i<n;i++) ans[i]=-oo;    for (int i=1;i<=tot;i++) son[fa[i]].push_back(i);    dfs(1);    for (int i=n-2;i>=0;i--)    {        num[i]+=num[i+1];        ans[i]=max(ans[i],ans[i+1]);    }    for (int i=0;i<n;i++) printf("%lld %lld\n",num[i],ans[i]==-oo?0:ans[i]);}
原创粉丝点击