Leetcode 659. 分割数组为连续子序列
题目
给你一个按升序排序的整数数组 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, 10000]
题解
我们先尽可能地构造长的连续子序列。即如果存在nums[i]-1的连续子序列,我们就可能将nums[i]放到这个连续子序列上。否则,我们需要判断是否存在nums[i]+1以及nums[i]+2,存在,则可以构造连续子序列;否则,则return false。
我们用哈希表time记录数字可使用次数,哈希表tail记录每个连续子序列的最后一个数字。详细过程见代码
代码
bool isPossible(vector<int>& nums) {
unordered_map<int,int> time;
unordered_map<int,int> tail;
int n = nums.size();
for(int i=0; i<n; i++)
time[nums[i]]++;
for(int i=0; i<n; i++){
if(time[nums[i]] == 0) continue; //该数字已全部使用
if(tail[nums[i]-1]!=0){ //当前数字可以和某个连续子序列进行合并
time[nums[i]]--;
tail[nums[i]-1]--;
tail[nums[i]]++;
}else{
if(time[nums[i]+1]!=0 && time[nums[i]+2]!=0){ //可以构造一个新的连续子序列
time[nums[i]]--;
time[nums[i]+1]--;
time[nums[i]+2]--;
tail[nums[i]+2]++;
}else return false;
}
}
return true;
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/split-array-into-consecutive-subsequences
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。