品酒大会 BZOJ 4199

品酒大会

【问题描述】

【输入格式】

【输出格式】

【样例输入】

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

【数据范围】

---

题解：

根据题意可得"r相似”也是“r - 1相似”

那么我们只要求出了所有最大为 r 相似的对数，就可以利用后缀和求出所有r相似的个数

考虑一瓶酒与另一瓶酒如果是 r 相似的，那么与其中一瓶酒 k (k > r) 相似的酒与另一瓶酒最大也为 r 相似

所以用后缀数组求出 height 数组

然后按 height 从大到小排序

每次按顺序找出两个 height 相似的点的祖先

height 相似的对数累加上两个祖先块内的点数乘积

height 相似的最大值为两个块的最小值乘积和最大值乘积的较大值

用并查集合并，处理点的个数、最大值和最小值（美味度有负数）

最后跑一遍后缀和

 #include<algorithm>
 #include<iostream>
 #include<cstring>
 #include<cstdlib>
 #include<cstdio>
 #include<cmath>
 using namespace std;
 inline void Scan(int &x)
 {
     char c;
     int o = ;
     while((c = getchar()) < '' || c > '')
         if(c == '-') o = -;
     x = c - '';
     while((c = getchar()) >= '' && c <= '') x = x *  + c - '';
     x *= o;
 }
 const int me = ;
 int n;
 int w[me];
 int x[me];
 int sa[me], he[me];
 int val[me], fat[me], nex[me];
 int rank[me];
 long long ans_si[me], ans_mx[me];
 char s[me];
 struct Union
 {
     long long si, mx, mi;
 };
 Union un[me];
 inline void Sa()
 {
     int m = ;
     for(int i = ; i <= n; ++i) ++w[x[i] = s[i] - 'a' + ];
     for(int i = ; i <= m; ++i) w[i] += w[i - ];
     for(int i = n; i >= ; --i) sa[w[x[i]]--] = i;
     for(int k = ; k <= n; k <<= )
     {
         int p = ;
         for(int i = n; i >= n - k + ; --i) rank[++p] = i;
         for(int i = ; i <= n; ++i)
             if(sa[i] > k)
                 rank[++p] = sa[i] - k;
         for(int i = ; i <= m; ++i) w[i] = ;
         for(int i = ; i <= n; ++i) ++w[x[i]];
         for(int i = ; i <= m; ++i) w[i] += w[i - ];
         for(int i = n; i >= ; --i) sa[w[x[rank[i]]]--] = rank[i];
         m = ;
         for(int i = ; i <= n; ++i)
         {
             int u = sa[i], v = sa[i - ];
             if(x[u] != x[v] || x[u + k] != x[v + k]) rank[u] = ++m;
             else rank[u] = m;
         }
         if(n == m) break;
         for(int i = ; i <= n; ++i) swap(x[i], rank[i]);
     }
     int tot = ;
     int i, j;
     for(i = ; i <= n; i ++)
     {
         if (tot) tot --;
         j = sa[rank[i] - ];
         while (s[j + tot] == s[i + tot]) tot ++;
         he[rank[i]] = tot;
     }
 }
 inline bool rule(const int &x, const int &y)
 {
     return he[x] > he[y];
 }
 inline int Find(const int &x)
 {
     return (x != fat[x]) ? fat[x] = Find(fat[x]) : x;
 }
 inline void Un(const int &x, const int &y)
 {
     un[x].si += un[y].si;
     un[x].mx = max(un[x].mx, un[y].mx);
     un[x].mi = min(un[x].mi, un[y].mi);
     fat[y] = x;
 }
 int main()
 {
     Scan(n);
     scanf("%s", s + );
     for(int i = ; i <= n; ++i)
     {
         Scan(val[i]);
         nex[i] = i + ;
         fat[i] = i;
     }
     Sa();
     for(int i = ; i <= n; ++i)
         ans_mx[i] = -;
     for(int i = ; i <= n; ++i)
         un[i] = (Union) {, val[sa[i]], val[sa[i]]};
     sort(nex + , nex + n, rule);
     for(int i = ; i < n; ++i)
     {
         int x = Find(nex[i] - ), y = Find(nex[i]);
         int z = he[nex[i]];
         ans_si[z] += un[x].si * un[y].si;
         ans_mx[z] = max(ans_mx[z], max(un[x].mi * un[y].mi, un[x].mx * un[y].mx));
         Un(x, y);
     }
     for(int i = n - ; i >= ; --i)
     {
         ans_si[i] += ans_si[i + ];
         ans_mx[i] = max(ans_mx[i], ans_mx[i + ]);
     }
     for(int i = ; i < n; ++i)
         printf("%lld %lld\n", ans_si[i], ans_si[i] ? ans_mx[i] : );
 }