给你一个整数数组 nums 和一个正整数 k,请你判断是否可以把这个数组划分成一些由 k 个连续数字组成的集合。
如果可以,请返回 true;否则,返回 false。
示例 1:
输入:nums = [1,2,3,3,4,4,5,6], k = 4
输出:true
解释:数组可以分成 [1,2,3,4] 和 [3,4,5,6]。
示例 2:
输入:nums = [3,2,1,2,3,4,3,4,5,9,10,11], k = 3
输出:true
解释:数组可以分成 [1,2,3] , [2,3,4] , [3,4,5] 和 [9,10,11]。
示例 3:
输入:nums = [3,3,2,2,1,1], k = 3
输出:true
示例 4:
输入:nums = [1,2,3,4], k = 3
输出:false
解释:数组不能分成几个大小为 3 的子数组。
提示:
- 1 <= k <= nums.length <= 10^5
- 1 <= nums[i] <= 10^9
主要思路:二分法+依次判断连续
Code:
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0;
int right = nums.size() - 1; // 定义target在左闭右闭的区间里,[left, right]
while (left <= right) { // 当left==right,区间[left, right]依然有效,所以用 <=
int middle = left + ((right - left) / 2);// 防止溢出 等同于(left + right)/2
if (nums[middle] > target) {
right = middle - 1; // target 在左区间,所以[left, middle - 1]
} else if (nums[middle] < target) {
left = middle + 1; // target 在右区间,所以[middle + 1, right]
} else { // nums[middle] == target
return middle; // 数组中找到目标值,直接返回下标
}
}
// 未找到目标值
return -1;
}
bool isPossibleDivide(vector<int>& nums, int k) {
if(nums.size()%k)
return false;
vector<int>(nums.size(), 0);
sort(nums.begin(),nums.end());
while(1)
{
int num=nums[0];
for(int i=0;i<k-1;i++)
{
// if(find(nums.begin(),nums.end(),num+1)==nums.end())
if(search(nums,num+1)==-1)
return false;
num++;
}
for(int i=0;i<k;i++)
{
// nums.erase(find(nums.begin(),nums.end(),num--));
nums.erase(nums.begin()+search(nums,num--));
}
if(nums.size()==0)
break;
}
return true;
}
};