【ICPC-256】hdu 1251 统计难题

最新推荐文章于 2020-05-27 20:49:58 发布

原创最新推荐文章于 2020-05-27 20:49:58 发布 · 1.5k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#tree #insert #null #search #struct

3. ACM-ICPC 专栏收录该内容

467 篇文章

订阅专栏

本文介绍了一种使用字典树（Trie）来高效处理字符串前缀匹配问题的方法。通过构建字典树并存储每个节点的字符串出现次数，可以快速查询以特定单词为前缀的字符串数量。

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

点击打开链接hdu 1251

题意：给点一序列的字符串，再给你一些单词，问以这个单词为前缀的字符串的个数，注意本身也是自己的前缀

思路：把给定的字符串建立一棵字典树，每一个节点保存的是当前节点为结尾的字符串出现的次数，那么对于给定的单词我们只要去查找字典树即可

代码：


#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int MAXN = 1000010;
const int N = 26;

struct Node{
    int cnt;
    Node* child[N]; 
};
Node node[MAXN];
Node* root;
int pos;

Node* newNode(){
    node[pos].cnt = 0;
    for(int i = 0 ; i < N ; i++)
        node[pos].child[i] = NULL;
    return &node[pos++];
}

void insert(char* str){
    int len = strlen(str);
    Node* s = root;
    for(int i = 0 ; i < len ; i++){
        int num = str[i]-'a';
        if(s->child[num] == NULL)
            s->child[num] = newNode();
        s = s->child[num];
        s->cnt++;
    }
}

int search(char* str){
    int len = strlen(str);
    Node* s = root;
    for(int i = 0 ; i < len ; i++){
        int num = str[i]-'a';
        if(s->child[num] == NULL)
            return 0;
        s = s->child[num];
    }
    return s->cnt;
}

int main(){
    pos = 0;
    root = newNode();
    char str[N];
    while(gets(str) && str[0] != '\0')
        insert(str); 
    while(scanf("%s" , str) != EOF)
        printf("%d\n" , search(str));
    return 0;
}