uva 1401 字典树模板

给一个串和一个字典（一些字符串）

将这个串分解为字典单词的连接（如 abcd = ab+cd =a+bcd )

问有多少方案

线性dp

枚举位置 i

f[i] += f[j] i<j , string(i,j) 为字典单词

直接枚举 j 和单词 明显超时

用到字典树，从 位置 i 开始在字典树上找单词，找到后更新 f[i]

#include <iostream>
#include <cstring>
#include <vector>
using namespace std;
 #define mod 20071027
 const int N=5e5;
 
 int n;
 int tot,ch[N][30],val[N];
 char str[N],tmp[N];
 int f[N],L[N];
 
 void insert(char *s,int v){
     int i,u=1,l=strlen(s);
     
     for(i=0;i<l;i++){
         int c=s[i]-'a';
         if(ch[u][c]==0){
             ++tot;
             memset(ch[tot],0,sizeof ch[tot]);
             ch[u][c]=tot; val[tot]=0;
         }
         u=ch[u][c];
     }
    val[u]=v;
 }
 void find(char *s,int len,int id){
     int i,u=1;
     for(i=0;i<len;i++){
         if(s[i]=='\0') break;
         int c=s[i]-'a';
         if(ch[u][c]==0) break;
         u=ch[u][c];
         if(val[u]!=0) 
          f[id]+=f[id+L[val[u]]],f[id]%=mod;
     }
 }
 int main(){
 //    
     int i,j,cas=0;
     
     while(cin>>str>>n){
         memset(f,0,sizeof f);
         tot=1; 
         memset(ch[1],0,sizeof ch[1]);
         for(i=1;i<=n;i++){
             cin>>tmp; insert(tmp,i); L[i]=strlen(tmp);
         }
         int l=strlen(str);
         f[l]=1;
        
        for(i=l-1;i>=0;i--){
            find(str+i,l-i,i);
        }
        printf("Case %d: %d\n",++cas,f[0]);
     }
 }