LeetCode 1121. 将数组分成几个递增序列
给你一个 非递减 的正整数数组 nums
和整数 K
,判断该数组是否可以被分成一个或几个 长度至少 为 K
的 不相交的递增子序列。
示例 1:
输入:nums = [1,2,2,3,3,4,4], K = 3
输出:true
解释:
该数组可以分成两个子序列 [1,2,3,4] 和 [2,3,4],每个子序列的长度都至少是 3。
示例 2:
输入:nums = [5,6,6,7,8], K = 3
输出:false
解释:
没有办法根据条件来划分数组。
提示:
1 <= nums.length <= 10^5
1 <= K <= nums.length
1 <= nums[i] <= 10^5
所给数组已是有序数组,要将数组分成长度至少为k的n个数组,只需考虑数组中出现次数最多的数的次数cnt乘K后是否小于等于数组的长度即可。(因为每个子序列中每个数都要求不相同,因此出现最多的数就要求被分到各个子序列中,那么就可以假设每个子序列的长度都为K,一共有cnt个数组,若其相乘后大于数组长度,那一定是不可划分的)
class Solution:
def canDivideIntoSubsequences(self, nums: List[int], K: int) -> bool:
from collections import Counter
l_num = len(nums)
if l_num < K:
return False
cnt = 0
c_nums = Counter(nums)
nums = list(set(nums))
for num in nums:
if c_nums[num] > cnt:
cnt = c_nums[num]
if cnt * K <= l_num:
return True
else:
return False