UVA Live-3942 Remember the Word(trie樹入門題)
阿新 • • 發佈:2019-01-26
【解題報告】
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;
}