Remember the Word UVA - 1401 (Trie+dp)

版权声明: https://blog.csdn.net/sgh666666/article/details/82470140

题意:

        给你一个由N个单词组成的词典,和一个字符串S。问你S由N中的单词组成的方法有多少种?字典中的单词可以重复使用但是不可重叠。

分析:

        d[i] = x 表示S串的后缀[i,L-1]串有多少种构造方式.L = strlen(S).字符从0到L-1下标.

d[i] = sum(d[i+len(x)]) 仅当[i,len(x-1)]区间正好是字典中的一个单词.

初始化:d[L] = 1.  最后求得d[0]即为所求答案.

为什么要倒着求?要运用Trie树,如果正着求.求d[5]的话,暴力的去扫0,4,去看他们到5是否是字典中的字符串.每次共调用5次find函数.显然倒着能够更好地运用Trie树.

#include<bits/stdc++.h>
#include<cstdio>
#include<iostream>

using namespace std;
const int maxn = 4e5+1111;


struct node
{
    int ch[maxn][26];
    int v[maxn];
    int sz;

    void init()
    {
        sz=1;
        memset(ch[0],0,sizeof ch[0]);
    }

    int idx(char c)
    {
        return c-'a';
    }

    void insert(char *s)
    {
        int u=0,n=strlen(s);
        for(int i=0;i<n;i++)
        {
            int id=idx(s[i]);
            if(ch[u][id]==0)
            {
                ch[u][id]=sz;
                memset(ch[sz],0,sizeof ch[sz]);
                v[sz++]=0;
            }
            u=ch[u][id];
        }
        v[u]=n;///v存放长度
    }

    void find(char *s,int len,vector<int>&vc)
    {
        int u=0;
        for(int i=0;i<len;i++)
        {
            int id=idx(s[i]);
            if(ch[u][id]==0)
                return;
            u=ch[u][id];///
            if(v[u])vc.push_back(v[u]);
        }
    }
};

node trie;
const int mod=20071027;
char s[300000+111],word[1111];
int  d[300000+111];
int main()
{
    int kase=1;
    while(~scanf("%s",s))
    {
        int w;
        cin>>w;
        trie.init();
        for(int i=0;i<w;i++)
        {
            scanf("%s",word);
            trie.insert(word);
        }
        memset(d,0,sizeof d);
        int l=strlen(s);
        d[l]=1;
        for(int i=l-1;i>=0;i--)
        {
            vector<int>vc;
            trie.find(s+i,l-i,vc);
            for(int j=0;j<vc.size();j++)
            {
                d[i]=(d[i]+d[ i+vc[j] ])%mod;
            }
        }
        printf("Case %d: %d\n",kase++,d[0]);
    }

}

猜你喜欢

转载自blog.csdn.net/sgh666666/article/details/82470140