HDU 2527

题目描述

         HDU 2527

分析

        霍夫曼编码的应用。

        本题没有必要构造一棵完整的霍夫曼树。只需利用霍夫曼编码的原理，每次挑选频率最低的两个元素进行合并。（显然，可以利用优先级队列，这里用数组来模拟）

源码

//每次挑出现频率最小的两个元素，应该用优先级队列！！！！！！！！！！

#include <stdio.h>
#include <limits.h>
#include <string.h>
int unionFre(int fre[26]);

int main()
{
	int frequency[26];    //记录每个字母出现的频率
	int n;
	int m;
	char input[10000];
	int letterNum;     //出现的小写字母的个数
	int len;
	int result;
	int i, j;

	scanf("%d", &n);
	while (n --)
	{
		scanf("%d", &m);
		scanf("%s", &input);
		//初始化
		memset(frequency, 0, sizeof(frequency));

		//统计每个字母出现的次数
		len = strlen(input);
		for (i = 0; i < len; i ++)
		{
			frequency[input[i]-'a'] ++;
		}
		//没出现的字母频率均设为最大值
		letterNum = 26;
		for (i = 0; i < 26; i ++)
		{
			if (frequency[i] == 0)
			{
				letterNum --;
				frequency[i] = INT_MAX;
			}
		}

		//如果字符串只由一个字母组成
		if (letterNum == 1)
		{
			result = frequency[input[0]-97];
			if (result <= m)
				printf("yes\n");
			else
				printf("no\n");

			continue;
		}

		//循环(letterNum-1)次，每次挑选出现频率最低的两个元素进行合并
		//并更新result值，加上被合并的两个元素的频率和
		result = 0;
		for (i = 0; i < letterNum-1; i ++)
		{
			result = result + unionFre(frequency);
		}

		if (result <= m)
			printf("yes\n");
		else
			printf("no\n");
	}

	return 0;
}

//合并函数
int unionFre(int fre[26])
{
	int min, secondMin;
	int minIndex, secondMinIndex;
	int temp;
	int i;

	min = fre[0];
	minIndex = 0;
	secondMin = fre[1];
	secondMinIndex = 1;

	if (min > secondMin)
	{
		temp = min;
		min = secondMin;
		secondMin = temp;

		temp = minIndex;
		minIndex = secondMinIndex;
		secondMinIndex = temp;
	}

	//找出频率最低的两个元素
	for (i = 2; i < 26; i ++)
	{
		if (fre[i] < min)
		{
			secondMin = min;
			secondMinIndex = minIndex;

			min = fre[i];
			minIndex = i;
		}
		else if (fre[i] < secondMin)
		{
			secondMin = fre[i];
			secondMinIndex = i;
		}
	}

	//合并两个元素
	fre[minIndex] = min + secondMin;
	fre[secondMinIndex] = INT_MAX;

	return (min + secondMin);
}