题目描述
有一个长度为 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)