详细思路在下面代码,核心思路:
由于是最优的方案,所以先将列表进行从小到大排序,
1. 选择一个大于 0 的整数,将它减去 1;
2. 选择连续 K 个大于 0 的整数,将它们各减去 1。
首先肯定选择2方案较快:选择连续 K 个大于 0 的整数,将它们各减去 1
如样例:[1,2,3,4] -->[0,1,3,4] -->[0,0,2,4]-->[0,0,1,3]-->[0,0,0,2]
由于经过了排序,所以只需要最后再判断一下最后一位数,是否有剩余就行了,
即选择1方案:选择一个大于 0 的整数,将它减去 1;
直接加上2次
# 蓝桥杯2022年第十三届省赛真题-最优清零方案
n,k = map(int,input().split())
a = list(map(int,input().split()))
# 样例输入
# 4 2
# 1 2 3 4
# 样例输出
# 6
a.sort(reverse=False)#从小到大排序,[1,2,3,4]
count = 0#用来计算次数
#默认用第一种
for i in range(0,len(a)-k+1):
while 0 not in a[i:i+k]:
#进行一次连续k的整数减1
for j in range(i,i+k):
a[j] -= 1
#print(a)
#计数加一
count += 1
count += a[len(a)-1]#因为是排序后,且两个两个计算故最后可能存在没有算尽的最大值,再加起来
print(count)
# 样例输入
# 4 2
# 1 2 3 4
# 样例输出
# 6
运行: