1. 程式人生 > >Remember the Word UVALive - 3942(字典樹trie)

Remember the Word UVALive - 3942(字典樹trie)

傳送門

題意:這個題是先給出一個長的字串,然後相應給出若干個單詞,問這個字串能由後面給出的若干個單詞構成的方案數,比如給出abcd,給出的若干個單詞是a,b,ab,cd,這樣就有兩種構成方法了

題解:最最重要的是得要想出遞推得出答案,用d[i]表示從給出的字串中的第i位開始的字尾所能構成的方案數,然後對於每一位我們直接d[i]=d[i]+d[i+len[p[j]]即可從後往前推出答案了,最後就是如果快速得出字首呢?使用字典樹

附上程式碼:


#include<bits/stdc++.h>

using namespace std;

const int maxnode=4e3*2e2+10;
const int sigma_size=26;

const int maxl=3e5+10;
const int maxw=4e3+10;
const int maxwl=1e2+10;
const int MOD=20071027;

int d[maxl],len[maxw],S;
char text[maxl],word[maxwl];

struct Trie{
    int ch[maxnode][sigma_size];
    int val[maxnode];
    int sz;
    void clear(){
        sz=1;
        memset(ch[0],0,sizeof(ch[0]));
    }
    int idx(char c){
        return c-'a';
    }
    void insert(char *s,int v){
        int u=0,n=strlen(s);
        for(int i=0;i<n;i++){
            int c=idx(s[i]);
            if(!ch[u][c]){
                memset(ch[sz],0,sizeof(ch[sz]));
                val[sz]=0;
                ch[u][c]=sz++;
            }
            u=ch[u][c];
        }
        val[u]=v;
    }
    void find_prefixes(char *s,int len,vector<int>&ans){
        int u=0;
        for(int i=0;i<len;i++){
            if(s[i]=='\0'){
                break;
            }
            int c=idx(s[i]);
            if(!ch[u][c]){
                break;
            }
            u=ch[u][c];
            if(val[u]!=0){
                ans.push_back(val[u]);
            }
        }
    }
};
Trie trie;

int main()
{
    int kase=1;
    while(scanf("%s%d",text,&S)!=EOF){
        trie.clear();
        for(int i=1;i<=S;i++){
            scanf("%s",word);
            len[i]=strlen(word);
            trie.insert(word,i);
        }
        memset(d,0,sizeof(d));
        int L=strlen(text);
        d[L]=1;
        for(int i=L-1;i>=0;i--){
            vector<int>p;
            trie.find_prefixes(text+i,L-i,p);
            for(int j=0;j<p.size();j++){
                d[i]=(d[i]+d[i+len[p[j]]])%MOD;
            }
        }
        printf("Case %d: %d\n",kase++,d[0]);
    }
    return 0;
}