标签：rt 专题 const Trie void 练习 tot char int

HDU 1251 统计难题

Solution:

Trie。插入字符串时，将前缀链上的所有结点的点权加一，代表该前缀出现次数。查询时直接返回出现次数即可。注意输入格式。

Code:

#include<bits/stdc++.h>
using namespace std;
const int N=11;
const int M=5e5+10;
char s[N];
int tr[M][26];
int endpos[M];
struct trie
{
	int tot;//点数
	int rt;//根
	void init(){
		tot=0;
		rt=0;
	}
	void insert1(char s[],int lens){
		int pos=rt;
		for(int i=0;i<lens;++i){
			int v=s[i]-'a';
			if(!tr[pos][v])tr[pos][v]=++tot;
			pos=tr[pos][v];
			endpos[pos]++;
		}
		
	}
	int query(char s[],int lens){
		int pos=rt;
		for(int i=0;i<lens;++i){
			int v=s[i]-'a';
			if(!tr[pos][v])return 0;
			pos=tr[pos][v];
		}
		return endpos[pos];
	}
};
int main(){
	trie T;
	T.init();
	while(cin.getline(s,N)){
		int len=strlen(s);
		if(len==0)break;
		T.insert1(s,len);
	}
	while(cin.getline(s,N)){
		int len=strlen(s);
		printf("%d\n", T.query(s,len));
	}
	return 0;
}

标签：rt,专题,const,Trie,void,练习,tot,char,int
来源： https://www.cnblogs.com/quuns/p/15005987.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。