蓝桥杯 谈判 C语言

题目描述

在很久很久以前，有 �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 
#include 
#include 
#include 
#include 
#include 
#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;
}