Remember the Word UVALive - 3942(字典樹trie)
阿新 • • 發佈:2018-11-10
題意:這個題是先給出一個長的字串,然後相應給出若干個單詞,問這個字串能由後面給出的若干個單詞構成的方案數,比如給出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; }