当前位置: 代码迷 >> 综合 >> 【NOI2015】bzoj4199 品酒大会【解法一】
  详细解决方案

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

热度:84   发布时间:2024-01-13 10:38:20.0

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

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

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
#define LL long long
const 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]);
}