给定一个已排序的正整数数组 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] 区间里所有的数。
所以我们最少需要添加一个数字。
示例 2:
输入: nums = [1,5,10], n = 20
输出: 2
解释: 我们需要添加 [2, 4]。
示例 3:
输入: nums = [1,2,2], n = 5
输出: 0
class Solution {
public:
int minPatches(vector<int>& nums, int n) {
}
};
思路:
如果用一个 n 位的列表初始为 1,表示 1-n 都不能被区间中的数表示出来,如果能被表示,对应值记为0(下标0对应1,类推)。循环把区间里的数能表示的值对应的列表都记为0,结束后还为 1 的则确定需要补充区间里的数,从最小的开始补充(因为补充大的较小的就不能被表示),把所有的和与补充的值相加,排除能被表示的值。直至均能被表示,得出结果。
但是求和数量一多也太难了吧,理智告诉我这样不对。
后面又去看题解了,怎么感觉这该是个数学问题。
数组本身有序,设能组合范围 [1,x - 1] 且 x 在数组中,则 x 与 [x+1,2x-1] 也可被覆盖,合并为 [1,2x-1];若 x 缺失,从贪心角度讲,补充 x 就可以保证数组中数字能填满到 2x-1,继续直至填到 n 即可。
class Solution {
public:
int minPatches(vector<int>& nums, int n) {
int patches=0; //打补丁的个数
long long x=1;
int length=nums.size();
int i=0;
while(x<=n){ //假定[1,x-1]均能被组合,判断x能否被组合,循环至能组合成n时跳出循环
if(i<length && nums[i]<=x){ //被组合区域从[1,x-1]扩展到[1,x+nums[i]-1]
x+=nums[i];
++i;
}
else{
x<<=1; //x*=2(x+=x) 即补充x后,可组合区域从[1,x-1]扩展到[1,2x-1]
++patches;
}
}
return patches;
}
};