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

/*
题意:给你一个母串,还有一本由字典串构成的字典,问你有几种方式可以用这本字典里的
单词构成母串。
思路:先用单词构造前缀树,然后用母串遍历查询,每次查询不会超过一百(字典串长度最长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]=1
	len=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。 果然还是有点难,不看题解还想不出来。 ~~

你可能感兴趣的:(数据结构树)