码迷,mamicode.com
首页 > 其他好文 > 详细

简单字典树 模板

时间:2018-08-18 13:16:34      阅读:147      评论:0      收藏:0      [点我收藏+]

标签:std   har   bit   bsp   namespace   return   ons   cout   delete   

#include<bits/stdc++.h>
using namespace std;
const int N = 2000000+10;
struct node{
    node *next[26];
    int num;
    bool ok;
    node() {
        for(int i=0;i<26;i++) {
            next[i] = NULL;
        }
        num = 0, ok = 0;
    }
};
int n;
char s[N];
node *root;
int ans = 0;
void dfs(node *rt) {
    if(rt == NULL) return ;
    if(rt->num <= 5) {
        ans++;
        return ;
    }
    for(int i=0;i<26;i++) {
        dfs(rt->next[i]);
    }
}

void ins(char *s) {
    int len = strlen(s);
    node * rt = root;
    for(int i=0;i<len; i++) {
        int id = s[i] - a;
        if(rt->next[id] == NULL) {
            rt->next[id] = new node();
        }
        rt = rt->next[id];
        rt->num++;
    }
    rt->ok = true;
}

void del(node *rt) {
    if(rt == NULL) return ;
    for(int i=0;i<26;i++) {
        if(rt->next[i] != NULL) del(rt->next[i]);
    }
    delete p;
}

int main () {
    int T; scanf("%d",&T);
    while (T--) {
        ans = 0;
        scanf("%d",&n);
        root = new node();
        for(int i=0;i<n;i++)
            scanf("%s",s) , ins(s);
        for(int i=0;i<26;i++) {
            if(root->next[i]!=NULL)
                dfs(root->next[i]);
        }
        cout << ans<<endl;
    }
    return 0;
}

 

简单字典树 模板

标签:std   har   bit   bsp   namespace   return   ons   cout   delete   

原文地址:https://www.cnblogs.com/Draymonder/p/9496600.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!