1. 程式人生 > >BZOJ_P4199 [NOI2015] 品酒大會(字尾陣列+並查集)

BZOJ_P4199 [NOI2015] 品酒大會(字尾陣列+並查集)

一年一度的“幻影閣夏日品酒大會”隆重開幕了。大會包含品嚐和趣味挑戰兩個環節,分別向優勝者頒發“首席品酒家”和“首席獵手”兩個獎項,吸引了眾多品酒師參加。

在大會的晚餐上,調酒師 Rainbow 調製了 nn 杯雞尾酒。這 nn 杯雞尾酒排成一行,其中第 ii 杯酒 (1≤i≤n1≤i≤n) 被貼上了一個標籤 sisi,每個標籤都是 2626 個小寫英文字母之一。設 Str(l,r)Str(l,r) 表示第 ll 杯酒到第 rr 杯酒的 r−l+1r−l+1 個標籤順次連線構成的字串。若 Str(p,po)=Str(q,qo)Str(p,po)=Str(q,qo),其中 1≤p≤po≤n1≤p≤po≤n,1≤q≤qo≤n1≤q≤qo≤n,p≠qp≠q,po−p+1=qo−q+1=rpo−p+1=qo−q+1=r,則稱第 pp 杯酒與第 qq 杯酒是“rr相似” 的。當然兩杯“rr相似” (r>1r>1)的酒同時也是“11 相似”、“22 相似”、……、“(r−1)(r−1) 相似”的。特別地,對於任意的 1≤p,q≤n1≤p,q≤n,p≠qp≠q,第 pp 杯酒和第 qq 杯酒都是“00相似”的。

在品嚐環節上,品酒師 Freda 輕鬆地評定了每一杯酒的美味度,憑藉其專業的水準和經驗成功奪取了“首席品酒家”的稱號,其中第 ii 杯酒 (1≤i≤n1≤i≤n) 的美味度為 aiai。現在 Rainbow 公佈了挑戰環節的問題:本次大會調製的雞尾酒有一個特點,如果把第 pp 杯酒與第 qq 杯酒調兌在一起,將得到一杯美味度為 apaqapaq 的酒。現在請各位品酒師分別對於 r=0,1,2,…,n−1r=0,1,2,…,n−1,統計出有多少種方法可以選出 22 杯“rr相似”的酒,並回答選擇 22 杯“rr相似”的酒調兌可以得到的美味度的最大值。

輸入格式
輸入檔案的第 11 行包含 11 個正整數 nn,表示雞尾酒的杯數。

第 22 行包含一個長度為 nn 的字串 SS,其中第 ii 個字元表示第 ii 杯酒的標籤。

第 33 行包含 nn 個整數,相鄰整數之間用單個空格隔開,其中第 ii 個整數表示第 ii 杯酒的美味度 aiai。

輸出格式
輸出檔案包括 nn 行。第 ii 行輸出 22 個整數,中間用單個空格隔開。第 11 個整數表示選出兩杯“(i−1)(i−1)相似”的酒的方案數,第 22 個整數表示選出兩杯“(i−1)(i−1)相似”的酒調兌可以得到的最大美味度。若不存在兩杯“(i−1)(i−1)相似”的酒,這兩個數均為 00。

樣例一
input

10
ponoiiipoi
2 1 4 7 4 8 3 6 4 7

output

45 56
10 56
3 32
0 0
0 0
0 0
0 0
0 0
0 0
0 0

explanation

用二元組 (p,q)(p,q) 表示第 pp 杯酒與第 qq 杯酒。

00 相似:所有 4545 對二元組都是 00 相似的,美味度最大的是 8×7=568×7=56。

11 相似:(1,8)(1,8) (2,4)(2,4) (2,9)(2,9) (4,9)(4,9) (5,6)(5,6) (5,7)(5,7) (5,10)(5,10) (6,7)(6,7) (6,10)(6,10) (7,10)(7,10),最大的 8×7=568×7=56。

22 相似:(1,8)(1,8) (4,9)(4,9) (5,6)(5,6),最大的 4×8=324×8=32。

沒有 3,4,5,…,93,4,5,…,9 相似的兩杯酒,故均輸出 00。

樣例二
input

12
abaabaabaaba
1 -2 3 -4 5 -6 7 -8 9 -10 11 -12

output

66 120
34 120
15 55
12 40
9 27
7 16
5 7
3 -4
2 -4
1 -4
0 0
0 0

樣例三
見樣例資料下載。

限制與約定
測試點編號 nn 的規模 aiai 的規模 備註
1 n=100n=100 ∣ai∣≤10000∣ai∣≤10000
2 n=200n=200
3 n=500n=500
4 n=750n=750
5 n=1000n=1000 ∣ai∣≤1000000000∣ai∣≤1000000000
6
7 n=2000n=2000
8
9 n=99991n=99991 ∣ai∣≤1000000000∣ai∣≤1000000000 不存在“1010相似”的酒
10
11 n=100000n=100000 ∣ai∣≤1000000∣ai∣≤1000000 所有 aiai 的值都相等
12 n=200000n=200000
13 n=300000n=300000
14
15 n=100000n=100000 ∣ai∣≤1000000000∣ai∣≤1000000000
16
17 n=200000n=200000
18
19 n=300000n=300000
20
時間限制:1s1s
空間限制:512MB

Sol:
字尾陣列求height[]
並查集維護最大值和次大值,最小值和次小值(有負數),統計方案數的時候就將兩個並查集的size相乘就可以了(2組合)。

#include<cstdio>
#include<climits>
#include<cstring>
#include<vector>
#include<algorithm>
#include<iostream>
using namespace std;
#define N 300005
typedef long long ll;
inline int in(int x=0,char ch=getchar(),int v=1){
    while(ch!='-'&&(ch>'9'||ch<'0')) ch=getchar();if(ch=='-') v=-1,ch=getchar();
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();return x*v;}
int t1[N],t2[N],c[N],sa[N],rk[N],ht[N];
int n,k,l1,l2;char s[N];
void DA(int m,int n){
    int *x=t1,*y=t2;
    for(int i=0;i<m;i++) c[i]=0;
    for(int i=0;i<n;i++) c[x[i]=s[i]]++;
    for(int i=1;i<m;i++) c[i]+=c[i-1];
    for(int i=n-1;~i;i--) sa[--c[x[i]]]=i;
    for(int k=1,p=0;k<=n;k<<=1,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;--i) sa[--c[x[y[i]]]]=y[i];
        swap(x,y);m=1,x[sa[0]]=0;
        for(int i=1;i<n;i++) x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&
            (y[sa[i]+k]==y[sa[i-1]+k])?m-1:m++;
        if(m>=n) break;
    }
}
void CalcHeight(int n){
    for(int i=0;i<=n;i++) rk[sa[i]]=i;
    for(int i=0,j,k=0;i<n;ht[rk[i++]]=k){
        j=sa[rk[i]-1];if(k) k--;
        while(s[i+k]==s[j+k]) k++;
    }
    ht[0]=0;
}
vector<int> g[N];
int f[N];long long sz[N];
long long max1[N],max2[N],min1[N],min2[N],a1[N],a[N];
long long ans[N][2];
inline long long max(long long a,long long b){return b>a?b:a;}
inline long long min(long long a,long long b){return b<a?b:a;}
inline int find(int x){return x==f[x]?x:f[x]=find(f[x]);}
inline void merge(int a,int b,int i){
    if(!a) return;
    int f1=find(a),f2=find(b);
    if(max1[f1]>max1[f2]) max2[f1]=max(max1[f2],max2[f1]);
    else max2[f1]=max(max2[f2],max1[f1]),max1[f1]=max1[f2];
    if(min1[f1]<min1[f2]) min2[f1]=min(min2[f1],min1[f2]);
    else min2[f1]=min(min1[f1],min2[f2]),min1[f1]=min1[f2];
    f[f2]=f1;ans[i][0]+=sz[f1]*sz[f2],sz[f1]+=sz[f2];;
    ans[i][1]=max(ans[i][1],max(max1[f1]*max2[f1],min1[f1]*min2[f1]));
}
char str[65];
void print(long long x){
    int l=0;if(!x){putchar('0');return;}
    if(x<0) putchar('-'),x=-x;
    while(x) str[++l]=x%10+'0',x/=10;
    while(l) putchar(str[l--]);
}
int main(){
    n=in();
    scanf("%s",s);n=strlen(s);
    DA(256,n+1);CalcHeight(n);
    for(int i=1;i<=n;i++) a1[i]=in(),ans[i][1]=LONG_LONG_MIN,max2[i]=LONG_LONG_MIN,min2[i]=LONG_LONG_MAX;
    for(int i=1;i<=n;i++) a[i]=a1[sa[i]+1],sz[i]=1,f[i]=i,g[ht[i]].push_back(i),max1[i]=a[i],min1[i]=a[i];
    for(int i=n-1;~i;i--){
        if(g[i].empty()) continue;
        ans[i][0]=ans[i+1][0],ans[i][1]=ans[i+1][1];
        for(int j=0,lim=g[i].size();j<lim;j++) merge(g[i][j]-1,g[i][j],i);
    }
    for(int i=0;i<n;i++) print(ans[i][0]),putchar(' '),print(ans[i][0]?ans[i][1]:0),putchar('\n');
    return 0;
}