hdu 1251 统计难题（字典树）

Posted 2022-12-08 jpphy0

目录

• 问题
• 分析
• 代码

问题

hdu 1251 统计难题 - https://acm.hdu.edu.cn/showproblem.php?pid=1251

分析

• trie[0]作为根

代码

/* hdu 1251 统计难题 */
#include<bits/stdc++.h>
using namespace std;
const int MXN = 500010;
int trie[MXN][26], num[MXN], tot = 0, len;
void insert(char s[])
	for(int p = 0, i = 0; i < len; ++i)
		if(trie[p][s[i]-'a'] == 0) trie[p][s[i]-'a'] = ++tot;
		p = trie[p][s[i]-'a'], ++num[p];
	

int query(char s[])
	int p = 0;
	for(int i = 0; i < len; ++i)
		if(trie[p][s[i]-'a'] == 0) return 0;
		p = trie[p][s[i]-'a'];
		
	return num[p];

int main()
	char s[11];
	memset(trie[0], 0, sizeof trie[0]);	
	while(gets(s) && (len = strlen(s))) insert(s);
	while(gets(s) && (len = strlen(s))) printf("%d\\n", query(s));
    return 0;