LA3942 背单词(trie树&&简单动规)

    xiaoxiao2021-03-25  142

    【问题描述】

      给出一个由S个不同单词组成的字典和一个长字符串。把这个字符串分解成若干个单词的连接(单词可以重复使用),有多少种方法?

      比如4个单词:{“a”,”b”,”cd”,”ab”},则”abcd”有两种分解方法:”a”+”b”+”cd” 或 “ab”+”cd”。

    【输入格式】

      第一行为长字符串(仅由小写字母组成)。   第二行一个整数n,表示字典包含的单词数。以下n行,每行为一个单词。

    【输出格式】

      输出长字符串的分解方案数除以 20071027 的余数。

    【输入样例】

    abcd 4 a b cd ab

    【输出样例】

    2

    【样例解释】

    时间限制:1秒 内存限制:64M

    【数据范围】

    n<=40000 长字符串长度不超过1 000 000,每个单词长度不超过100。

    【来源】

    大白书209页,LA3942

    建立一棵trie树然后在原字符串上进行简单的动态规划就可以了。 d(i)完成了前i-1个字母有多少中情况。 以每一个字母为起点,在trie树上找有没有对应的单词,比如说i到j个字母为一个单词,d(j)=d(j)+d(i),这样就可以得出答案了。

    详细代码如下:

    #include<cstdlib> #include<cstdio> #include<cstring> #include<iostream> using namespace std; const int maxn=2000005; const int mod=20071027; int ch[maxn][26]={0},cnt=0,n,root=0,m; bool vis[maxn]={0}; char s[1000005],c[105]; int d[1000005]; void in()//建trie树 { int p=root,i=0,k; while(1) { if(c[i]==0) { vis[p]=1; return; } int k=c[i]-'a'; if(!ch[p][k]) ch[p][k]=++cnt; p=ch[p][k]; i++; } } void work(int i) { int p=root,k,j=i; while(i<=m) { if(vis[p])//找到了对应的单词 { d[i]=(d[i]+d[j])%mod; } int k=s[i]-'a'; p=ch[p][k]; if(!p) break;//一定要记得break,没有找的就跳出来 i++; } } int main() { freopen("in.txt","r",stdin); freopen("ou.txt","w",stdout); scanf("%s",s); scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%s",c); in(); } d[0]=1; m=strlen(s); for(int i=0;i<m;i++)//动规 { work(i); } cout<<d[m]; return 0; }
    转载请注明原文地址: https://ju.6miu.com/read-14610.html

    最新回复(0)