前言
这道题耗时将近半个月,期间我复习了几处基础知识
最后在参考我同学的博客,终于通过最小堆构建最小优先级队列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;
}