题目地址:
https://leetcode.com/problems/patching-array/
给定一个排好序的数组 A A A和一个正整数 n n n,要求给 A A A添加若干数使得 [ 1 , n ] [1,n] [1,n]内的数都能写成 A A A里若干数的和( A A A的每个数用完了就不能继续用了)。问至少要添加多少个数。
我们把什么数都不选也作为一种选择,那么 0 0 0是一定能凑出来的,即 [ 0 , 1 ) [0,1) [0,1)一定能凑出来。假设 [ 0 , x ) [0,x) [0,x)已经能被集合 S S S凑出来,接下来加入了一个数 a a a,那么如果加上这个 a a a, [ a , a + x ) [a,a+x) [a,a+x)这个范围肯定是能凑出来的;如果 a ≤ x a\le x a≤x,那么 S ∪ { a } S\cup\{a\} S∪{a}能凑出 [ 0 , a + x ) [0,a+x) [0,a+x)可以更新右端点;如果 a > x a>x a>x,那么能凑出的范围是 [ 0 , x ) ∪ [ a , a + x ) [0,x)\cup [a,a+x) [0,x)∪[a,a+x),其中 x x x凑不出来,并且如果继续加入大于等于 a a a的数的话, x x x也是不可能凑的出来的,所以此时必须得加入一个能把 x x x凑出来的数,显然加入 x x x最好,可以使得能被凑出来的区间最长。加入 x x x之后能凑出的范围是 [ 0 , x ) ∪ [ x , 2 x ) = [ 0 , 2 x ) [0,x)\cup [x,2x)=[0,2x) [0,x)∪[x,2x)=[0,2x),即右端点延伸到 2 x 2x 2x。算法如下,先设初始区间是 [ 0 , 1 ) [0,1) [0,1),然后遍历 A A A,设当前能凑出的区间是 [ 0 , x ) [0,x) [0,x),如果加入 A [ i ] A[i] A[i]可以使得能凑出的区间还是个完整的区间,则加入之并得出新的右端点;否则,说明必须得加入 x x x,也更新右端点。直到能把 n n n覆盖到,循环结束。代码如下:
public class Solution {
public int minPatches(int[] nums, int n) {
long x = 1;
int res = 0;
for (int i = 0; x <= n; ) {
// 如果nums[i]可以拓展右端点,并且维持区间的完整性,则选取之,并更新右端点
if (i < nums.length && nums[i] <= x) {
x += nums[i++];
} else {
// 否则说明需要添加新数,加入x并更新右端点,计数加1
x += x;
res++;
}
}
return res;
}
}
时间复杂度 O ( l A + log n ) O(l_A+\log n) O(lA+logn),空间 O ( 1 ) O(1) O(1)。