hdu4287字典树

最新推荐文章于 2021-08-23 23:45:30 发布

贫嘴小李子的幸福生活

最新推荐文章于 2021-08-23 23:45:30 发布

阅读量420

点赞数

CC 4.0 BY-SA版权

分类专栏： C++足迹

本文链接：https://blog.youkuaiyun.com/u013491262/article/details/51137393

C++足迹专栏收录该内容

274 篇文章

订阅专栏

本文探讨了Trie结构在C++中的实现及其在字符串查询中的应用，包括节点构建、添加字符串和查询功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <queue>
#include <string>
#include <string.h>
#include <map>
#include <vector>

const int nodeSize = 5001*6 ;
const int alphaSize = 26 ;
struct Trie{
    int root ;
    int totel ;
    int next[nodeSize][alphaSize] ;
    int cnt[nodeSize] ;

    int newNode(){
        memset(next[totel] , 0 , sizeof(next[totel])) ;
        cnt[totel] = 0 ;
        return totel++ ;
    }

    void clear(){
         totel = 0 ;
         root = newNode() ;
    }

    void add(char *s){
         int now = root ;
         for(int i = 0 ; s[i] ; i++){
              int son = s[i] - 'a' ;
              if(! next[now][son])
                    next[now][son] = newNode() ;
              now = next[now][son] ;
         }
         cnt[now]++ ;
    }

};

Trie trie ;

char dir[10][5] = {"" , "" , "abc" , "def" , "ghi" , "jkl" , "mno" , "pqrs" , "tuv" , "wxyz"} ;

int query(char *s , int now){
    if(*s == 0) return trie.cnt[now] ;
    int res = 0 ;
    int d = *s - '0' ;
    for(int i = 0 ; i < strlen(dir[d]) ; i++){
        int son = dir[d][i] - 'a' ;
        if(! trie.next[now][son]) continue ;
        res += query(s+1 , trie.next[now][son]) ;
    }
    return res ;
}

int main(){

    int t , n , m ;
    scanf("%d" , &t) ;
    while(t--){
        scanf("%d%d" , &n , &m) ;
        for(int i = 0 ; i < n ; i++) scanf("%s" , ask[i]) ;

        trie.clear() ;
        while(m--){
            scanf("%s" , word) ;
            trie.add(word) ;
        }

        for(int i = 0 ; i < n ; i++) printf("%d\n" , query(ask[i] , trie.root)) ;

    }

    return 0 ;
}