bzoj4199 [Noi2015]品酒大会

传送门
貌似bzoj并没有题目描述······那从COGS上弄过来吧
【题目描述】
一年一度的“幻影阁夏日品酒大会”隆重开幕了。大会包含品尝和趣味挑战两个环节，分别向优胜者颁发“首席品酒家”和“首席猎手”两个奖项，吸引了众多品酒师参加。
在大会的晚餐上，调酒师 Rainbow 调制了 n 杯鸡尾酒。这 n 杯鸡尾酒排成一行，其中第i杯酒（1≤i≤n）被贴上了一个标签 si，每个标签都是 26 个小写英文字母之一。设Str(l,r)表示第 l 杯酒到第 r 杯酒的r−l+1个标签顺次连接构成的字符串。若 Str(p,po)=Str(q,qo)，其中1≤p≤po≤n，1≤q≤qo≤n，p≠q，po−p+1=qo−q+1=r，则称第 p 杯酒与第 q 杯酒是“r相似” 的。当然两杯“r相似”（r>1）的酒同时也是“1 相似”、“2 相似”、…、“(r−1) 相似”的。特别地，对于任意的 1≤p,q≤n，p≠q，第 p 杯酒和第 q杯酒都是“0相似”的。
在品尝环节上，品酒师 Freda 轻松地评定了每一杯酒的美味度，凭借其专业的水准和经验成功夺取了“首席品酒家”的称号，其中第 i 杯酒 （1≤i≤n）的美味度为 ai。现在 Rainbow 公布了挑战环节的问题：本次大会调制的鸡尾酒有一个特点，如果把第 p 杯酒与第 q 杯酒调兑在一起，将得到一杯美味度为 ap∗aq 的酒。现在请各位品酒师分别对于 r=0,1,2,…,n−1，统计出有多少种方法可以选出 2 杯“r相似”的酒,并回答选择 2 杯“r相似”的酒调兑可以得到的美味度的最大值。
【输入格式】
输入文件的第 1 行包含 1 个正整数 n，表示鸡尾酒的杯数。
第 2 行包含一个长度为 n 的字符串 S，其中第 i 个字符表示第 i 杯酒的标签。
第 3 行包含 n 个整数，相邻整数之间用单个空格隔开，其中第 i 个整数表示第 i 杯酒的美味度 ai。
【输出格式】
输出文件包括 n 行。第 i 行输出 2 个整数，中间用单个空格隔开。第 1 个整数表示选出两杯“(i−1)相似”的酒的方案数，第 2 个整数表示选出两杯“(i−1)相似”的酒调兑可以得到的最大美味度。若不存在两杯“(i−1)相似”的酒，这两个数均为 0。
【样例输入】
10
ponoiiipoi
2 1 4 7 4 8 3 6 4 7
【样例输出】
45 56
10 56
3 32
0 0
0 0
0 0
0 0
0 0
0 0
0 0
【提示】
用二元组 (p,q) 表示第 p 杯酒与第 q 杯酒。
0 相似：所有 45 对二元组都是 0 相似的，美味度最大的是 8×7=56。
1 相似：(1,8) (2,4) (2,9) (4,9) (5,6) (5,7) (5,10) (6,7) (6,10) (7,10)，最大的 8×7=56。
2 相似：(1,8) (4,9) (5,6)，最大的 4×8=32。
没有 3,4,5,…,9 相似的两杯酒，故均输出 0。
n<=300000,∣ai∣≤1000000000
【数据范围】

然而只有洛谷上有数据范围啊
【来源】
NOI2015

补一个洛谷上的测试数据：
样例输入：
12
abaabaabaaba
1 -2 3 -4 5 -6 7 -8 9 -10 11 -12
样例输出：
66 120
34 120
15 55
12 40
9 27
7 16
5 7
3 -4
2 -4
1 -4
0 0
0 0

题解

非常恶心的一道后缀数组题，从昨天晚上开始做，一直调到今天晚上8点
数据处理和普通后缀数组没什么区别，只是后面需要用到并查集来维护每个子串的连通性，以及所在连通块的size和最大最小值（数据有负值，有可能答案由两个负数相乘）。一定要注意自己维护的是哪个字符串以及它的表示！
ps：我把答案初值设为-1e17竟然只得了75分，对自己数据范围估计的能力已经放弃了

CODE：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=300010;
struct edge
{
    int nxt,to;
}a[N];
ll ans[N],Max[N],Min[N],tot[N],size[N];
char s[N];
int t1[N],t2[N],sa[N],height[N],rank[N],c[N],f[N],head[N];
int n,num,lv;
inline ll max(ll a,ll b){return a>b?a:b;}
inline ll min(ll a,ll b){return a<b?a:b;}
inline void add(int x,int y)
{
    a[++num].nxt=head[x],a[num].to=y,head[x]=num;
}
int find(int n)
{
    if(f[n]!=n) f[n]=find(f[n]);
    return f[n];
}
inline bool cmp(int *y,int a,int b,int k)
{
    int r1=y[a],r2=y[b];
    int r3=a+k>=n?-1:y[a+k];
    int r4=b+k>=n?-1:y[b+k];
    return r1==r2&&r3==r4;
}
inline void makesa()
{
    int *x=t1,*y=t2,m=30;
    for(int i=0;i<m;i++) c[i]=0;
    for(int i=0;i<n;i++) c[x[i]=s[i]-'a']++;
    for(int i=1;i<m;i++) c[i]+=c[i-1];
    for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
    for(int k=1;k<=n;k<<=1)
    {
        int p=0;
        for(int i=n-k;i<n;i++) y[p++]=i;
        for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        for(int i=0;i<m;i++) c[i]=0;
        for(int i=0;i<n;i++) c[x[y[i]]]++;
        for(int i=1;i<m;i++) c[i]+=c[i-1];
        for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1;x[sa[0]]=0;
        for(int i=1;i<n;i++)
          x[sa[i]]=cmp(y,sa[i],sa[i-1],k)?p-1:p++;
        if(p>=n) break;
        m=p;
    }
}
inline void makeheight()
{
    for(int i=0;i<n;i++) rank[sa[i]]=i;
    int k=0;
    for(int i=0;i<n;i++)
      if(rank[i])
      {
        int j=sa[rank[i]-1];
        if(k) k--;
        while(s[i+k]==s[j+k]) k++;
        height[rank[i]]=k;
      }
}
inline void merge(int i,int x,int y)
{
    f[y]=x;
    ans[i]=max(ans[i],Max[sa[x]]*Max[sa[y]]);
    ans[i]=max(ans[i],Min[sa[x]]*Min[sa[y]]);
    Max[sa[x]]=max(Max[sa[x]],Max[sa[y]]);
    Min[sa[x]]=min(Min[sa[x]],Min[sa[y]]);
    tot[i]+=size[sa[x]]*size[sa[y]];
    size[sa[x]]+=size[sa[y]];
}
inline void solve()
{
    for(int i=1;i<n;i++)
      add(height[i],i);
    for(int i=n-1;i>=0;i--)
      for(int j=head[i];j;j=a[j].nxt)
      {
        int x=find(a[j].to),y=find(a[j].to-1);
        merge(i,x,y);
      }
    for(int i=n-2;i>=0;i--)
      tot[i]+=tot[i+1],ans[i]=max(ans[i],ans[i+1]);
    for(int i=0;i<n;i++)
      if(tot[i]) printf("%lld %lld\n",tot[i],ans[i]);
      else printf("0 0\n");
}
int main()
{
    scanf("%d",&n);
    scanf("%s",s);
    for(int i=0;i<n;i++)
      scanf("%d",&lv),f[i]=i,Max[i]=Min[i]=lv,size[i]=1,ans[i]=-1e18;
    makesa();
    makeheight();
    solve();
    return 0;
}