最优二叉树

菜鸟生成记(46)

最优二叉树

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

//int arr[8]={2,3,4,5,7};
//int arr[8]={2,3,4,7,14,15,12};
class Node{
	public:
		int w;//权值 
		int l,r;//左右孩子key (值为-1，表示没有左右孩子结点)
		int parents;//父亲的key (值为-1，表示没有父结点)
		int flag;//访问标记 (值为-1，表示未被访问过)
	public:
		Node(){
			w=0;
			flag=r=l=parents=-1;
		}
		Node(int w){
			this->w=w;
			flag=l=r=parents=-1;
		}
		Node(int l,int r,int w){
			this->l=l;
			this->r=r;
			this->w=w;
			flag=parents=-1;
		}	
};
vector<Node> v;//在数组v中构建哈夫曼树 
int n;//结点数 

void input();//输入结点个数n和n个结点的w(权值) 
int findMin();//获取未被选过的最小值的key 
void createTree();//构建哈弗曼树 
int findRoot();//查询根节点的key 
void dfs01(int rt);//前序遍历 
void dfs02(int rt);//中序遍历 
void dfs03(int rt);//后序遍历 
void print();//打印结点之间的关系 
/*
7
2 3 4 7 14 15 12

*/
int main(){
	input();//输入 
	createTree();//建树 
	print();//打印结点之间的关系 
	cout<<endl;
	
	int root=findRoot();//查询根节点的key 
	cout<<"根结点key:"<<root<<endl;
	cout<<"前序遍历哈弗曼树:"<<endl;
	dfs01(root);//前序遍历 
	cout<<"\n中序遍历哈弗曼树:"<<endl;
	dfs02(root);//中序遍历 
	cout<<"\n后序遍历哈弗曼树:"<<endl;
	dfs03(root);//后序遍历 
	return 0;
}
void input(){
	cout<<"输入初始结点个数："<<endl;
	cin>>n;
	cout<<"输入["<<n<<"]个初始结点的权值："<<endl;
	for(int i=0;i<n;i++){
		int w;
		cin>>w;
		Node *node=new Node(w);//构建结点 
		v.push_back(*node);//装入结点 
	}
}
int findMin(){//获取未被选过的最小值的key 
	int minKey=-1;
	int minValue=1000;
	for(int i=0;i<v.size();i++){
		if(minValue>v[i].w&&v[i].flag==-1){
			
			minValue=v[i].w;
			minKey=i;
		}
	}
	v[minKey].flag=1; //选中标记 (标记当前最小值)
	return minKey;
}
void createTree(){
	int k=n;
	for(int i=n;i<2*n-1;i++){
		//寻找最小的两个结点的key 
		int minKey1=findMin();
		int minKey2=findMin();
		
		// 最小的两个结点的w(权值)求和，作为它们父结点的w(权值) 
		int wSum=v[minKey1].w+v[minKey2].w;
		
		//设置父亲 
		v[minKey1].parents=k;
		v[minKey2].parents=k;
		
		//父节点 
		Node *newNode;
		if(v[minKey1].w<v[minKey2].w)//权值w小的作为左孩子 
		newNode=new Node(minKey1,minKey2,wSum);
		else
		newNode=new Node(minKey2,minKey1,wSum);
		
		//装入父结点 
		v.push_back(*newNode);
		k++;		
	}
}
int findRoot(){//查询根节点的key 
	int parents=-1;
	for(int i=0;i<v.size();i++){
		if(v[i].parents==-1){//父结点==-1，(为根结点) 
			parents=i;
		}
	}
	return parents;
}
void dfs01(int rt){//前序遍历 
	if(rt!=-1){
		cout<<v[rt].w<<" ";
		dfs01(v[rt].l);
		dfs01(v[rt].r);
	}
}
void dfs02(int rt){//中序遍历 
	if(rt!=-1){
		dfs02(v[rt].l);
		cout<<v[rt].w<<" ";
		dfs02(v[rt].r);
	}
}
void dfs03(int rt){//后序遍历 
	if(rt!=-1){
		dfs03(v[rt].l);
		dfs03(v[rt].r);
		cout<<v[rt].w<<" ";
	}
}
void print(){//打印结点之间的关系 
	for(int i=0;i<n;i++){
		printf("key:%d ",i);
		printf("w:%d ",v[i].w);
		printf("left:%d right:%d  ",v[i].l,v[i].r);
		printf("parents:%d \n",v[i].parents);
		cout<<"-----------------"<<endl;
	}	
//	cout<<"\n=======================\n"<<endl;
	for(int i=n;i<2*n-1;i++){
		printf("key:%d ",i);
		printf("w:%d ",v[i].w);
		printf("left:%d right:%d  ",v[i].l,v[i].r);
		printf("parents:%d \n",v[i].parents);
		cout<<"-----------------"<<endl;
	}	
}

哈夫曼树构造过程

代码执行结果