题目描述
一年一度的“幻影阁夏日品酒大会”隆重开幕了。大会包含品尝和趣味挑战 两个环节,分别向优胜者颁发“首席品酒家”和“首席猎手”两个奖项,吸引了众多品酒师参加。
在大会的晚餐上,调酒师 Rainbow 调制了 n 杯鸡尾酒。这 n 杯鸡尾酒排成一行,其中第 n 杯酒 (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 。
输入输出样例
输入样例#1:
1 2 3 |
10 ponoiiipoi 2 1 4 7 4 8 3 6 4 7 |
输出样例#1:
1 2 3 4 5 6 7 8 9 10 |
45 56 10 56 3 32 0 0 0 0 0 0 0 0 0 0 0 0 0 0 |
输入样例#2:
1 2 3 |
12 abaabaabaaba 1 -2 3 -4 5 -6 7 -8 9 -10 11 -12 |
输出样例#2:
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 |
说明
【样例说明 1】
用二元组 (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 。
【时限1s,内存512M】
Solution
从大到小扫height数组,用并查集维护。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 |
#include<iostream> #include<cstdio> #include<algorithm> #include<cmath> #include<cstring> #define ll long long using namespace std; const int N=300100; int n,A[N],f[N],pos[N]; ll Max[N],Max1[N],Min[N],Min1[N],Num[N],ans[N],Max_ans[N]; struct SA { char s[N]; int cnta[N],sa[N],rank[N*2],a[N],b[N],cntb[N],tsa[N],height[N]; void Get_SA() { for (int i=1;i<=256;i++) cnta[i]=0; for (int i=1;i<=n;i++) cnta[(int)s[i]]++; for (int i=1;i<=256;i++) cnta[i]+=cnta[i-1]; for (int i=n;i>=1;i--) sa[cnta[(int)s[i]]--]=i; rank[sa[1]]=1; for (int i=2;i<=n;i++) rank[sa[i]]=rank[sa[i-1]]+(s[sa[i]]!=s[sa[i-1]]); for (int i=1;rank[sa[n]]!=n;i<<=1) { for (int j=1;j<=n;j++) a[j]=rank[j],b[j]=rank[j+i]; for (int j=0;j<=n;j++) cnta[j]=0,cntb[j]=0; for (int j=1;j<=n;j++) cnta[a[j]]++,cntb[b[j]]++; for (int j=1;j<=n;j++) cnta[j]+=cnta[j-1],cntb[j]+=cntb[j-1]; for (int j=n;j>=1;j--) tsa[cntb[b[j]]--]=j; for (int j=n;j>=1;j--) sa[cnta[a[tsa[j]]]--]=tsa[j]; rank[sa[1]]=1; for (int j=2;j<=n;j++) rank[sa[j]]=rank[sa[j-1]]+(a[sa[j]]!=a[sa[j-1]]||b[sa[j]]!=b[sa[j-1]]); } } void Get_Height() { int len=0; for (int i=1;i<=n;i++) { if (len) len--; while (s[i+len]==s[sa[rank[i]-1]+len]) len++; height[rank[i]]=len; } } }S; bool cmp(int x,int y) { return S.height[x]>S.height[y];} int get(int x) { if (f[x]==x) return x;else f[x]=get(f[x]); return f[x]; } int main() { scanf("%d",&n); scanf("%s",S.s+1); for (int i=1;i<=n;i++) scanf("%d",&A[i]); S.Get_SA(); S.Get_Height(); for (int i=1;i<=n;i++) { f[i]=i; Max[i]=A[S.sa[i]];Max1[i]=-1LL<<30; Min[i]=A[S.sa[i]];Min1[i]=1LL<<30; Num[i]=1; pos[i]=i+1; } for (int i=0;i<=n;i++) Max_ans[i]=-1LL<<60; sort(pos+1,pos+n,cmp); for (int i=1;i<=n-1;i++) { int x=get(f[pos[i]]),y=get(f[pos[i]-1]); if (Max[y]>Max[x]) Max1[x]=Max[x],Max[x]=Max[y];else if (Max[y]>Max1[x]) Max1[x]=Max[y]; if (Max1[y]>Max1[x]) Max1[x]=Max1[y]; if (Min[y]<Min[x]) Min1[x]=Min[x],Min[x]=Min[y];else if (Min[y]<Min1[x]) Min1[x]=Min[y]; if (Min1[y]<Min1[x]) Min1[x]=Min1[y]; ans[S.height[pos[i]]]+=Num[y]*Num[x]; Num[x]+=Num[y]; if (Max1[x]!=-1LL<<30) Max_ans[S.height[pos[i]]]=max(Max_ans[S.height[pos[i]]],Max[x]*Max1[x]); if (Min1[x]!=1LL<<30) Max_ans[S.height[pos[i]]]=max(Max_ans[S.height[pos[i]]],Min[x]*Min1[x]); Max_ans[S.height[pos[i]]]=max(Max_ans[S.height[pos[i]]],Max[x]*Max1[x]); f[y]=x; } for (int i=n-1;i>=0;i--) ans[i]+=ans[i+1],Max_ans[i]=max(Max_ans[i],Max_ans[i+1]); for (int i=0;i<=n-1;i++) printf("%lld %lld\n",ans[i],(Max_ans[i]!=-1LL<<60)?Max_ans[i]:0LL); return 0; } |