hdu 1251 统计难题（求前缀出现了多少次）

weixin_30383279

于 2019-08-10 08:22:00 发布

阅读量82

点赞数

CC 4.0 BY-SA版权

文章标签： php

原文链接：http://www.cnblogs.com/LLLAIH/p/11330433.html

本文详细解析了一道ACM编程竞赛题目，介绍了如何使用字典树（Trie Tree）进行字符串查找和计数，包括插入字符串到字典树、在字典树中查找字符串并返回匹配的字符串数量。代码实现采用C++，展示了完整的字典树操作过程。

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

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1251

#include <iostream>
#include <cstring>
#include <cstdio>
#define mem(a,b) memset(a,b,sizeof(a));
using namespace std;
typedef long long ll;
const int maxn = 500005;
const ll INF = 0x3f3f3f3f;
int tot,n,trie[maxn][26],sum[maxn];
bool vis[maxn],flag;
void Insert(char *s,int rt)
{
    int len = strlen(s);
    for(int i = 0; i < len; i++)
    {
        int k = s[i] -'a';
        if(trie[rt][k] == 0) trie[rt][k] = ++tot;
        sum[trie[rt][k]]++;
        rt = trie[rt][k];
    }
}
int Find(char *s,int rt)
{
    int len = strlen(s);
    for(int i = 0; i < len; i++)
    {
        int k = s[i] -'a';
        rt = trie[rt][k];
        if(rt == 0) return 0;
    }

    return sum[rt];
}
int main()
{
    tot = 0;
    int rt = 0;
    flag = false;
    char s[105];
    while(1)
    {
        gets(s);
        if(s[0] == NULL)
        {
            break;
        }
            Insert(s,rt);
    }
    while(cin >> s) {
            cout << Find(s,rt) << endl;
    }
    return 0;
}