目录
一、题目描述
给你一个按升序排序的整数数组 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]
二、解题思路
唉,这题一开始没什么思路…一开始着眼去找数字出现次数的规律,发现无迹可寻,后面没在深想了,要是再想想数字必须连续,基本上就出来了。
两种方法:
- 贪心地去构建序列,贪心体现在添加数字时都往已存在的子序列后面添加,因为如果重新建一个序列的话有可能后面没有满足长度要求的数字了;如果没有可添加的子序列,那么必须新建,而且新建的长度必须等于3,如果没有足够的数字则说明不能成功分割数组。
- 哈希+小顶堆(哈希的key表示当前序列的结尾为key,value为由以key为结尾的所有子序列的长度,因为以key为结尾的子序列可能不止一个)记录所有的序列,再看所构建的序列是否满足要求。
三、代码实现
#include <bits/stdc++.h>
using namespace std;
//哈希+小顶堆(哈希的key表示当前序列的结尾为key,value为由以key为结尾的所有子序列的长度,因为以key为结尾的子序列可能不止一个)
bool isPossible(vector<int>& nums) {
unordered_map<int, priority_queue<int, vector<int>, greater<int>>> mp;
int n = nums.size();
for (int i = 0; i < n; ++i) {
//如果不存在以nums[i] - 1结尾的序列则开始一个长度为1,以nums[i]结尾的序列
if (mp.find(nums[i]) == mp.end()) {
//先定义,后面会用到
mp[nums[i]] = priority_queue<int, vector<int>, greater<int>>();
}
if (mp.find(nums[i] - 1) != mp.end()) {
//要取最短的那条子序列,优先加在这一条后面
int preLength = mp[nums[i] - 1].top();
mp[nums[i] - 1].pop();
if (mp[nums[i] - 1].empty()) {
mp.erase(nums[i] - 1);
}
mp[nums[i]].push(preLength + 1);
} else {
mp[nums[i]].push(1);
}
}
for (auto ite = mp.begin(); ite != mp.end(); ++ite) {
if (ite->second.top() < 3) {
return false;
}
}
return true;
}
//贪心策略:每次都把当前数加在最短的那条序列(如果存在可以加的序列的话)
bool isPossible1(vector<int>& nums) {
unordered_map<int, int> numCnt, tail;
for (auto &x : nums) {
numCnt[x]++;
}
for (auto &x : nums) {
int cnt = numCnt[x];
if (cnt > 0) {
//如果存在以x - 1结尾的序列,就直接接在后面
if (tail[x - 1] > 0) {
numCnt[x]--;
tail[x - 1]--;
tail[x]++;
} else {
//如果不存在就需要以x结尾开始一个长度为3(保证满足题目长度要求)的新序列
if (numCnt[x + 1] > 0 && numCnt[x + 2] > 0) {
numCnt[x]--;
numCnt[x + 1]--;
numCnt[x + 2]--;
tail[x + 2]++;
} else {
return false;
}
}
}
}
return true;
}
int main() {
vector<int> arr = { 1,2,3,3,4,5 };
cout << isPossible1(arr);
return 0;
}