1. 程式人生 > >UVA Live-3942 Remember the Word(trie樹入門題)

UVA Live-3942 Remember the Word(trie樹入門題)

【解題報告】
trie樹入門題目。我們知道給定字串和給定的S個單詞,求把這個字串分解成若干個單詞的方案數。
不難想到遞推求解。
dp[i]表示text[i..LEN]這個字串可以分解成若干個單詞的方案數

    dp[i]=sum( dp[i+len(x)] )(x表示一個可以作為i~LEN的字首的單詞)

初始化使dp[len]=1,表示如果字串text[i..LEN]可以表示為一個單詞時,方案數為1.

如何查詢單詞字首?可以預處理單詞,構造一顆trie樹,儲存的結點資訊為單詞長度,然後列舉字串的數位i,遍歷trie樹查詢即可。

【參考程式碼】

#include<iostream>
#include<cstdio> #include<cstring> #include<vector> #include<algorithm> using namespace std; const int mod=20071027; const int maxnode=4*1e5+10; char text[300000+10]; int n; int dp[300000+10]; vector<int>P; struct TRIE{ int ch[maxnode][26]; int val[maxnode]; int sz; void
clear(){ sz=1; memset( ch,0,sizeof(ch) ); } int idx( char c){ return c-'a'; } void insert( char* s, int v ) { int L=strlen(s),u=0; for( int i=0; i<L; i++ ) { int c=idx(s[i]); if( !ch[u][c] ) { val[sz]=0; ch[u][c]=sz++; } u=ch[u][c]; } val[u]=v; } void
find_pre( const char* S ) { int L=strlen(S),u=0; for( int i=0; i<L; i++ ) { if(S[i]=='\0')break; int c=idx(S[i]); if(!ch[u][c])break; if(val[ch[u][c]] )P.push_back(val[ch[u][c]]); u=ch[u][c]; } } }trie; int main() { int kase=0; while(~scanf("%s",text)) { scanf("%d",&n); trie.clear(); memset( dp,0,sizeof(dp) ); for( int i=1; i<=n;i++ ) { char word[100+10]; scanf("%s",word); int v=strlen(word); trie.insert( word, v ); } int L=strlen(text); dp[L]=1; for( int i=L-1; i>=0; i-- ) { P.clear(); trie.find_pre( text+i ); for( int j=0; j<P.size(); j++ ) { dp[i]=(dp[i]+dp[i+P[j]])%mod; } } printf("Case %d: %d\n",++kase,dp[0]); } return 0; }