Trie implementation -- method II

struct TrieNode {
    bool isEnd;
    vector<TrieNode*> children;
    
    TrieNode():isEnd(false){
        children.resize(26, NULL);
    }
};


class Trie{
    private :
        TrieNode* root;

    public :
    Trie(){
        root = new TrieNode();
    }
    
    void insert(const string& str) {
        TrieNode* runner = root;
        for( int i = 0; i < str.size(); i++){
            int idx = str[i] - 'a';
            if(!runner->children[idx]) {
                runner->children[idx] = new TrieNode();
            }
            runner = runner->children[idx];
        }
        runner->isEnd = true;
    }
    
    bool searchWord(const string& str) {
        TrieNode* runner = root;
        for( int i = 0; i < str.size(); i++ ) {
            int idx = str[i] - 'a';
            if( !runner->children[idx] ) return false;
            runner = runner->children[idx];
        }
        return runner->isEnd;
    }
    
    bool searchPre(const string& pre) {
    }
    
    vector<string> dfs_traverse(){
        string oneRst;
        vector<string> rst;
        helper(root, oneRst, rst);
        return rst;
    }
    
    void helper( TrieNode* runner, string& oneRst, vector<string>& rst ) {
        if(!runner) return;

        if(runner->isEnd) {
            rst.push_back(oneRst);
        }
        
        for( int i = 0; i < 26; i++ ) {
            TrieNode* child = runner->children[i];
            if(child) {
                oneRst.push_back('a' + i);
                helper(child, oneRst, rst);
                oneRst.pop_back();
            }
        }
    }


};


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值