蓝桥杯 小明的衣服 C语言

题目描述

小明买了 �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;
}