【YbtOJ Trie - 1】前缀统计

本文讲解了如何使用Trie树数据结构解决字符串查询问题,通过统计以特定字符序列结尾的字符串数量,实现高效查找。通过实例展示了处理输入32个字符和一系列查询的操作过程和输出结果。

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

前缀统计


题目

在这里插入图片描述


输入

在这里插入图片描述

输出

对于每个询问,输出一个整数表示答案

每个答案占一行


输入样例

3 2
ab
bc
abc
abc
efg

输出样例

2
0

解题思路

我们把nnn个字符都存入TrieTrieTrie,然后记录一个特殊值来表示字符串的结尾是TrieTrieTrie上的节点xxx的字符串个数


程序如下

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>

using namespace std;

int n, m, e[1000001], trie[1000001][26], tot = 1, ans;

char s[1000001], ss[1000001];

void handle()
{
	scanf("%s", s + 1);
	int len = strlen(s + 1), p = 1;
	for(int i = 1; i <= len; ++i)
	{
		int c = s[i] - 'a';
		if(!trie[p][c]) 
			trie[p][c] = ++tot;
		p = trie[p][c];
	}
	e[p]++;
}

int main()
{
	scanf("%d%d",&n,&m);
	for(int i = 1; i <= n; ++i)
		handle();
	for(int i = 1; i <= m; ++i)
	{
		scanf("%s", ss + 1);
		int len = strlen(ss + 1), p = 1;
        ans = 0;
		for(int j = 1; j <= len; ++j)
		{
			int c = ss[j] - 'a';
			if(trie[p][c])
				p = trie[p][c], ans += e[p];
			else break;
		}
		printf("%d\n",ans);
	}
	return 0;
} 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值