题目链接
难度:中等 类型:数组、滑动窗口
在一个长度无限的数轴上,第 i 颗石子的位置为 stones[i]。如果一颗石子的位置最小/最大,那么该石子被称作端点石子。
每个回合,你可以将一颗端点石子拿起并移动到一个未占用的位置,使得该石子不再是一颗端点石子。
值得注意的是,如果石子像 stones = [1,2,5] 这样,你将无法移动位于位置 5 的端点石子,因为无论将它移动到任何位置(例如 0 或 3),该石子都仍然会是端点石子。
当你无法进行任何移动时,即,这些石子的位置连续时,游戏结束。
要使游戏结束,你可以执行的最小和最大移动次数分别是多少? 以长度为 2 的数组形式返回答案:answer = [minimum_moves, maximum_moves] 。
示例1
输入:[7,4,9]
输出:[1,2]
解释:
我们可以移动一次,4 -> 8,游戏结束。
或者,我们可以移动两次 9 -> 5,4 -> 6,游戏结束。
示例2
输入:[6,5,4,3,10]
输出:[2,3]
解释:
我们可以移动 3 -> 8,接着是 10 -> 7,游戏结束。
或者,我们可以移动 3 -> 7, 4 -> 8, 5 -> 9,游戏结束。
注意,我们无法进行 10 -> 2 这样的移动来结束游戏,因为这是不合要求的移动。
示例3
输入:[100,101,104,102,103]
输出:[0,0]
解题思路
1.排序
2.最大值:max(移到最左,移到最右)
3.最小值:
- 连续值a<总长度n,外面有几个值就移几步,所有可能去最小
- 连续值a == n-1, 移动2步即可
代码实现
class Solution(object):
def numMovesStonesII(self, stones):
"""
:type stones: List[int]
:rtype: List[int]
"""
stones.sort()
i, n = 0, len(stones)
min_moves = n
max_moves = max(stones[-1]-stones[1], stones[-2]-stones[0])-n+2
for j in range(n):
while stones[j]-stones[i] >= n:
i += 1
if j-i+1 == n-1 and stones[j]-stones[i]==n-2:
min_moves = min(2, min_moves)
else:
min_moves = min(min_moves, n-(j-i+1))
return [min_moves, max_moves]