POJ 1432 Decoding Morse Sequences

题意

给出一个字母的译码表，给出一个有n个单词的字典，给出一串编码。问再给出字典中有多少种不同的解码方式。

分析

首先编码只有-和.，可以很方便地建立一棵Trie树，还是最熟悉的二叉方式。
然后令opt[i]表示区间[i..len]的方案数
则有opt[i]=sigma{[i..j]组成的单词的个数*opt[j+1]|i<=j

代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<bitset>
using namespace std;
const int INF=0x3f3f3f3f,MAXN=10000;
typedef bitset<MAXN+3> data;
const char* morse[26] = {
    ".-",   "-...", "-.-.", "-..",
    ".",    "..-.", "--.",  "....",
    "..",   ".---", "-.-",  ".-..",
    "--",   "-.",   "---",  ".--.",
    "--.-", ".-.",  "...",  "-",
    "..-",  "...-", ".--",  "-..-",
    "-.--", "--.."
};
struct Trie{
    int ch[4*20*MAXN+3][2],cnt[4*20*MAXN+3],tot;
    void init(){
        memset(ch,0,sizeof(ch));
        memset(cnt,0,sizeof(cnt));
        tot=0;
    }
    void insert(const data &d,int len){
        int cur=0;
        for(int i=0;i<len;i++){
            if(!ch[cur][d[i]])ch[cur][d[i]]=++tot;
            cur=ch[cur][d[i]];
        }
        cnt[cur]++;
    }
}T;
int getmorse(char *s,data &d){
    int len=strlen(s),p=0;
    d=0;
    for(int i=0;i<len;i++){
        int tl=strlen(morse[s[i]-'A']);
        for(int j=0;j<tl;j++){
            if(morse[s[i]-'A'][j]=='.')d[p++]=0;
            else d[p++]=1;
        }
    }
    return p;
}
char str[MAXN+3],tmp[MAXN+3];
int n,opt[MAXN+3];
void input(){
    scanf("%s",str);
    scanf("%d",&n);
    T.init();
    data d;
    for(int i=1;i<=n;i++){
        scanf("%s",tmp);
        T.insert(d,getmorse(tmp,d));
    }
}
int dp(){
    int len=strlen(str);
    opt[len]=1;
    for(int i=len-1;i>=0;i--){
        opt[i]=0;
        int cur=0;
        for(int j=i;j<len;j++){
            if(str[j]=='.'){
                if(!T.ch[cur][0])break;
                cur=T.ch[cur][0];
            }
            else{
                if(!T.ch[cur][1])break;
                cur=T.ch[cur][1];
            }
            if(T.cnt[cur]!=0)opt[i]+=T.cnt[cur]*opt[j+1];
        }
    }
    return opt[0];
}
int main(){
    int T;scanf("%d",&T);
    while(T--){
        input();
        printf("%d\n",dp());
    }
    return 0;
}