本題練習前綴樹(Trie),又稱字典樹。
Trie結構體模板:
?
struct Trie { int sz; int ch[maxnode][sigma_size]; int val[maxnode]; void init() { 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; int 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; } int find(char *s) { int u = 0; int n = strlen(s); for(int i=0;i<n;i++) { int c = idx(s[i]); if(!ch[u][c]) return 0; u = ch[u][c]; } return val[u]; } };
對于本題而言,DP+Trie可以解決。
?
其中,d[i]代表以s[i-1]結尾的分解方案。
?
#include <stdio.h> #include <string.h> #include <math.h> #include <stdlib.h> #include <time.h> #include <vector> #include <set> #include <map> #include <algorithm> using namespace std; #define maxnode 4010*100 #define sigma_size 26 #define MOD 20071027 struct Trie { int sz; int ch[maxnode][sigma_size]; int val[maxnode]; void init() { 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; int 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; } int find(char *s) { int u = 0; int n = strlen(s); for(int i=0;i<n;i++) { int c = idx(s[i]); if(!ch[u][c]) return 0; u = ch[u][c]; } return val[u]; } }; Trie trie; char s[300005]; int d[300005];//d[i]代表以s[i-1]結尾的分解方案 int main() { #ifndef ONLINE_JUDGE freopen("in.txt", "r", stdin); #endif int n; char temp[105]; int cas = 0; while(scanf(" %s",s)!=EOF) { cas++; memset(d,0,sizeof(d)); d[0] = 1; trie.init(); scanf(" %d",&n); for(int i=0;i<n;i++) { scanf(" %s",temp); trie.insert(temp,1); } int len = strlen(s); for(int i=0;i<len;i++) { int u = 0; for(int j=i+1;j<=len;j++) { int c = trie.idx(s[j-1]); if(!trie.ch[u][c]) break; if(trie.val[trie.ch[u][c]]) { d[j] += d[i]; if(d[j]>=MOD) d[j] -= MOD; } u = trie.ch[u][c]; } } printf("Case %d: %d\n",cas,d[len]); } }
?
?
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索:z360901061

微信掃一掃加我為好友
QQ號聯系: 360901061
您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點擊下面給點支持吧,站長非常感激您!手機微信長按不能支付解決辦法:請將微信支付二維碼保存到相冊,切換到微信,然后點擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對您有幫助就好】元
