Remember the Word UVALive - 3942(字典树trie)
程序员文章站
2022-05-18 20:26:33
...
题意:这个题是先给出一个长的字符串,然后相应给出若干个单词,问这个字符串能由后面给出的若干个单词构成的方案数,比如给出abcd,给出的若干个单词是a,b,ab,cd,这样就有两种构成方法了
题解:最最重要的是得要想出递推得出答案,用d[i]表示从给出的字符串中的第i位开始的后缀所能构成的方案数,然后对于每一位我们直接d[i]=d[i]+d[i+len[p[j]]即可从后往前推出答案了,最后就是如果快速得出前缀呢?使用字典树
附上代码:
#include<bits/stdc++.h>
using namespace std;
const int maxnode=4e3*2e2+10;
const int sigma_size=26;
const int maxl=3e5+10;
const int maxw=4e3+10;
const int maxwl=1e2+10;
const int MOD=20071027;
int d[maxl],len[maxw],S;
char text[maxl],word[maxwl];
struct Trie{
int ch[maxnode][sigma_size];
int val[maxnode];
int sz;
void clear(){
sz=1;
memset(ch[0],0,sizeof(ch[0]));
}
int idx(char c){
return c-'a';
}
void insert(char *s,int v){
int u=0,n=strlen(s);
for(int i=0;i<n;i++){
int c=idx(s[i]);
if(!ch[u][c]){
memset(ch[sz],0,sizeof(ch[sz]));
val[sz]=0;
ch[u][c]=sz++;
}
u=ch[u][c];
}
val[u]=v;
}
void find_prefixes(char *s,int len,vector<int>&ans){
int u=0;
for(int i=0;i<len;i++){
if(s[i]=='\0'){
break;
}
int c=idx(s[i]);
if(!ch[u][c]){
break;
}
u=ch[u][c];
if(val[u]!=0){
ans.push_back(val[u]);
}
}
}
};
Trie trie;
int main()
{
int kase=1;
while(scanf("%s%d",text,&S)!=EOF){
trie.clear();
for(int i=1;i<=S;i++){
scanf("%s",word);
len[i]=strlen(word);
trie.insert(word,i);
}
memset(d,0,sizeof(d));
int L=strlen(text);
d[L]=1;
for(int i=L-1;i>=0;i--){
vector<int>p;
trie.find_prefixes(text+i,L-i,p);
for(int j=0;j<p.size();j++){
d[i]=(d[i]+d[i+len[p[j]]])%MOD;
}
}
printf("Case %d: %d\n",kase++,d[0]);
}
return 0;
}
上一篇: Recursion