字典树 01字典树【数据结构】

题目1:字典树
HDU 1251 统计难题
Problem Description
Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).

Input
输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.

注意:本题只有一组测试数据,处理到文件结束.

Output
对于每个提问,给出以该字符串为前缀的单词的数量.

Sample Input
banana band bee absolute acm ba b band abc

Sample Output
2 3 1 0

不知道字典树的推荐一个链接:大佬写的的字典树

ac代码:

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=5e5+5;
int ans=1;
struct node{
	int tot;
	int child[30];
	node(){
		tot=0;
		memset(child,0,sizeof child);
	}
}tree[maxn];
void ins(char s[]){
	int len=strlen(s);
	int k=0;
	for(int i=0;i<len;i++){
		int c=s[i]-'a';
		if(tree[k].child[c]==0){
			tree[k].child[c]=ans++;
		}
		k=tree[k].child[c];
		tree[k].tot++;
	}
}
int sear(char s[]){
	int len=strlen(s);
	int k=0;
	for(int i=0;i<len;i++){
		int c=s[i]-'a';
		if(tree[k].child[c]==0) return 0;
		k=tree[k].child[c];
	}
	return tree[k].tot;
}
int main(){
	char s[11];
	while(cin.getline(s,50)){
		if(strlen(s)==0) break;
		ins(s);
	}
	while(cin.getline(s,50)){
		cout<<sear(s)<<endl;
	}
	return 0;
}

题目2:01字典树
hdu 4825 Xor Sum

Problem Description
Zeus 和 Prometheus 做了一个游戏,Prometheus 给 Zeus 一个集合,集合中包含了N个正整数,随后 Prometheus 将向 Zeus 发起M次询问,每次询问中包含一个正整数 S ,之后 Zeus 需要在集合当中找出一个正整数 K ,使得 K 与 S 的异或结果最大。Prometheus 为了让 Zeus 看到人类的伟大,随即同意 Zeus 可以向人类求助。你能证明人类的智慧么?

Input
输入包含若干组测试数据,每组测试数据包含若干行。
输入的第一行是一个整数T(T < 10),表示共有T组数据。
每组数据的第一行输入两个正整数N,M(<1=N,M<=100000),接下来一行,包含N个正整数,代表 Zeus 的获得的集合,之后M行,每行一个正整数S,代表 Prometheus 询问的正整数。所有正整数均不超过2^32。

Output
对于每组数据,首先需要输出单独一行”Case #?:”,其中问号处应填入当前的数据组数,组数从1开始计算。
对于每个询问,输出一个正整数K,使得K与S异或值最大。

Sample Input
2 3 2 3 4 5 1 5 4 1 4 6 5 6 3

Sample Output
Case #1:
4
3
Case #2:
4

01字典树概述:
01字典树其实和字典树类似,建树查找都类似,结构体也相似。可以看作是把一个整数变成二进制数串,再将他们像字典树那样存进去。只是在对求异或最大数目的时候加了一些操作首先,将一个数由高位到低位(也是为后面贪心思想做铺垫)录入到字典树中。然后取一个贪心的思想,每次寻找路径(即遍历每一位)的时候,尽量选择异或结果为1的那条边,因为01字典树每个节点对应两条边,这样就可以达到最大。

ac代码:

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<algorithm>
#include<math.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
int t,n,m,q,ans=0,add=1;
int a[maxn];
struct node{
	int val;
	int cnt[2];
}tree[maxn*32];
void add_num(int x){
	int u=0;
	for(int i=31;i>=0;i--){
		int c=(x>>i)&1;  //右移31位后,第一位就是原来的31位,再&1就能得到当前位是否为1
		if(tree[u].cnt[c]==0){
			tree[u].cnt[c]=add++; 
		}
		u=tree[u].cnt[c];
	}
	tree[u].val=x;  //路径末尾节点记录数字的十进制位值
}
int query(int x){
	int k=0;
	for(int i=31;i>=0;i--){
		int c=(x>>i)&1;//右移31位后,第一位就是原来的31位,再&1就能得到当前位是否为1
		if(tree[k].cnt[c^1]) k=tree[k].cnt[c^1]; /*
		这里,就是为了保证每次取最大。
		我们可以举个例子,假如当前位为1,则异或1得到的就是0,
		若0存在,我们就取第0位。假如当前是0,那么异或出来就为1,
		若1存在,我们就取0,然后我们发现这样就
		达到了每次取最大的效果*/
		else k=tree[k].cnt[c];
	}
	return tree[k].val; //返回路径末尾节点得到数字的十进制位值
}
int main(){
	scanf("%d",&t);
	while(t--){
		for(int i=0;i<maxn*32;i++){
			for(int j=0;j<2;j++){
				tree[i].cnt[j]=0;
			}
		}
		ans++;
		printf("Case #%d:\n",ans);
		scanf("%d%d",&n,&m);
		for(int i=0;i<n;i++) scanf("%d",&a[i]);
		for(int i=0;i<n;i++) add_num(a[i]);
		while(m--){
			scanf("%d",&q);
			printf("%d\n",query(q));
		}
	}
	return 0;
}
发布了43 篇原创文章 · 获赞 56 · 访问量 5127

猜你喜欢

转载自blog.csdn.net/tran_sient/article/details/97263399