[BZOJ4199][Noi2015]品酒大会 树形DP+后缀自动机

由于要找后缀的前缀，所以先用反串建立SAM。

link边组成了后缀树。

两个子串的最长公共前缀是LCA的step

树形dp即可。

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #include<cstdlib>
 #include<cmath>
 #include<algorithm>
 #define maxn 1200000
 using namespace std;
 struct edge {
     int to,next;
 }e[maxn];
 int n,a[maxn];
 char s[maxn];
 int head[maxn],d;
 long long ans[maxn],mx[maxn],mn[maxn],sum[maxn];
 struct data {
     int last,cnt;
     int son[maxn][],link[maxn],step[maxn],v[maxn],size[maxn];
     int tmp[maxn];
     data() {last=cnt=;v[]=;}
     void extend(int c,int val) {
         int p=last,np=last=++cnt;step[np]=step[p]+;v[np]=val,size[np]=;
         while(p&&!son[p][c]) son[p][c]=np,p=link[p];
         if(!p) link[np]=;
         else {
             int q=son[p][c];
             if(step[q]==step[p]+) link[np]=q;
             else {
                 int nq=++cnt;v[nq]=;
                 memcpy(son[nq],son[q],sizeof(son[q]));
                 link[nq]=link[q];
                 link[q]=link[np]=nq;
                 step[nq]=step[p]+;
                 while(son[p][c]==q&&p) son[p][c]=nq,p=link[p];
             }
         }
     }
     void add(int u,int v) {e[d].next=head[u];e[d].to=v;head[u]=d++;}
     void build() {for(int i=;i<=cnt;i++) add(link[i],i),ans[i]=-21474836470000000000000LL;}
     void query(int x) {
         mx[x]=-;mn[x]=;
         if(v[x]!=) mx[x]=mn[x]=v[x];
         for(int i=head[x];i>=;i=e[i].next) {
             int to=e[i].to;
             query(to);
             if(mx[x]!=-&&mn[x]!=&&mx[to]!=-&&mn[to]!=) ans[step[x]]=max(ans[step[x]],max(mx[x]*mx[to],mn[to]*mn[x]));
             mx[x]=max(mx[x],mx[to]);mn[x]=min(mn[x],mn[to]);
             sum[step[x]]+=1LL*size[x]*size[to];size[x]+=size[to];
         }
 
     }
     void getans() {
         for(int i=n-;i>=;i--) sum[i]+=sum[i+],ans[i]=max(ans[i],ans[i+]);
         for(int i=;i<=n-;i++) if(sum[i]) printf("%lld %lld\n",sum[i],ans[i]); else printf("0 0\n");
     }
 
 }sam;
 
 int main() {
     memset(head,-,sizeof(head));
     scanf("%d",&n);
     scanf("%s",s+);
     for(int i=;i<=n;i++) scanf("%d",&a[i]);
     for(int i=n;i>=;i--) sam.extend(s[i]-'a',a[i]);
     sam.build();
     sam.query();
     sam.getans();
 
 }