LeetCode 330 按要求补齐数组 题解
给定一个已排序的正整数数组 nums,和一个正整数 n 。从 [1, n] 区间内选取任意个数字补充到 nums 中,使得 [1, n] 区间内的任何数字都可以用 nums 中某几个数字的和来表示。请输出满足上述要求的最少需要补充的数字个数。
示例 1:
输入: nums = [1,3], n = 6
输出: 1
解释:
根据 nums 里现有的组合 [1], [3], [1,3],可以得出 1, 3, 4。
现在如果我们将 2 添加到 nums 中, 组合变为: [1], [2], [3], [1,3], [2,3], [1,2,3]。
其和可以表示数字 1, 2, 3, 4, 5, 6,能够覆盖 [1, 6] 区间里所有的数。
所以我们最少需要添加一个数字。
思路:贪心算法
核心逻辑:
- 值域为
[0,x]
的若干个数加上 k,可扩大值域至[0,x+k]
,注意k∈[0,x+1]
。
证明:值域为[0, x]
,说明可以完全覆盖该区域内所有的连续的整数,都加上k,新出现的值域
为[k,x+k]
,那么必须要k<=x+1
,才可以将这两个值域连起来,形成[0,x+k],
- 将 k 视为 nums已有的数 或者 要插入的数
- 根据题意,nums 必须有 数字1
显然从[0,1]
开始模拟扩展,一般地,我们取 k 最大,也就是k = x+1
假设 nums 空,我们这样模拟[0,1] -> [0,2] -> [0,4] -> [0,8] ----> [0,y],
直到y >= n
- 贪心的地方在于我们取的k是最大能取到的值,也就是k=x+1,这样也是为了满足题目要求的补充的数字个数最少
- 因此当
nums(i) ∈ [0,y+1]
时,可取之为nums(i) 并扩大值域, 否则视为插入一个数 y+1。
class Solution {
public int minPatches(int[] nums, int n) {
if(n == 0){
return 0;
}
int i = 0, res = 0;
//加起来的最大值可能超出int范围(因为y>=n),需要用long
long y = 0;//值域[0,y]
while(y < n){
if(i < nums.length && nums[i] <= y+1){
//值域扩大至[0,nums[i]+y]
y += nums[i];
i++;
}else{
//值域扩大至[0,2y+1]
y += y+1;
res++;
}
}
return res;
}
}