暑假- Trie树-(E - Remember the Word)

来源:互联网 发布:制作地图软件 编辑:程序博客网 时间:2024/05/23 01:53
/*题意:给你一个母串,还有一本由字典串构成的字典,问你有几种方式可以用这本字典里的单词构成母串。思路:先用单词构造前缀树,然后用母串遍历查询,每次查询不会超过一百(字典串长度最长100),所以时间复杂度不会超过3 * 10 ^ 7,令d[i]表示从字符i开始的字符串的分解方案数DP:从前往后推:dp[j] += dp[i - 1];预处理dp[0]=1;母串从1-len.    从后往前推:dp[i]=sum{dp[i+dp[x]] | 单词x是母串[i...len]的前缀}*/#include<iostream>#include<cstring>#include<stdio.h>using namespace std;const int MOD=20071027;//求模const int MAXM=300010;//母串的长度const int MAXN=400010;//树的深度(4000*100)int child[MAXN][26];//孩子节点int value[MAXN];//节点的值int dp[MAXM];//dp数组char temp[MAXM];//母串char ch[105];//字典串(构树串)int trieN,len;//当前构建树的深度,母串长度void inset()//清0.{memset(child,0,sizeof(child));memset(value,0,sizeof(value));memset(dp,0,sizeof(dp));trieN=0;//trie树根节点dp[0]=1;//母串从1-len,所以初始化dp[0]=1len=strlen(temp);}void create(char s[])//建树{int x=0;for(int i=0;i<strlen(s);i++){int d=s[i]-'a';if(child[x][d]==0)//不存在孩子节点时{child[x][d]=++trieN;}x=child[x][d];}value[x]++;//把信息保存在字符串的最后一个字母中}void Sreach(char s[]){for(int i=1;i<len;i++)//从前往后推。{int x=0;//每次都从根节点开始。for(int j=i;j<len;j++){int d=s[j]-'a';if(child[x][d]==0){break;}x=child[x][d];if(value[x]){dp[j]=(dp[i-1]+dp[j])%MOD;}}}}int main(){int ans=0;while(scanf("%s",temp+1)!=EOF){temp[0]='1';//预处理(因为要从前往后推,母串从1-len,所以0的位置随便赋值一个不可能的字符)inset();//初始化int n;cin>>n;for(int i=0;i<n;i++)//建树{cin>>ch;create(ch);}Sreach(temp);//搜索cout<<"Case "<<++ans<<": "<<dp[len-1]<<endl;}return 0;}


字典树+DP。 果然还是有点难,不看题解还想不出来。 ~~

0 0
原创粉丝点击