hihocoder 1024:Trie树

hihocoder 1024:Trie树

题目链接：http://hihocoder.com/problemset/problem/1014

对于trie树，要找出具有确定前缀的所有匹配的个数，这个任务可以从两个方面来考虑。

第一方面

找出具有确定前缀的所有字符串匹配是trie树的功能之一：通过自上而下，由根节点不断找到下一字符对应的子节点，遍历可以得到前缀串的最深节点，即末尾字符对应的节点。从当前结点出发，DFS深度优先搜索所有孩子，一旦有节点满足为字符串节点（字符串位设置为true），即当前节点代表的串是一个完整的串，就加入返回字符串集合中。这样不断深搜所有节点，这样就能得到所有满足条件的字符串集合了。

第二方面

1. 由于这里不需要得到所有具有相应前缀匹配的字符串的具体值，而是要得到所有具有前缀的字符串的数目，所以这应该是一个计数问题，而不是一个得到所有解的问题。
2. 由于遍历字典树的过程是自上而下的，上一个节点之能移动到下一个节点，而下一个节点不能到达上一个节点。因此对于这个数据结构，我们只能记录之前的状态，对于后面的状态的统计我们是不知情的。
3. 为了达到统计所有具有相应前缀的字符串的数目，我们在每一个节点上设置一个计数器，计数器初始化为0，表示前缀为该节点对应字符串的字符串的数目，这样每次插入遍历时，如果经过该节点则待插入字符串一定具有该节点的前缀匹配，因此计数器+1。
4. 这样经过插入的字典树，每个结点的计数器的值都代表能与该节点前缀匹配的字符串的数目

源代码

#include<map>
#include<iostream>
using namespace std;

/**
	字典树节点
**/
class TrieNode{
public:
    int prefixN = 0;
    map<char, TrieNode*> children;
};

/**
	字典树
**/
class Trie{
public:
    TrieNode* root = new TrieNode();

    int countParent(string str){
        TrieNode* now = root;

        for(int i = 0; i < str.size(); i++){
            if(now->children.count(str[i]) == 0){
                return 0;
            }
            now = now->children[str[i]];
        }

        return now->prefixN;
    }

    void insert(string str){
        TrieNode* now = root;

        for(int i = 0; i < str.size(); i++){
            if(now->children.count(str[i]) == 0){
                now->children[str[i]] = new TrieNode();
            }
            now = now->children[str[i]];
            (now->prefixN)++;
        }
    }
};

int main(){
    int num;
    string tmp;
    Trie trie;

	// 字符串插入到字典树
    cin >> num;
    while(num--){
        cin >> tmp;
        trie.insert(tmp);
    }

	// 从字典树中查找对应匹配树
    cin >> num;
    while(num--){
        cin >> tmp;
        cout << trie.countParent(tmp) << endl;
    }

    return 0;
}

Think More

1. 计数器初始化为0还是1呢
2. 插入计数器自增时需要判断当前节点是字符串节点吗