正数数组的最小不可组成和-进阶问题

正数数组的最小不可组成和-进阶问题

题目描述

给定一个正数数组arr，其中所有的值都为整数，以下是最小不可组成和的概念

• 把arr每个子集内的所有元素加起来会出现很多值，其中最小的记为min，最大的记为max
• 在区间[min, max]上，如果有数不可以被arr某一个子集相加得到，那么其中最小的那个数是arr的最小不可组成和
• 在区间[min, max]上，如果所有的数都可以被arr的某一个子集相加得到，那么max+1是arr的最小不可组成和

请写函数返回正数数组arr的最小不可组成和

保证1一定出现过！

[要求]

时间复杂度为$O(nlogn)$，额外空间复杂度为$O(1)$

输入描述：

第一行一个整数N，表示数组长度。
接下来一行N个整数表示数组内的元素。

输出描述：

输出一个整数表示数组的最小不可组成和

示例1

输入

3
1 2 5

输出

4

说明

3 = 1 + 2

示例2

输入

5
1 2 3 4 5

输出

16

备注：

$1⩽N⩽10^5$
$1⩽ar{r}_i⩽10^9$

---

题解：

相比于上一题 正数数组的最小不可组成和 ，这题多了一个条件：1一定出现。于是此题可以进行优化：

• 将 arr 排序，则 arr[0] == 1
• 从左往右计算每个位置的 i 的 range ，range 表示计算到 arr[i] 时，[1, range] 区间内的所有正数都可以被 arr[0…i-1] 的某一个子集加出来，初始 range = 0
• 若 arr[i] > range + 1，因为 arr[i] 已经排序，后面都不会出现 range + 1，可以直接得到结果 range + 1；若 arr[i] <= range + 1，说明 [1, arr[i] + range] 区间上的所有正数都可以被 arr[0…i] 的一个子集加出来，所以令 range += arr[i]，继续下一个位置

简要证明：

假设当前遍历到位置 i，若要凑出区间 [1, arr[i] + range]，我们现在需要凑出来的是区间 [range+1, range+arr[i]]，假设 arr[0…i-1] 可以凑出 range - x ，则:

• 若 range - x + arr[i] = range + 1，得到 x = arr[i] - 1；

• 若 range - x + arr[i] = range + 2，得到 x = arr[i] - 2；

• …

• 若 range - x + arr[i] = range + arr[i] ，得到 x = 0；

若 arr[i] > range + 1，则 x = arr[i] - 1 > range ，arr[0…i-1] 肯定无法凑出来（最大值为 range） ，而后面不会出现 range + 1 ，所以 range + 1 就是答案。

代码：

#include <cstdio>
#include <algorithm>

using namespace std;

const int N = 100000;

int n;
int a[N];

int main(void) {
    scanf("%d", &n);
    for ( int i = 0; i < n; ++i ) scanf("%d", a + i);
    sort( a, a + n );
    long range = 0;
    for ( int i = 0; i < n; ++i ) {
        if ( a[i] > range + 1 ) break;
        else range += a[i];
    }
    return 0 * printf("%ld\n", range + 1);
}