题目描述
在很久很久以前,有 �n 个部落居住在平原上,依次编号为 11 到 �n。第 �i 个部落的人数为 ��ti。
有一年发生了灾荒。年轻的政治家小蓝想要说服所有部落一同应对灾荒,他能通过谈判来说服部落进行联合。
每次谈判,小蓝只能邀请两个部落参加,花费的金币数量为两个部落的人数之和,谈判的效果是两个部落联合成一个部落(人数为原来两个部落的人数之和)。
输入描述
输入的第一行包含一个整数 �n,表示部落的数量。
第二行包含 �n 个正整数,依次表示每个部落的人数。
其中,1≤�≤1000,1≤��≤1041≤n≤1000,1≤ti≤104。
输出描述
输出一个整数,表示最小花费。
输入输出样例
示例 1
输入
4
9 1 3 5
输出
31
运行限制
- 最大运行时间:1s
- 最大运行内存: 128M
解题思路
按从小到大排序,首先从数组取出最小的两个部落合并,合并后按照部落规模从小到大的顺序再插入数组,用c语言模仿c++的优先队列解决哈夫曼树问题
代码
#include <stdio.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <math.h>
#define LL long long
#define MAX(A, B) (((A) > (B))? (A): (B))
LL n, t[1005];
int cmp(const void* p, const void* q);
void swap(LL* a, LL* b);
int main(int argc, char* argv[]) {
LL i, h, r, ans = 0;
scanf("%lld", &n);
for (i = 0; i < n; i++) scanf("%lld", &t[i]);
qsort(t, n, sizeof(t[0]), cmp);
h = 0; r = n - 1;
while (h < r) {
t[h + 1] += t[h];
ans += t[h + 1];
h++;
for (i = h; t[i] > t[i + 1] && i < r; i++) swap(&t[i], &t[i+1]);
}
printf("%lld\n", ans);
return 0;
}
int cmp(const void* p, const void* q) {
int* a = (int*)p;
int* b = (int*)q;
if (*a > *b) return 1;
else return -1;
}
void swap(LL* a, LL* b) {
LL t;
t = *a;
*a = *b;
*b = t;
return;
}