NOIP2023模拟19联测40 诡异键盘

题目大意

有一个键盘，上面有$n+1$个按键，按下按键$1\le i\le n$会打印出字符串$S_i$，按下按键$n+1$会删掉结尾的$K$个字符，如果不足$K$个字符则全部删完，问打印出$S$最少要按多少次。

有$T$组数据。

$1\le T\le 100,10\le n\le 5\times 10^3,1\le \sum K\le 2\times 10^3$

$1\le \sum (\sum |S_i|)\le 10^6,1\le \sum |S|\le 5\times 10^3$

时间限制$2000ms$，空间限制$512MB$。

题解

考虑$DP$，设$f_i$表示打出前$i$个字符需要的最小操作次数。那我们要求的就是打出$S$的第$i$个字符到第$j$个字符需要的最小操作次数。

先考虑如何得到$S$中$[i,j]$这一段。我们选择一个前$j-i+1$个字符与$S$的$[i,j]$中的字符相同的$S_p$，然后将$S_p$打出并删到只剩下$S$的$[i,j]$这部分即可。我们可以建一棵字典树，每次加入一个$S_i$时更新从根节点到达路径上每个点的最小操作次数。为了求这里的最小操作次数，我们还需要求出删除若干个字符所需要的最小操作次数，这个可以用一个类似“同余”的最短路来求出，用$\text{dijkstra}$可以$O(K^2)$解决（连边是$O(k^2)$的，求最短路是$O(K\log K)$的）。

得到了带到每个位置的最小操作次数之后，枚举每个$i$，然后在字典树上按$S$从$i$往后枚举，设枚举到$j$，则用$f_i$来更新$f_j$。$DP$的时间复杂度为$O(|S{|}^2)$。

总时间复杂度为$O(\sum (\sum |S_i|+|S{|}^2+K^2))$。

可以参考代码帮助理解。

code

#include<bits/stdc++.h>
using namespace std;
const int N=5000,M=1000000,K=2000;
int T,n,k,s1,now,bg[N+5],len[N+5],z[K+5],vs[K+5],cm[K+5];
int tot;
char s[N+5],t[M+5],ss[M+5];
long long dis[K+5],to[M+5],f[N+5];
struct node{int x;long long dis;bool operator<(const node ax)const{return dis>ax.dis;}
};
vector<pair<int,int>>g[K+5];
priority_queue<node>q;
struct trie{int a[26];
}w[M+5];
void pt(){int len=strlen(ss+1);for(int i=1;i<=len;i++){t[++now]=ss[i];}
}
void init(){for(int i=0;i<k;i++){z[i]=1e9;vs[i]=cm[i]=0;g[i].clear();}for(int i=1;i<=n;i++){len[i]=bg[i+1]-bg[i];z[len[i]%k]=min(z[len[i]%k],len[i]);vs[len[i]%k]=1;}for(int i=1;i<=n;i++){if(z[len[i]%k]!=len[i]) continue;if(!vs[len[i]%k]) continue;vs[len[i]%k]=0;for(int j=0;j<k;j++){g[(j+len[i])%k].push_back({j,1+(j+len[i])/k});}}for(int i=0;i<k;i++) dis[i]=1e16;dis[0]=dis[k]=0;q.push((node){0,0});while(!q.empty()){int u=q.top().x;q.pop();if(cm[u]) continue;cm[u]=1;for(auto p:g[u]){int v=p.first,w=p.second;if(dis[u]+w<dis[v]){dis[v]=dis[u]+w;q.push((node){v,dis[v]});}}}
}
void pt(int tw){int q,vq=1;for(int i=1;i<=len[tw];i++){q=t[bg[tw]+i-1]-'a';if(!w[vq].a[q]){w[vq].a[q]=++tot;to[tot]=1e16;}vq=w[vq].a[q];to[vq]=min(to[vq],1ll+(len[tw]-i)/k+dis[(len[tw]-i)%k]);}
}
void solve(int tw){int q,vq=1;for(int i=tw+1;i<=s1;i++){q=s[i]-'a';if(!w[vq].a[q]) return;vq=w[vq].a[q];f[i]=min(f[i],f[tw]+to[vq]);}
}
int main()
{
//	freopen("keyboard.in","r",stdin);
//	freopen("keyboard.out","w",stdout);scanf("%d",&T);while(T--){scanf("%d%d",&n,&k);now=0;for(int i=1;i<=n;i++){scanf("%s",ss+1);bg[i]=now+1;pt();}bg[n+1]=now+1;scanf("%s",s+1);s1=strlen(s+1);init();tot=1;for(int i=1;i<=n;i++) pt(i);for(int i=0;i<=s1;i++) f[i]=1e16;f[0]=0;for(int i=0;i<s1;i++) solve(i);if(f[s1]>=1e16) printf("-1\n");else printf("%lld\n",f[s1]);for(int i=1;i<=tot;i++){for(int j=0;j<26;j++) w[i].a[j]=0;}}return 0;
}