Display Substring

#include<bits/stdc++.h> using namespace std; typedef long long ll; // sa[i]: 排名是i位的是第幾個后綴 // rk[i]: 第i個后綴的排名是多少 // height[i]: sa[i]與sa[i-1] const int N=100010; char s[N]; int rk[N],sa[N],cnt[N],height[N]; int x[N],y[N]; int n,m; void rsort()// x[i] 第一關鍵字 y[i] 第二關鍵字 基數排序 {for(int i=1;i<=m;i++) cnt[i]=0;for(int i=1;i<=n;i++) cnt[x[i]]++;for(int i=1;i<=m;i++) cnt[i]+=cnt[i-1];for(int i=n;i;i--) sa[cnt[x[y[i]]]--]=y[i]; } void SA() {n=strlen(s+1);m=300;for(int i=1;i<=n;i++) x[i]=s[i],y[i]=i;rsort();for(int k=1;k<=n;k<<=1){int p=0;for(int i=n-k+1;i<=n;i++) y[++p]=i;// 第二關鍵字為空字符排在最前面for(int i=1;i<=n;i++) if(sa[i]>k) y[++p]=sa[i]-k;rsort();swap(x,y);x[sa[1]]=1,p=1;for(int i=2;i<=n;i++)x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p:++p);if(p==n) break;m=p;}for(int i=1;i<=n;i++) rk[sa[i]]=i;// 求heightfor(int i=1,j=0;i<=n;i++){if(j) --j;while(s[i+j]==s[sa[rk[i]-1]+j]) j++;height[rk[i]]=j;} } int v[200];ll K; int sum[N]; bool check(int x) {ll cnt=0;for(int i=1;i<=n;i++){// [sa[i],r)的權值小于x// sa[i]~n sum[sa[i]-1]+xint r=upper_bound(sum+sa[i],sum+n+1,sum[sa[i]-1]+x)-sum;cnt+=max(0,r-sa[i]-height[i]);}return (cnt>=K); } int main() {int Tc;scanf("%d",&Tc);while(Tc--){scanf("%d%lld%s",&n,&K,s+1);for(int i=0;i<26;i++) scanf("%d",&v[i+'a']);SA();ll tot=0;for(int i=1;i<=n;i++) tot+=n-sa[i]+1-height[i];if(tot<K) {puts("-1");continue;}int l=100;for(int i=1;i<=n;i++) {l=min(l,v[s[i]]);sum[i]=sum[i-1]+v[s[i]];}int r=sum[n];while(l<r){int mid=l+r>>1;if(check(mid)) r=mid;else l=mid+1;}printf("%d\n",l);}return 0; }

總結

以上是生活随笔為你收集整理的2021“MINIEYE杯”中国大学生算法设计超级联赛（4）Display Substring（后缀数组+二分）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。