品酒大會
題目描述
一年一度的“幻影閣夏日品酒大會”隆重開幕了。大會包含品嘗和趣味挑戰 兩個環節,分別向優勝者頒發“首席品酒家”和“首席獵手”兩個獎項,吸引了眾多品酒師參加。
在大會的晚餐上,調酒師 Rainbow 調制了 $n$ 杯雞尾酒。這 $n$ 杯雞尾酒排成一行,其中第 $n$ 杯酒 ($1 ≤ i ≤ n$) 被貼上了一個標簽 $s_i$ ,每個標簽都是 $26$ 個小寫 英文字母之一。設 $str(l, r)$ 表示第 $l$ 杯酒到第 $r$ 杯酒的 $r - l + 1$ 個標簽順次連接構成的字符串。若 $str(p, p_0) = str(q, q_0)$,其中 $1 ≤ p ≤ p_0 ≤ n$, $1 ≤ q ≤ q_0 ≤ n$, $p ≠ q$,$p_0-p+1 = q_0 - 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$) 的 美味度為 $a_i$ 。現在 Rainbow 公布了挑戰環節的問題:本次大會調制的雞尾酒有一個特點,如果把第 $p$ 杯酒與第 $q$ 杯酒調兌在一起,將得到一杯美味度為 $a_p\times a_q$ 的 酒。現在請各位品酒師分別對於 $r = 0,1,2,⋯,n-1$ ,統計出有多少種方法可以 選出 $2$ 杯“ $r$ 相似”的酒,並回答選擇 $2$ 杯“$r$ 相似”的酒調兌可以得到的美味度的最大值。
輸入輸出格式
輸入格式:第 $1$ 行包含 $1$ 個正整數 $n$ ,表示雞尾酒的杯數。
第 $2$ 行包含一個長度為 $n$ 的字符串 $S$,其中第 $i$ 個字符表示第 $i$ 杯酒的標簽。
第 $3$ 行包含 $n$ 個整數,相鄰整數之間用單個空格隔開,其中第 $i$ 個整數表示第 $i$ 杯酒的美味度 $a_i$ 。
輸出格式:包括 $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
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
說明
【樣例說明 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】
分析
建出后綴樹以后,因為任意兩個后綴的LCP都是他們的LCA,所以在樹上dp,維護大小和最值就可以了。
時間復雜度\(O(n)\)
co int N=6e5;
int last=1,tot=1;
int ch[N][26],fa[N],len[N],pos[N],ref[N];
void extend(int c,int po){
int p=last,cur=last=++tot;
len[cur]=len[p]+1,pos[cur]=po,ref[po]=cur;
for(;p&&!ch[p][c];p=fa[p]) ch[p][c]=cur;
if(!p) fa[cur]=1;
else{
int q=ch[p][c];
if(len[q]==len[p]+1) fa[cur]=q;
else{
int clone=++tot;
memcpy(ch[clone],ch[q],sizeof ch[q]);
fa[clone]=fa[q],len[clone]=len[p]+1,pos[clone]=pos[cur];
fa[cur]=fa[q]=clone;
for(;ch[p][c]==q;p=fa[p]) ch[p][c]=clone;
}
}
}
int son[N][26],siz[N],max[N][2],min[N][2];
ll num[N],ans[N];
void dfs(int p){
for(int c=0,q;c<26;++c)if(q=son[p][c]){
dfs(q);
num[len[p]]+=(ll)siz[p]*siz[q],siz[p]+=siz[q];
if(max[q][0]>max[p][1]){
max[p][1]=max[q][0];
if(max[p][1]>max[p][0]) std::swap(max[p][1],max[p][0]);
}
if(max[q][1]>max[p][1]){
max[p][1]=max[q][1];
if(max[p][1]>max[p][0]) std::swap(max[p][1],max[p][0]);
}
if(min[q][0]<min[p][1]){
min[p][1]=min[q][0];
if(min[p][1]<min[p][0]) std::swap(min[p][1],min[p][0]);
}
if(min[q][1]<min[p][1]){
min[p][1]=min[q][1];
if(min[p][1]<min[p][0]) std::swap(min[p][1],min[p][0]);
}
}
if(siz[p]>=2) ans[len[p]]=std::max(ans[len[p]],std::max((ll)max[p][0]*max[p][1],(ll)min[p][0]*min[p][1]));
}
char s[N];
int n;
int main(){
read(n),scanf("%s",s+1);
for(int i=n;i;--i) s[i]-='a',extend(s[i],i);
for(int i=1;i<=tot;++i) max[i][0]=max[i][1]=-1e9,min[i][0]=min[i][1]=1e9;
for(int i=1;i<=n;++i){
int p=ref[i];
siz[p]=1,max[p][0]=read(min[p][0]);
}
for(int i=0;i<n;++i) ans[i]=-1e18; // edit 2: range of INFINITY
for(int i=1;i<=tot;++i) son[fa[i]][s[pos[i]+len[fa[i]]]]=i;
dfs(1);
for(int i=n-2;i>=0;--i) num[i]+=num[i+1],ans[i]=std::max(ans[i],ans[i+1]);
for(int i=0;i<n;++i) printf("%lld %lld\n",num[i],ans[i]==-1e18?0:ans[i]);
return 0;
}
字符串
有兩個長度為 \(n\) 的由小寫字母組成的字符串 \(a,b\),取出他們所有長為 \(k\) 的子串(各有 \(n-k+1\) 個),這些子串分別組成集合 \(A,B\)。現在要修改 \(A\) 中的串,使得 \(A\) 和 \(B\) 完全相同。可以任意次選擇修改 \(A\) 中一個串的一段后綴,花費為這段后綴的長度。總花費為每次修改花費之和,求總花費的最小值。
對於所有數據,\(1\le k\le n\le 1.5\times 10^5\)。
題解
將兩個串的反串放后綴自動機上,那么長度為\(k\)的串在parent樹上是一些沒有祖先關系的點。
然后要做個匹配。串\(x,y\)在節點\(f\)處匹配的代價是\(k-\text{len}_f\)。
顯然越下面匹配越好,貪心地樹形DP即可。
時間復雜度\(O(n)\)。
CO int N=6e5;
int last=1,tot=1;
array<int,26> ch[N];
int fa[N],len[N],siz[N];
void extend(int c){
int x=last;
if(ch[x][c]){
int y=ch[x][c];
if(len[y]==len[x]+1) {last=y; return;}
int clone=last=++tot;
ch[clone]=ch[y],fa[clone]=fa[y],len[clone]=len[x]+1;
fa[y]=clone;
for(;ch[x][c]==y;x=fa[x]) ch[x][c]=clone;
return;
}
int cur=last=++tot;
len[cur]=len[x]+1;
for(;x and !ch[x][c];x=fa[x]) ch[x][c]=cur;
if(!x) {fa[cur]=1; return;}
int y=ch[x][c];
if(len[y]==len[x]+1) {fa[cur]=y; return;}
int clone=++tot;
ch[clone]=ch[y],fa[clone]=fa[y],len[clone]=len[x]+1;
fa[cur]=fa[y]=clone;
for(;ch[x][c]==y;x=fa[x]) ch[x][c]=clone;
}
char a[N],b[N];
vector<int> to[N];
int main(){
int n=read<int>(),K=read<int>();
scanf("%s%s",a+1,b+1);
for(int i=n;i>=1;--i) extend(a[i]-'a'),siz[last]+=i<=n-K+1;
last=1;
for(int i=n;i>=1;--i) extend(b[i]-'a'),siz[last]-=i<=n-K+1;
for(int i=2;i<=tot;++i) to[fa[i]].push_back(i);
int64 ans=0;
function<void(int)> dfs=[&](int x)->void{
for(int y:to[x]){
dfs(y);
if((int64)siz[x]*siz[y]<0 and len[x]<K)
ans+=(int64)min(abs(siz[x]),abs(siz[y]))*(K-len[x]);
siz[x]+=siz[y];
}
};
dfs(1);
printf("%lld\n",ans);
return 0;
}