2811. 判断是否能拆分数组
https://leetcode.cn/problems/check-if-it-is-possible-to-split-array/
Solution
思维技巧
判断是否存在 长度为 2 的子数组和 >= m
,然后就可以像剥洋葱一样,每次脱离一个长度为 1 的子数组了。特判 原数组长度 <= 2
的时候可以直接拆成两个长度为 1 的子数组。
class Solution:
def canSplitArray(self, nums: List[int], m: int) -> bool:
return len(nums) <= 2 or any(x + y >= m for x, y in pairwise(nums))
2808. 使循环数组所有元素相等的最少秒数
https://leetcode.cn/problems/minimum-seconds-to-equalize-a-circular-array/
Solution
枚举
枚举最后要变成的元素,考虑两个最近的相同数字下标 i, j
,把 i
到 j
都变成 nums[i]
的操作次数为 (j - i) // 2
class Solution:
def minimumSeconds(self, nums: List[int]) -> int:
pos = defaultdict(list)
for i, x in enumerate(nums):
pos[x].append(i)
ans = n = len(nums)
for a in pos.values():
a.append(a[0] + n) # 将第一个下标加入数组
mx = max((j - i) // 2 for i, j in pairwise(a))
ans = min(ans, mx)
return ans