[LeetCode110双周赛&LeetCode357周赛] 2808. 使循环数组所有元素相等的最少秒数,2811. 判断是否能拆分数组

参考灵神直播和代码

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 ,把 ij 都变成 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

你可能感兴趣的:(算法,算法,leetcode)