题目地址
https://leetcode-cn.com/problems/split-array-into-consecutive-subsequences/
题目描述
给你一个按升序排序的整数数组 num(可能包含重复数字),请你将它们分割成一个或多个长度至少为 3 的子序列,其中每个子序列都由连续整数组成。
如果可以完成上述分割,则返回 true ;否则,返回 false 。
示例 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 <= nums.length <= 10000
主要思路
seq用来记录哪些元素可以被接到其他子序列后面。
遍历数组元素,优先判断元素是否能够接到其他序列后面,如果不能,则以自己开头构建一个长度至少为 3 的序列,如果序列也不能构建,则返回false
代码实现
javascript
var isPossible = function(nums) {
let freq = new Map(), seq = new Map();
for (let i of nums) {
freq.has(i) ? freq.set(i, freq.get(i)+1) : freq.set(i, 1);
}
for (let i of nums) {
if (freq.get(i) === 0) continue;
if (seq.has(i) && seq.get(i).length>0) {
freq.set(i, freq.get(i)-1);
let new_seq = seq.get(i).pop();
new_seq.push(i);
// 对seq.get(i+1)得到的数组进行push操作,不用再set
seq.has(i+1) ? seq.get(i+1).push(new_seq) : seq.set(i+1, [new_seq]);
} else if (freq.get(i)>0 && freq.get(i+1)>0 && freq.get(i+2)>0){
freq.set(i, freq.get(i)-1);
freq.set(i+1, freq.get(i+1)-1);
freq.set(i+2, freq.get(i+2)-1);
seq.has(i+3) ? seq.get(i+3).push([i, i+1, i+2]) : seq.set(i+3, [[i, i+1, i+2]]);
} else {
return false;
}
}
return true;
};