【UVA1401】Remember the Word Trie+dp
阿新 • • 發佈:2018-12-19
題目大意:給定一個字串和一個字串集合,問從集合中選出若干個串組成給定母串的不同方案數。
題解:有些類似於揹包問題。狀態很好表示,為:\(dp[i]\) 表示母串前 i 個字元的不同方案數,因此,有狀態轉移方程為:\(dp[i]=\Sigma dp[j],s[j+1...i]=s_0,s_0\in set\) ,可以發現若列舉 \(j < i\) 作為決策集合的話,時間複雜度將是 \(O(n^2)\) 的。優化:可以用 Trie 來直接進行匹配,具體操作如下:將每個集合中的串倒序插入 Trie 中,再對每個 i 進行倒序匹配,若匹配成功則嘗試決策轉移,時間複雜度為 \(O(n*depth)\)
程式碼如下
#include <bits/stdc++.h> #define cls(a,b) memset(a,b,sizeof(a)) using namespace std; const int maxn=4e5+10; const int mod=20071027; int trie[maxn][26],tot=1;bool ed[maxn]; int kase,n,dp[maxn]; char s[maxn],s2[101]; void insert(int now,int idx){ if(!idx){ed[now]=1;return;} int ch=s2[idx]-'a'; if(!trie[now][ch])trie[now][ch]=++tot; now=trie[now][ch],insert(now,idx-1); } void go(int now,int idx,int i){ if(ed[now])dp[i]=((long long)dp[i]+dp[idx])%mod; if(!idx)return; else if(!trie[now][s[idx]-'a'])return; else go(trie[now][s[idx]-'a'],idx-1,i); } void read_and_parse(){ scanf("%d",&n); for(int i=1;i<=n;i++)scanf("%s",s2+1),insert(1,strlen(s2+1)); } void solve(){ dp[0]=1; for(int i=1;i<=strlen(s+1);i++)go(1,i,i); printf("Case %d: %d\n",++kase,dp[strlen(s+1)]); } void init(){ cls(trie,0),cls(ed,0),cls(dp,0),tot=1; } int main(){ while(scanf("%s",s+1)!=EOF){ init(); read_and_parse(); solve(); } return 0; }