uvaliva3942(trie树)

题目连接：https://vjudge.net/problem/UVALive-3942

trie树

dp[i]=sum(dp[i+len(x)]%mod;

dp[i]表示从字符i开始的字符串的分解方案方案数，x是s[i……L]的前缀

#include<cstdio>
#include<cstring>
const int maxnode=500010;
const int sigma=26;
const int N=300010;
const int mod=20071027;
int ch[maxnode][sigma];
int val[maxnode];
int sz;
inline int id(char c) {return c-'a';}

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

void inser(char *s,int v)
{
    int u=0;
    for(int i=0;s[i];i++)
    {
        int c=id(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;
}
int num,dp[N],st;
void query(char *s)
{
    int u=0;
    for(int i=0;s[i];i++)
    {
        int c=id(s[i]);
        if(ch[u][c])
        {
            u=ch[u][c];
            if(val[u]) num=(num+dp[st+i+1])%mod;  //
        }
        else return;
    }
}
int main()
{
    char s[N],p[110];
    int k=0;
    while(scanf("%s",s)!=EOF)
    {
        trie_init();
        int m;
        scanf("%d",&m);
        while(m--)
        {
            scanf("%s",p);
            inser(p,1);
        }
        int len=strlen(s);
        dp[len]=1;
        for(int i=len-1;i>=0;i--)
        {
            num=0;
            st=i;
            query(&s[i]);
            dp[i]=num;
        }
        printf("Case %d: %d
",++k,dp[0]);
    }
    return 0;
}

lrj：

// LA3942 Remember the Word
// Rujia Liu
#include<cstring>
#include<vector>
using namespace std;

const int maxnode = 4000 * 100 + 10;
const int sigma_size = 26;

// 字母表为全体小写字母的Trie
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'; } // 字符c的编号

  // 插入字符串s，附加信息为v。注意v必须非0，因为0代表“本结点不是单词结点”
  void insert(const 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;  // 中间结点的附加信息为0
        ch[u][c] = sz++; // 新建结点
      }
      u = ch[u][c]; // 往下走
    }
    val[u] = v; // 字符串的最后一个字符的附加信息为v
  }

  // 找字符串s的长度不超过len的前缀
  void find_prefixes(const char *s, int len, vector<int>& ans) {
    int u = 0;
    for(int i = 0; i < len; i++) {
      if(s[i] == '') break;
      int c = idx(s[i]);
      if(!ch[u][c]) break;
      u = ch[u][c];
      if(val[u] != 0) ans.push_back(val[u]); // 找到一个前缀
    }
  }
};

#include<cstdio>
const int maxl = 300000 + 10; // 文本串最大长度
const int maxw = 4000 + 10;   // 单词最大个数
const int maxwl = 100 + 10;   // 每个单词最大长度
const int MOD = 20071027;

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

int main() {
  int kase = 1;
  while(scanf("%s%d", text, &S) == 2) {
    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
", kase++, d[0]);
  }
  return 0;
}