老晨Leetcode 659 分割数组为连续子序列 · Split Array into Consecutive Subsequences

给定一个整数数组 nums. 现在需要将 nums 拆分成若干个 (至少1个) 子序列, 并且每个子序列至少包含 3 个连续的整数.
返回是否能做到这样的拆分.

Example 2:

Input: [1,2,3,3,4,4,5,5]
Output: True
Explanation:
You can split them into two consecutive subsequences :
1, 2, 3, 4, 5
3, 4, 5

Example 3:

Input: [1,2,3,4,4,5]
Output: False
本题贪心的思路不是那么容易想到
我们首先先把每个数出现的次数存起来放在 count map里
然后把已经成形的sequence (>=3)存起来 (只需要存最后可以放在结尾的数,比如说已经有123了,下一个我可以放4,那就把4作为key)
有可能是112233, 可以放两个4, 所以tail map存的是 :
key: 下一个能放哪个数
value: 能放几个这个数

遍历每个数的时候,
1.如果这个数已经用完了,就跳过
2.如果可以加到已经成型的sequence,就加上
3.如果加不进去,再试着以这个数为开头,把n+1, n+2也都找到,组成一个序列
(每次组序列的时候,都要满足3个才行)
4.都不满足,就return false

public boolean isPossible(int[] a) {
        Map<Integer, Integer> count = new HashMap();// num -> count
        Map<Integer, Integer> tail = new HashMap();// tail -> how many this tail are needed
        for(int i = 0; i < a.length; i++) {
            int cur = count.getOrDefault(a[i], 0);
            count.put(a[i], cur+1);
        } 
        for(int i = 0; i < a.length; i++) {
            if(count.get(a[i]) <= 0) continue;
            if(tail.getOrDefault(a[i], 0) > 0) {
                tail.put(a[i], tail.get(a[i]) - 1);
                tail.put(a[i]+1, tail.getOrDefault(a[i]+1,0) +1);                
            } else if(count.getOrDefault(a[i]+1,0) > 0 && count.getOrDefault(a[i]+2,0) >0) {
                count.put(a[i]+1, count.get(a[i]+1) - 1);
                count.put(a[i]+2, count.get(a[i]+2) - 1);
                tail.put(a[i]+3, tail.getOrDefault(a[i]+3,0) + 1);
            } else {
                return false;
            }
            count.put(a[i], count.get(a[i]) - 1);
        }
        return true;
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
题目描述:给定一个非负整数数组nums和一个整数m,你需要将这个数组分成m个非空的连续数组。设计一个算法使得这m个子数组的最大和最小。 解题思路: 这是一个典型的二分搜索题目,可以使用二分查找来解决。 1. 首先确定二分的左右边界。左边界为数组最大的值,右边界为数组所有元素之和。 2. 在二分搜索的过程,计算出分割数组的组数count,需要使用当前的间值来进行判断。若当前的间值不够分割成m个子数组,则说明mid值偏小,将左边界更新为mid+1;否则,说明mid值偏大,将右边界更新为mid。 3. 当左边界小于等于右边界时,循环终止,此时的左边界即为所求的结果。 具体步骤: 1. 遍历数组,找到数组的最大值,并计算数组的总和。 2. 利用二分查找搜索左右边界,从左边界到右边界间的值为mid。 3. 判断当前的mid值是否满足题目要求,若满足则更新右边界为mid-1; 4. 否则,更新左边界为mid+1。 5. 当左边界大于右边界时,循环终止,返回左边界即为所求的结果。 代码实现: ```python class Solution: def splitArray(self, nums: List[int], m: int) -> int: left = max(nums) right = sum(nums) while left <= right: mid = (left + right) // 2 count = 1 total = 0 for num in nums: total += num if total > mid: total = num count += 1 if count > m: left = mid + 1 else: right = mid - 1 return left ``` 时间复杂度分析:二分搜索的时间复杂度为O(logN),其N为数组的总和,而遍历数组的时间复杂度为O(N),因此总的时间复杂度为O(NlogN)。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值