[leetcode] 330. Patching Array

Given a sorted positive integer array nums and an integer n, add/patch elements to the array such that any number in range [1, n]inclusive can be formed by the sum of some elements in the array. Return the minimum number of patches required.

Example 1:
nums = [1, 3]n = 6
Return 1.

Combinations of nums are [1], [3], [1,3], which form possible sums of: 1, 3, 4.
Now if we add/patch 2 to nums, the combinations are: [1], [2], [3], [1,3], [2,3], [1,2,3].
Possible sums are 1, 2, 3, 4, 5, 6, which now covers the range [1, 6].
So we only need 1 patch.

Example 2:
nums = [1, 5, 10]n = 20
Return 2.
The two patches can be [2, 4].

Example 3:
nums = [1, 2, 2]n = 5
Return 0.

这道题是求用数组中元素组合出n范围内的所有数字所需添加的最少元素,题目难度为Hard。

我们不妨拿一个具体例子来理顺题目思路,假定nums = [1, 2, 3, 8, 24], n = 50。我们需要逐个查看[1, 50]范围内的数字能否通过组合nums中元素来构成,用cur表示当前检查到的数字。容易看出cur = 1~6时,[1, 6]范围内的数字能够通过组合nums前三个元素来构成。cur = 7时,不能通过组合nums中前三个元素构成,而下一个待使用的元素8又太大了,不能通过组合nums中元素得出7,因此需要在nums中添加元素,可选的元素为1~7,如何选择添加的元素是本题的关键,如果我们选择1,则在前三个元素基础上可组合出的范围扩大到[1, 7],选2的话扩大到[1, 8],其他同理,如果选7则范围扩大到[1, 13],可以看出选择的数字越大可以扩展的范围也越大,有利于后续数字的组合,减少在nums中添加元素的可能性,因此这种情况下我们选择在nums中加入7,通过这种贪婪策略即可保证给nums添加的元素最少。在nums中加入7之后,当前能够组合出的数字范围是[1, 13],cur = 14,此时nums中下一个待使用元素是8,cur > 8,通过加入8能够组合出14,因此不必添加元素,加入8之后又能将组合范围扩大到[1, 21]。此时cur = 22,nums中下一个待使用元素是24,cur < 24,回到了和cur = 7时同样的问题,在nums中加入22,组合范围扩大到[1, 43]。此时cur = 44, nums中下一个待使用元素是24,加入24可以将组合范围扩大到[1, 67],因此n = 50时只需要在nums中加入7和22即可,返回2。

拿idx表示nums数组中当前使用到的元素位置,如果当前检查的数字cur < nums[idx],表明需要在nums中添加元素,此时将cur加入nums中可将cur扩大到cur*2的位置;如果当前检查的数字cur >= nums[idx],此时cur可以通过组合nums中元素来构成,不需要添加元素,在加入nums[idx]之后可以将cur扩大到cur + nums[idx],同时idx向前进1。持续上述检查直到数字n能够通过nums中数字组合得出即可。具体代码:

class Solution {
public:
    int minPatches(vector<int>& nums, int n) {
        int cnt = 0, idx = 0;
        long long cur = 1;
        while(cur <= n) {
            if(idx < nums.size() && cur >= nums[idx]) {
                cur += nums[idx++];
            }
            else {
                cur += cur;
                ++cnt;
            }
        }
        return cnt;
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值