蓝桥杯————平均（贪心算法）

题目描述

有一个长度为 n 的数组（n 是 10 的倍数），每个数 ai​ 都是区间 [0,9] 中的整数。小明发现数组里每种数出现的次数不太平均，而更改第 i 个数的代价为 bi​，他想更改若干个数的值使得这 10种数出现的次数相等（都等于 n/10），请问代价和最少为多少。

思路

主要运用贪心算法的解决方案，这边大概说下贪心算法思路(转载)：

贪心算法是指在对问题求解时，总是做出在当前看来是最好的选择。也就是说，不从整体最优上加以考虑，只做出在某种意义上的局部最优解。贪心算法不是对所有问题都能得到整体最优解，关键是贪心策略的选择，选择的贪心策略必须具备无后效性，即某个状态以前的过程不会影响以后的状态，只与当前状态有关。
解题的一般步骤是：
1.建立数学模型来描述问题；
2.把求解的问题分成若干个子问题；
3.对每一子问题求解，得到子问题的局部最优解；
4.把子问题的局部最优解合成原来问题的一个解。                    
tips:原文链接：https://blog.csdn.net/qq_32400847/article/details/51336300

那么对于本题来说，就是要让所有存在的整数呢，出现的次数都相同，主要思路就是利用二维数组来存放ai 和 bi,再定义一个字典来存放数组中ai的数量，例如a1有3个，a2有2个这样，然后再用这个数量和每个数字应该出现的次数进行比较，定义一个num，num = n//10，然后再取出最小的a即可。文字描述可能较为抽象，下面放代码：

代码部分

n = int(input())

lst = []

num = n / 10

dic = {}

for i in range(n):

    a, b = map(int, input().split())

    lst.append((a, b))

    if a in dic.keys():#记录a的数量

        dic[a] += 1

    else:

        dic[a] = 1

lst = sorted(lst, key=lambda x: x[1]) #以代价大小进行排序

value = 0

for i in range(n):

    a, b = lst[i]#取出最小代价的a

    if dic.get(a, 0) > num: #判断a的数量是否大于num

        value += b    

        dic[a]-=1  #计算完a的数量要减1

print(value)