hdu2846 Repository（字典树）

通过字典树，还学习了后缀的思想。

 

本题要求的是一个字符串集合中含有某子串的个数(以前也做过string判断子串，这次只看字典树嘻嘻^ ^)，普通字典树不好判断，因为查找起点可能在树中。这时就要引入后缀的思想，把样本字符串的所有后缀输入字典树，从而待查找的字符串，肯定是这些后缀的一个前缀，最后只需返回后缀数目即可。

 

注意abab的情况，后缀数有可能重复，所以每次前面添加一个编号num，若此编号不同于所给编号(此后缀在本要查的字符串中没有)才加1，这样就避免了重复。

 

#include <stdio.h>
#include <algorithm>
#include <stdlib.h>
#include <string.h>
#include <iostream>

using namespace std;

typedef long long LL;

const int N = 35;
const int INF = 1e8;

struct Trie
{
    int sum;
    int num;
    Trie *next[26];
    Trie()
    {
        num = 0;
        sum = 0;
        for(int i = 0; i < 26; i ++)
            next[i] = 0;
    }
}root;

void inserttrie(char *str, int num0)
{
    int k = 0;
    Trie *p = &root;
    while(str[k] != '\0')
    {
        int id = str[k] - 'a';
        if(p -> next[id] == 0) p -> next[id] = new Trie;
        p = p -> next[id];
        if(p -> num != num0) p -> sum ++;
        p -> num = num0;
        k ++;
    }
}

int findtrie(char *str)
{
    int k = 0;
    Trie *p = &root;
    while(str[k] != '\0' && p -> next[str[k] - 'a']) p = p -> next[str[k ++] - 'a'];
    return str[k] == '\0' ? p -> sum : 0;
}

int main()
{
   // freopen("in.txt", "r", stdin);
    char s[N];
    int p, q;
    scanf("%d", &p);
    for(int i = 1; i <= p; i ++)
    {
        scanf("%s", s);
        int len = strlen(s);
        for(int j = 0; j < len; j ++)
        {
            inserttrie(s + j, i);
        }
    }
    scanf("%d", &q);
    for(int i = 0; i < q; i ++)
    {
        scanf("%s", s);
        printf("%d\n", findtrie(s));
    }
    return 0;
}