搬水果(九度oj)

前言

这道题耗时将近半个月,期间我复习了几处基础知识
  • 贪心算法
  • 堆排序
  • 哈夫曼树
最后在参考我同学的博客,终于通过最小堆构建最小优先级队列ac了这道题!

优先级队列参考链接: http://blog.csdn.net/zinss26914/article/details/8614022

推荐一下我同学的博客,内容很好而且人也很犀利 :  http://blog.csdn.net/cscmaker/article/details/8138870

题目

题目描述:
    在一个果园里,小明已经将所有的水果打了下来,并按水果的不同种类分成了若干堆,小明决定把所有的水果合成一堆。每一次合并,小明可以把两堆水果合并到一起,消耗的体力等于两堆水果的重量之和。当然经过 n‐1 次合并之后,就变成一堆了。小明在合并水果时总共消耗的体力等于每次合并所耗体力之和。
    假定每个水果重量都为 1,并且已知水果的种类数和每种水果的数目,你的任务是设计出合并的次序方案,使小明耗费的体力最少,并输出这个最小的体力耗费值。例如有 3 种水果,数目依次为 1,2,9。可以先将 1,2 堆合并,新堆数目为3,耗费体力为 3。然后将新堆与原先的第三堆合并得到新的堆,耗费体力为 12。所以小明总共耗费体力=3+12=15,可以证明 15 为最小的体力耗费值。
输入:
    每组数据输入包括两行,第一行是一个整数 n(1<=n<=10000),表示水果的种类数,如果 n 等于 0 表示输入结束,且不用处理。第二行包含 n 个整数,用空格分隔,第 i 个整数(1<=ai<=1000)是第 i 种水果的数目。
输出:
对于每组输入,输出一个整数并换行,这个值也就是最小的体力耗费值。输入数据保证这个值小于 2^31。
样例输入:
3
9 1 2
0
样例输出:
15


ac代码

#include <stdio.h>
#include <stdlib.h>

#define NUM 10001

void minHeapIfy(int *A, int i, int n);
void buildMinHeap(int *A, int n);
int heapExtractMin(int *A, int n);
void heapIncreaseKey(int *A, int i, int key);
int moveFruit(int *A, int n);

int main()
{
	int i , n, power, weight[NUM];

	while(scanf("%d", &n) != EOF && n != 0)
	{
		//接收客户端参数
		for(i = 1; i <= n; i ++)
		{
			scanf("%d", &weight[i]);
		}

		//贪心选择
		power = moveFruit(weight, n);
		printf("%d\n", power);
	}

	return 0;
}

/**
 * Description:维护以i为根节点的最小堆
 */
void minHeapIfy(int *A, int i, int n)
{
	int min, loc, r, l, change;

	for(min = i; min <= n;)
	{
		l = min * 2;
		r = min * 2 + 1;
		loc = min;

		if(l <= n && A[l] < A[min])
			min = l;
		if(r <= n && A[r] < A[min])
			min = r;
		
		if(loc != min)
		{
			change = A[min];
			A[min] = A[loc];
			A[loc] = change;
		}else
		{
			break;
		}
	}
}

/**
 * Description:去掉并返回堆中最小的元素
 */
int heapExtractMin(int *A, int n)
{
	int max = A[1];

	A[1] = A[n];
	minHeapIfy(A, 1, n - 1);

	return max;
}

/**
 * Description:建立最小堆
 */
void buildMinHeap(int *A, int n)
{
	int i;

	for(i = n / 2; i >= 1; i --)
	{
		minHeapIfy(A, i, n);
	}
}

/**
 * Description:将元素插入到最小优先队列
 */
void heapIncreaseKey(int *A, int i, int key)
{
	int parent, change;

	for(A[i] = key, parent = i / 2; i >= 1 && A[parent] > A[i] && parent >= 1;)
	{
		change = A[parent];
		A[parent] = A[i];
		A[i] = change;
		i = parent;
		parent = i / 2;
	}
}


int moveFruit(int *A, int n)
{
	int power, i, lchild, rchild, parent, j;

	buildMinHeap(A, n);

	for(i = 1, j = n, power = 0; i < n; i ++)
	{
		lchild = heapExtractMin(A, j);
		j -= 1;
		rchild = heapExtractMin(A, j);

		parent = lchild + rchild;
		power += lchild + rchild;
	
		heapIncreaseKey(A, j, parent);	
	}

	return power;
}


你可能感兴趣的:(搬水果(九度oj))