问题描述 :
给你一个按升序排序的整数数组 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]
输入说明 :
首先输入num的长度n,然后输入n个整数
输出说明 :
输出“true”或“false”,不包括引号。
输入范例 :
8
1 2 3 3 4 4 5 5
输出范例 :
true
算法思想:贪心
使用两个map,cnt和tail
cnt[i]:存储的是原数组中数字i出现的次数
tail[i]:存储以数字i结尾的符合题意要求的子序列的个数
1.先去寻找一个长度为3的连续子序列i,i+1,i+2,找到后则cnt[i],cnt[i+1],cnt[i+2]对应的数字消耗1
,并将tail[i+2]加1
2.如果后续发现有能够接在这个连续子序列的数字i+3,则延长以i+2结尾的子序列
即tail[i+2]--,tail[i+3]++
在不满足1,2两点的条件下
3.如果cnt[i] = 0,说明这个数字消耗完,接着遍历下一个数字即可
如果cnt[i] !=0,说明这个数字多出来了,返回false即可
#include<iostream>
#include<vector>
#include<map>
#include<algorithm>
using namespace std;
class Solution {
public:
bool isPossible(vector<int> &nums) {
map<int,int> cnt; //存储数字出现的次数
map<int,int> tail; //存储以i结尾的序列个数
for(auto num : nums) {
cnt[n]++;
}
for(auto num : nums) {
if(cnt[num] == 0) {
continue;
}
else if(cnt[num] > 0 && tail[num-1] > 0) {
tail[num-1]--;
tail[num]++;
cnt[num]--;
}
else if(cnt[num] > 0 && cnt[num+1] >0 &&cnt[num+2] > 0) {
tail[num+2]++;
cnt[num]--;
cnt[num+1]--;
cnt[num+2]--;
}
else {
return false;
}
}
return true;
}
};
int main() {
vector<int> nums;
int n,data;
cin>>n;
for(int i = 0; i < n; i++) {
cin>>data;
nums.push_back(data);
}
Solution s;
bool res = s.isPossible(nums);
if(res==true) {
cout<<"true";
}
else {
cout<<"false";
}
return 0;
}