搬水果(九度oj)
前言
这道题耗时将近半个月,期间我复习了几处基础知识
- 贪心算法
- 堆排序
- 哈夫曼树
最后在参考我同学的博客,终于通过最小堆构建最小优先级队列ac了这道题!
推荐一下我同学的博客,内容很好而且人也很犀利 : 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; }