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

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

热度:50   发布时间:2024-01-13 10:38:09.0

解法一【后缀树+树形dp】见【这里】

把后缀数组中的元素按height从大到小合并并在合并的过程中统计答案,用并查集维护集合,记录大小和最值。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define LL long long
const int maxn=1000010;
const LL oo=5e18;
char s[maxn];
int sa[maxn],rank[maxn],height[maxn],cnt[maxn],f[maxn],
a[maxn],fa[maxn],que[maxn],
n;
LL size[maxn],mn1[maxn],mn2[maxn],mx1[maxn],mx2[maxn],num[maxn],ans[maxn];
int cmp(int x,int y)
{return height[x]>height[y];
}
int find(int x)
{return x==fa[x]?x:fa[x]=find(fa[x]);
}
int main()
{int m=26,p,x,y;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++) cnt[rank[i]=s[i]-'a'+1]++;for (int i=2;i<=m;i++) cnt[i]+=cnt[i-1];for (int i=n;i;i--) sa[cnt[rank[i]]--]=i;for (int k=1;;k<<=1){p=0;for (int i=n-k+1;i<=n;i++) f[++p]=i;for (int i=1;i<=n;i++)if (sa[i]>k) f[++p]=sa[i]-k;for (int i=1;i<=m;i++) cnt[i]=0;for (int i=1;i<=n;i++) cnt[rank[f[i]]]++;for (int i=2;i<=m;i++) cnt[i]+=cnt[i-1];for (int i=n;i;i--) sa[cnt[rank[f[i]]]--]=f[i];for (int i=1;i<=n;i++) f[i]=rank[i];rank[sa[1]]=1;for (int i=2;i<=n;i++)if (f[sa[i]]==f[sa[i-1]]&&f[sa[i]+k]==f[sa[i-1]+k])rank[sa[i]]=rank[sa[i-1]];else rank[sa[i]]=rank[sa[i-1]]+1;m=rank[sa[n]];if (m>=n) break;}for (int i=1;i<=n;i++){if (height[rank[i]]=height[rank[i-1]]) height[rank[i]]--;while (s[i+height[rank[i]]]==s[sa[rank[i]-1]+height[rank[i]]]) height[rank[i]]++;}for (int i=1;i<n;i++) que[i]=i+1;sort(que+1,que+n,cmp);for (int i=1;i<=n;i++){fa[i]=i;size[i]=1;mx1[i]=mn1[i]=a[i];mx2[i]=-oo;mn2[i]=oo;}for (int i=1;i<n;i++) ans[i]=-oo;for (int i=1;i<n;i++){x=find(sa[que[i]]);y=find(sa[que[i]-1]);num[height[que[i]]]+=size[x]*size[y];size[x]+=size[y];if (mn1[y]<=mn1[x]){mn2[x]=mn1[x];mn1[x]=mn1[y];mn2[x]=min(mn2[x],mn2[y]);}else mn2[x]=min(mn2[x],mn1[y]);if (mx1[y]>=mx1[x]){mx2[x]=mx1[x];mx1[x]=mx1[y];mx2[x]=max(mx2[x],mx2[y]);}else mx2[x]=max(mx2[x],mx1[y]);fa[y]=x;ans[height[que[i]]]=max(ans[height[que[i]]],max(mx1[x]*mx2[x],mn1[x]*mn2[x]));}for (int i=n-2;i>=0;i--){ans[i]=max(ans[i],ans[i+1]);num[i]+=num[i+1];}for (int i=0;i<n;i++) printf("%lld %lld\n",num[i],ans[i]==-oo?0:ans[i]);
}