SDUT_3039 迷之好奇 【字典树】

题目链接

题目描述

给定n个数字的集合，对于数字x，集合中有多少个数字在x前面添加任意数字可以得到。

解题思路

首先先将输入的字符逆序，这样就可以转化为前缀的问题。
由测试数据可以看出，当要查询的x为集合中的某一数字时，是不进行计数的。
将数字逆序构建字典树，cnt用来计算字符在字典树特定位置中出现的次数，在最后的位置cnt需要减去1，因为完整的字符不符合。当要查询一个前缀的时候，只需要输出他最后一个字符的cnt即可。

代码部分

#include <bits/stdc++.h>
using namespace std;
const int en = 10;
const int maxn = 1e6;
struct Trie
{
    int cnt;
    Trie *nex[en];
} T[maxn];
int top;
inline int idx(char c)
{
    return c - '0';
}
inline Trie * CreatTrie()
{
    Trie *p = &T[top ++];
    p -> cnt = 0;
    for(int i = 0; i< en; ++ i) p -> nex[i] = NULL;
    return p;
}
inline Trie * Init()
{
    top = 0;
    return CreatTrie();
}
void Insert(Trie *root, string s)
{
    Trie *p = root;
    for(int i = 0; i < s.size(); ++ i)
    {
        int temp = idx(s[i]);
        if(p -> nex[temp] == NULL) p -> nex[temp] = CreatTrie();
        p = p -> nex[temp];
        p -> cnt ++;
    }
    p -> cnt --;
}
int Search(Trie *root, string s)
{
    Trie *p = root;
    for(int i = 0; i < s.size(); ++ i)
    {
        int temp = idx(s[i]);
        if(p -> nex[temp] == NULL) return 0;
        p = p -> nex[temp];
    }
    return p -> cnt;
}
int main()
{
    int n, m;
    string s;
    while(cin >> n)
    {
        Trie *root = Init();
        for(int i = 1; i <= n; ++ i) cin >> s, reverse(s.begin(), s.end()), Insert(root, s);
        cin >> m;
        for(int i = 1; i <= m; ++ i) cin >> s, reverse(s.begin(), s.end()), cout<< Search(root, s) << endl;
    }
    return 0;
}