小知识点:
list.index(x,start,end),在列表的start到end的范围内查找x第一次出现的序列
题目:
给定一个长度为 N 的数列 A1, A2, · · · , AN。现在小蓝想通过若干次操作将这个数列中每个数字清零。
每次操作小蓝可以选择以下两种之一:
1. 选择一个大于 0 的整数,将它减去 1;
2. 选择连续 K 个大于 0 的整数,将它们各减去 1。
小蓝最少经过几次操作可以将整个数列清零?
输入格式
输入第一行包含两个整数 N 和 K。
第二行包含 N 个整数 A1, A2, · · · , AN。
输出格式
输出一个整数表示答案。
(1)代码:
n, k = map(int, input().split()) nums = list(map(int, input().split())) i = 0 ans = 0 while i < n: if i > n - k: # 当接近数组末尾时,特别处理 break min_value = min(nums[i:i+k]) min_idx = nums.index(min_value, i, i+k) for x in range(i, min(n, i+k)): nums[x] -= min_value i = min_idx + 1 ans += min_value # 处理数组末尾的元素 for x in range(n): ans += nums[x] print(ans) (2)解析:
对本题的思路为先设定一个开始的序列i,将i赋值为0,然后将前k个元素依次减去前k个元素中的最小值,再用index函数找出最小值的序列,此时最小值相应序列上的值为0,此时再将开始的序列变成0的序列,接着循环,最后当剩余元素不足k个,跳出循环,进入单独处理阶段。
单独阶段:
此时的列表已经为经过减法无法再减的列表,此时再将每个不为0的数加到ans上,就可以得到相应的操作次数
(3)注意事项:
在指定范围内([i, i+k)
)查找 min_value
第一次出现的索引。
**index()
函数的参数详解**
python
复制
nums.index(min_value, start, end)
参数 | 说明 |
---|---|
min_value |
要查找的值(这里是当前窗口的最小值)。 |
start |
查找的起始索引(包含)。 |
end |
查找的结束索引(不包含)。如果省略,默认查找到列表末尾。 |
注意:
- 如果找不到
min_value
,会抛出ValueError
。 - 在本题中,
min_value
一定存在于nums[i:i+k]
中(因为它是通过min(nums[i:i+k])
计算得到的),所以不会报错。