题目地址:
https://leetcode.com/problems/minimum-replacements-to-sort-the-array/description/
给定一个长 n n n数组 a a a,每一步允许将其中某个数 a [ i ] a[i] a[i]替换为两个数,且这两个数之和要等于 a [ i ] a[i] a[i](对于新产生的数也可以接着替换)。问最少要做多少次操作可以使得整个数组是非严格递增的。
我们从后向前考虑数组,如果第一次发现 a [ i ] > a [ i + 1 ] a[i]>a[i+1] a[i]>a[i+1]了,我们才需要考虑做操作。如果不然,设某另一个最优方案里操作的最右边的数不是 i i i,但是 a [ i ] a[i] a[i]又必须被操作,所以这个最优方案操作的数就是 a [ j ] , j > i a[j],j>i a[j],j>i,操作会将 a [ j ] a[j] a[j]变成若干更小的数,这个时候原方案肯定比这个最优方案更优(因为它限制少)。接下来考虑要将 a [ i ] a[i] a[i]怎么做拆分。拆分必须满足 a [ i ] = b 1 + b 2 + . . . + b k , b 1 ≤ b 2 ≤ . . . ≤ b k ≤ a [ i + 1 ] a[i]=b_1+b_2+...+b_k,b_1\le b_2\le ...\le b_k\le a[i+1] a[i]=b1+b2+...+bk,b1≤b2≤...≤bk≤a[i+1] k − 1 k-1 k−1就是拆分的次数。为了使得后面继续拆分限制尽量少,我们希望 b 1 b_1 b1尽可能的大,且希望 k k k尽可能小。容易证明 b 1 ≤ ⌊ a [ i ] k ⌋ b_1\le \lfloor \frac{a[i]}{k}\rfloor b1≤⌊ka[i]⌋且这个值是能取到的,而直接让 k k k尽可能小就能同时达到 b 1 b_1 b1尽可能大的目的。因为 b k ≤ a [ i + 1 ] b_k\le a[i+1] bk≤a[i+1],所以 a [ i ] ≤ k b k ≤ k a [ i + 1 ] a[i]\le kb_k\le ka[i+1] a[i]≤kbk≤ka[i+1],所以 k ≥ ⌈ a [ i ] a [ i + 1 ] ⌉ k\ge \lceil\frac{a[i]}{a[i+1]}\rceil k≥⌈a[i+1]a[i]⌉。考虑等号能否取到。
如果 a [ i + 1 ] ∣ a [ i ] a[i+1]|a[i] a[i+1]∣a[i],那么 k = a [ i ] a [ i + 1 ] k=\frac{a[i]}{a[i+1]} k=a[i+1]a[i]可以成立,此时 b 1 = b 2 = . . . = a [ i + 1 ] b_1=b_2=...=a[i+1] b1=b2=...=a[i+1]就行了。
如果不能整除, k = ⌈ a [ i ] a [ i + 1 ] ⌉ k= \lceil\frac{a[i]}{a[i+1]}\rceil k=⌈a[i+1]a[i]⌉也可以成立,只需要 b 1 , b 2 , . . . b_1,b_2,... b1,b2,...形如 ⌊ a [ i ] k ⌋ , ⌊ a [ i ] k ⌋ , . . . , ⌊ a [ i ] k ⌋ + 1 , . . . \lfloor\frac{a[i]}{k}\rfloor,\lfloor\frac{a[i]}{k}\rfloor,...,\lfloor\frac{a[i]}{k}\rfloor+1,... ⌊ka[i]⌋,⌊ka[i]⌋,...,⌊ka[i]⌋+1,...即可。代码如下:
class Solution {
public:
using ll = long long;
ll minimumReplacement(vector<int>& a) {
int n = a.size();
ll res = 0;
for (int i = n - 2; i >= 0; i--)
if (a[i] > a[i + 1]) {
int k = (a[i] + a[i + 1] - 1) / a[i + 1];
res += k - 1;
a[i] /= k;
}
return res;
}
};
时间复杂度 O ( n ) O(n) O(n),空间 O ( 1 ) O(1) O(1)。