完美的服务器太垃圾了吧,LGD 加油啊。
题目链接:
分割数组为连续子序列 - 力扣(LeetCode)leetcode-cn.com题目描述:
输入一个按升序排序的整数数组(可能包含重复数字),你需要将它们分割成几个子序列,其中每个子序列至少包含三个连续整数。返回你是否能做出这样的分割?
示例 1:
输入: [1,2,3,3,4,5]
输出: True
解释:
你可以分割出这样两个连续子序列 :
1, 2, 3
3, 4, 5
示例 2:
输入: [1,2,3,3,4,4,5,5]
输出: True
解释:
你可以分割出这样两个连续子序列 :
1, 2, 3, 4, 5
3, 4, 5
示例 3:
输入: [1,2,3,4,4,5]
输出: False
提示:
- 输入的数组长度范围为 [1, 10000]
解题思路:
本题要将原数组分为若干个子序列,每个子序列都至少包含三个连续整数,那我们先观察一下,到底什么策略才能让结果符合它的要求。
以示例 2 为例,[1,2,3,3,4,4,5,5],可以分割成 [1,2,3,4,5] 和 [3,4,5],那是一开始先尽可能按长的分配,再看剩下来的怎么分配么?
如果运用此策略对付示例 1,即 [1,2,3,3,4,5],我们就会得到 [1,2,3,4,5] 和 [1],就不能得到 [1,2,3] 和 [3,4,5] 的答案了。所以并不能只采取让现有子序列尽可能长的单一策略,我们还得考虑能不能和后面的数组成新的子序列。
于是乎,我们设置两个 map,一个叫 numsCntMap,numsCntMap[num] 用来表示剩余 num 的个数,一个叫numsEndCntMap,numsEndCntMap[num] 表示以 num 结尾的连续子序列个数。
对每一个 nums 中的数 num,如果 numsEndCntMap 中存在以 num-1 结尾的连续超过 3 个的序列,则把 num 接在以 num-1 结尾的序列上。如果不存在,则查找后续是否有 num+1 和 num+2。如果还不存在,就返回 false。(注意,条件顺序不能颠倒,必须先匹配前面的 num-1)。
代码如下:
class Solution {
public:
bool isPossible(vector<int>& nums) {
unordered_map<int, int> numsCntMap; // numsCntMap[num]表示的是num剩余的个数
unordered_map<int, int> numsEndCntMap; // numsEndCntMap[num]表示的是以num结尾的连续子序列(长度不小于3)个数
// 统计各个元素出现的次数
for(auto num : nums){
++numsCntMap[num];
}
for(auto num : nums){
if(numsCntMap[num] == 0){ // 此元素没有剩余,已经被使用完了
continue;
}
--numsCntMap[num]; // 剩余个数自减
if(numsEndCntMap.count(num-1) && numsEndCntMap[num-1]){
// 存在以num - 1结尾的连续子序列(长度不小于3),则将num放入
--numsEndCntMap[num-1]; // 以num - 1结尾的连续子序列(长度不小于3)自减,因为被num放入使用了一个
++numsEndCntMap[num]; // 以num结尾的连续子序列(长度不小于3)自增,刚刚创建了一个
}
else if(numsCntMap.count(num+1) && numsCntMap[num+1] && numsCntMap.count(num+2) && numsCntMap[num+2]){
// 否则查找后面两个元素,凑出一个合法的序列
--numsCntMap[num+1];
--numsCntMap[num+2];
++numsEndCntMap[num+2];
}
else{
// 两种方法都不行,则凑不出,比如[1,2,3,4,4,5]中的第二个4,
return false;
}
}
return true;
}
};
如果有任何疑问,欢迎提出。如果有更好的解法,也欢迎告知。