题目描述
小明买了 �n 件白色的衣服,他觉得所有衣服都是一种颜色太单调,希望对这些衣服进行染色,每次染色时,他会将某种颜色的所有衣服寄去染色厂,第 �i 件衣服的邮费为 ��ai 元,染色厂会按照小明的要求将其中一部分衣服染成同一种任意的颜色,之后将衣服寄给小明, 请问小明要将 �n 件衣服染成不同颜色的最小代价是多少?
输入描述
第一行为一个整数 �n ,表示衣服的数量。
第二行包括 �n 个整数 �1,�2...��a1,a2...an 表示第 �i 件衣服的邮费为 ��ai 元。
(1≤�≤105,1≤��≤1091≤n≤105,1≤ai≤109 )
输出描述
输出一个整数表示小明所要花费的最小代价。
输入输出样例
示例 1
输入
5
5 1 3 2 1
输出
25
运行限制
最大运行时间:1s
最大运行内存: 256M
思路
正难则反 将不同颜色的衣服染成白色 这里用c语言写c++的优先队列 来模仿哈夫曼树解决问题
#include <stdio.h>
#include <stdlib.h>
#define LL long long
LL cost[100005];
int cmp(const void* p, const void* q);
LL BinSearch_low(LL low, LL high, LL key);
int main(int argc, char* argv[]) {
LL n, sum = 0, i, j, k, l, r;
scanf("%lld", &n);
for (i = 0; i < n; i++) scanf("%lld", &cost[i]);
qsort(cost, n, sizeof(cost[0]), cmp);
l = 0; r = n - 1;
while (l < r) {
sum += cost[l] + cost[l + 1];
cost[l + 1] += cost[l];
l++;
i = BinSearch_low(l, n-1, cost[l]);
k = cost[l];
for (j = l; j < i; j++) cost[j] = cost[j + 1];
cost[i] = k;
}
printf("%lld\n", sum);
return 0;
}
int cmp(const void* p, const void* q) {
LL* a = (LL*)p;
LL* b = (LL*)q;
if (*a > *b) return 1;
else return -1;
}
LL BinSearch_low(LL low, LL high, LL key) {
LL mid;
while (low <= high) {
mid = (low + high) / 2;
if (cost[mid] >= key) high = mid - 1;
else low = mid + 1;
}
return high;
}