Acwing第43场周赛B题(两个数列)

4315. 两个数列

题目描述

有两个正整数数列 $a1,a2,\dots ,an$ 和 $b1,b2,\dots ,bn$。

现在，已知的信息有：

1. 数列 $a$ 的各个元素的值。
2. 数列 $b$ 的各个元素之和 $s$。
3. 对于任意的 $1\le i\le n$，满足 $1\le b_i\le a_i$ 成立。

由上述信息，我们可知对于元素 $b_i$，其可能的取值范围为 $[1,a_i]$，但是受到已知条件的约束，它可能无法取到其中一些数值。

我们的任务就是计算每个 $b_i$ 在其可能的取值范围内，无法取到的数值的数量。

例如，如果 n = 2 ， a = { 4 , 4 } ， s = 8 n=2，a=\{4,4\}，s=8 n=2，a={4,4}，s=8，则数列 $b$ 中的每一个元素都不能小于 $4$（否则，另一个元素就要大于 $4$，这是不可能的），也就是说 $b_1$ 和 $b_2$ 在其可能的取值范围 $[1,4]$ 内，均无法取到 $1,2,3$，无法取到的数值的数量均为 $3$。

输入格式

第一行包含两个整数 $n$ 和 $s$。

第二行包含 $n$ 个整数 $a_1,a_2,\dots ,a_n$。

输出格式

共一行，$n$ 个整数，其中第 $i$ 个整数表示 $b_i$ 在其可能的取值范围内，无法取到的数值的数量。

数据范围

前三个测试点满足 $1\le n\le 2$。
所有测试点满足 $1\le n\le 2\times 10^5，n\le s\le {\sum }_{i=1}^n{a}_i，1\le a_i\le 10^6$。

输入样例1：

2 8
4 4

输出样例1：

3 3

输入样例2：

1 3
5

输出样例2：

4

输入样例3：

2 3
2 3

输出样例3：

0 1

解题思路

其实就是解一个不等式，用数学推导公式~
下面我们来推导一下公式！
已知有两个正整数数列：$$a1,a2,\dots ,an$$
$$b1,b2,\dots ,bn$$
$b$的各个元素之和为$S$，即$S=b_1+b_2+...+b_n$;
令$S_i=b_1+b_2+...b_{i-1}+b_{i+1}...+b_n$，$Sum=a_1+a_2+...+a_n$；
又因为，对于任意的 $1\le i\le n$，满足 $1\le b_i\le a_i$ 成立;
则：$n-1\le S_i\le Sum-a_i$；

$$-n+1\ge -S_i\ge -Sum+a_i$$
$$S-n+1\ge S-S_i\ge S-Sum+a_i$$
又因为$S-S_i=b_i$，则有
$$S-n+1\ge b_i\ge S-Sum+a_i$$

最终我们只需要取$1\le b_i\le a_i$集合和$S-n+1\ge b_i\ge S-Sum+a_i$集合的交集即可！~

又因为$n\le S\le Sum$
即$b_i$的取值范围大概在$[n-Sum+a_i,Sum-n+1]$，


由于测试点满足 $1\le n\le 2\times 10^5，n\le s\le {\sum }_{i=1}^n{a}_i，1\le a_i\le 10^6$。


所以集合$S-n+1\ge b_i\ge S-Sum+a_i$的取值范围在$[1,2\times 10^{11}]$与集合$1\le b_i\le a_i$(取值范围大约在$[1,10^6]$)，所以数据保证是一定有交集的，所以情况三不存在，即集合长度$d=min(b,d)-max(a,c)+1$；

c++代码

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long LL;

const int N = 2e5 + 10;

int n;
LL s;
int a[N];

LL get(LL a, LL b, LL c, LL d)
{
    return min(b, d) - max(a, c) + 1;
}

int main()
{
    scanf("%d%lld", &n, &s);
    
    LL sum = 0;
    /*读入a数组,并计算a1+a2+...+an*/
    for (int i = 1; i <= n; i ++ )
    {
        scanf("%d", &a[i]);
        sum += a[i];
    }
    
    for (int i = 1; i <= n; i ++ )//输出答案~
        printf("%d ", a[i] - get(s - sum + a[i], s - n + 1, 1, a[i]));
        
    return 0;
}

时间复杂度：$O(n)$

空间复杂度：$O(n)$