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;}