【字典树】poj1451 T9

第一次做字典树的题目，吐血啊简直。。。。。。最后调试了一个多小时，最终AC了，代码丑陋不堪，惭愧，实在惭愧。。。。

#include<iostream>
#include<queue>
#include<vector>
#include<cstring>
using namespace std;

//字典树的每个节点中存储权重和指针两部分 
struct node{
	node* ptr[26];
	int weight[26];
};
node root;

struct element{
	node* pt;
	vector<char> v;
};

int build_tree(char str[30],int wei);
int delete_tree(node* tmp_root);
int query_dic(char query[30]);
int search(int* min_pos,int* max_pos,char num);
void initial(node* tmp); 

int main(){
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	int t,n,wei,query_num;
	char query[30],str[30];
	cin>>t;	
	for(int i=0;i<t;i++){
		initial(&root);
		cout<<"Scenario #"<<i+1<<":"<<endl;
		cin>>n;
		for(int j=0;j<n;j++){ 
			cin>>str>>wei;
			build_tree(str,wei);
		}		
		cin>>query_num;
		for(int j=0;j<query_num;j++){
			cin>>query;
			//进行字典查询并输出结果 
			query_dic(query);
			cout<<endl; 			
		} 
		delete_tree(&root);		
		cout<<endl;
	}
	return 0;
}

//初始化一个新建节点的过程 
void initial(node* tmp){
	for(int i=0;i<26;i++){
		tmp->ptr[i]=NULL;
		tmp->weight[i]=0;
	}
	return;
}

//将一个新的单词插入到字典树中，同时更新权值 
int build_tree(char str[30],int wei){
	int len=strlen(str),cur=0;
	int pos; 
	node* tmp=&root;
	//将每个单词的字母逐个插入字典树中 
	while(len--){
		pos=str[cur++]-'a';
		if(tmp->ptr[pos]==NULL){
			tmp->ptr[pos]=new node;
			initial(tmp->ptr[pos]);
		}
		//相同前缀，权重相加 
		tmp->weight[pos]+=wei;
		tmp=tmp->ptr[pos];
	}
	return 0;
}

//进行查询操作，本程序最核心的部分 
int query_dic(char query[30]){
	int len=strlen(query),weight;
	int min_pos,max_pos,max_weight,tmp;
	query[len-1]='\0'; len--;
	
	//为查找做变量定义上的准备 
	element ele;
	node* tmp_ptr,max_node;
	element max_ele,tmp_ele;
	queue<element> q;
	while(q.empty()==false)
		q.pop();
	ele.pt=&root;
	ele.v.clear();
	q.push(ele);
	
	bool find=true;
	//进行查找的过程 
	for(int i=0;i<len;i++){
		//此时查找失败 
		if(find==false){
			cout<<"MANUALLY"<<endl;
			continue;
		}
		max_weight=-1;	
		tmp=q.size();		
		for(int j=0;j<tmp;j++){
			tmp_ele=q.front();
			q.pop();
			//寻找起点与终点 
			search(&min_pos,&max_pos,query[i]);				
			//对一个按键的所有可能情况进行尝试 
			for(int k=min_pos;k<=max_pos;k++){				
				if(tmp_ele.pt->ptr[k]!=NULL){
					element* ele_ptr=new element;					
					ele_ptr->pt=tmp_ele.pt;
					ele_ptr->v=tmp_ele.v;
					(ele_ptr->v).push_back('a'+k);
					//更新最大值 
					if((ele_ptr->pt->weight[k])>max_weight){
						max_weight=ele_ptr->pt->weight[k];
						max_ele=*ele_ptr;
					}
					ele_ptr->pt=ele_ptr->pt->ptr[k];					
					q.push(*ele_ptr);					
				}
			}
		}
		//对应于查找失败的情形 
		if(q.empty()==true){
			find=false;
			//cout<<"悲剧啊！"<<endl; 
			cout<<"MANUALLY"<<endl;
		}		
		//查找成功 
		else{		
			int length=max_ele.v.size();
			for(int k=0;k<length;k++)
				cout<<(max_ele.v)[k];			
			cout<<endl;
		}					
	}	
	return 0;
} 

//搜索起点和终点的过程 
int search(int* min_pos,int* max_pos,char num){
	int goal=num-'2';
	//采用了最苦逼的枚举法。。。。 
	if(num<'7'){
		*min_pos=goal*3;
		*max_pos=*min_pos+2;
	}		
	else if(num=='7'){
		*min_pos='p'-'a';
		*max_pos='s'-'a';
	}
	else if(num=='8'){
		*min_pos='t'-'a';
		*max_pos='v'-'a';
	}
	else if(num=='9'){
		*min_pos='w'-'a';
		*max_pos='z'-'a';
	}
	else{
		*min_pos=0;
		*max_pos=0;
	}
	return 0;
} 

//递归删除字典树，只保留根节点 
int delete_tree(node* tmp_root){
	for(int i=0;i<26;i++){
		//先将子树删除，再删除根节点 
		if(tmp_root->ptr[i]!=NULL)
			delete_tree(tmp_root->ptr[i]);
	}
	if(tmp_root!=(&root))
		delete tmp_root;
	return 0;
}