【id:85】【20分】H. DS二叉树--赫夫曼树的构建与编码（含代码框架）

题目描述

给定n个权值，根据这些权值构造huffman树，并进行huffman编码

以下代码框架仅供参考，不强制一定按照下面框架

大家以课本算法为主，注意数组访问是从位置1开始

要求：赫夫曼的构建中，默认左孩子权值不大于右孩子权值

输入

第一行输入t，表示有t个测试实例
第二行先输入n，表示第1个实例有n个权值，接着输入n个权值，权值全是小于1万的正整数
依此类推

输出

逐行输出每个权值对应的编码，格式如下：权值-编码
即每行先输出1个权值，再输出一个短划线，再输出对应编码，接着下一行输入下一个权值和编码。
以此类推

输入输出样例

输入样例1 <-复制
1
5 15 4 4 3 2

输出样例1
15-1
4-010
4-011
3-001
2-000

代码（未完全通过）

#include<iostream>
#include<string>
#include<stack>
using namespace std;

struct huffnode
{
	int weight=0;
	int lchild=0, rchild=0, parent=0;
};

huffnode biao[1000];

void select(int wei)
{
	int i;
	int min = 1000;
	int s1, s2;
	for (i = 1; i <=wei; i++)
	{
		if (biao[i].parent == 0 && biao[i].weight < min)
		{
			min = biao[i].weight;
			s1 = i;
		}
	}
	biao[s1].parent = wei + 1;
	biao[wei + 1].lchild = s1;

	min = 1000;
	for (i = 1; i <=wei; i++)
	{
		if (biao[i].parent == 0 && biao[i].weight < min)
		{
			min = biao[i].weight;
			s2 = i;
		}
	}
	biao[s2].parent = wei + 1;
	biao[wei + 1].rchild = s2;

	biao[wei + 1].weight = biao[s1].weight + biao[s2].weight;
}

void createtree(int ma[],int n)
{
	int i = 1;
	int m = 2 * n - 1;
	for (i = 1; i <= n; i++)
		biao[i].weight = ma[i - 1];

	for (i = n + 1; i <=m; i++)
		select(i - 1);
}

void huffcode(int n)
{
	char code[1000][20];
	int i;
	int shu;
	for (i = 1; i <= n; i++)
	{
		stack<char>s;
		cout << biao[i].weight << "-";
		int j = i;
		while (biao[j].parent != 0)
		{
			int k = biao[j].parent;
			if (biao[k].lchild == j)
			{
				s.push('0');
			}
			else
			{
				s.push('1');
			}
			
			j = k;
		}
		j = 0;
		while (!s.empty())
		{
			code[i][j++] = s.top();
			s.pop();
		}
		code[i][j] = '\0';
		cout << code[i] << endl;
	}
}

int main()
{
	int t;
	cin >> t;
	int ma[1000];
	while (t--)
	{
		int n;
		cin >> n;
		for (int i = 0; i < n; i++)
			cin >> ma[i];
		createtree(ma, n);
		huffcode(n);
	}
}

（by 归忆）