3992 树上有猴（二分、求解区间交集）

1. 问题描述：

一棵大树上初始时有若干个猴子（也可能没有）。接下来 n 个时刻，每个时刻树上猴子的数量都会发生变动。第 i 时刻的变动数量为 ai，ai > 0 表示有 ai 只猴子上了树，ai < 0 表示有 |ai| 只猴子下了树。已知，在任意时刻树上的猴子总数都没有超过 w，当然也不可能小于 0。请问，初始时的猴子数量共有多少种可能性？例如，当 n = 3，w = 5 且 a1 = 2，a2 = 1，a3 = −3 时，初始时的猴子数量可能为 0，1，2 个。

输入格式

第一行包含两个整数 n，w。第二行包含 n 个整数 a1，a2，…，an。

输出格式

一个整数，表示初始时的猴子数量共有多少种可能性。如果无解，即初始时有多少只猴子都不满足题目要求，则输出 0。

数据范围

前六个测试点满足，1 ≤ n ≤ 10，1 ≤ w ≤ 10。
所有测试点满足，1 ≤ n ≤ 1000，1 ≤ w ≤ 10 ^ 9，−10 ^ 6 ≤ ai ≤ 10 ^ 6。

输入样例1：

3 5
2 1 -3

输出样例1：

3

输入样例2：

2 4
-1 1

输出样例2：

4

输入样例3：

4 10
2 4 1 2

输出样例3：

2
来源：https://www.acwing.com/problem/content/description/3995/

2. 思路分析：

一开始的时候想到是二分，因为一开始的时候树上的猴子最少有0个，最多有w个，当树上的猴子一开始有x个的时候需要满足增加的猴子与减少的猴子之后树上的猴子仍然满足这个最小限制与最大限制，所以想到使用二分枚举出树上最少猴子与最多猴子的数量，它们的差值 + 1就是答案，此外还需要判断是否存在无解的情况。其实我们仔细分析一下其实是满足n个不等式要求的x的取值范围，所以本质上需要求解n个区间的交集：

3. 代码如下：

二分：

from typing import List


class Solution:
    # 判断当前树上mid的时候否满足树上的猴子的数量小于等于w
    def check1(self, mid: int, m: int, a: List[int]):
        count = mid
        for i in range(len(a)):
            count += a[i]
            if count > m:
                return False
        return True
    
    # 判断是否当前mid的时候是否树上的猴子小于0
    def check2(self, mid: int, a: List[int]):
        count = mid
        for i in range(len(a)):
            count += a[i]
            if count < 0: return False
        return True

    def process(self):
        n, m = map(int, input().split())
        a = list(map(int, input().split()))
        s = m
        # 判断是否答案小于0
        for x in a:
            s += x
            if s < 0: return 0
        s = 0
        # 判断答案是否大于m
        for x in a:
            s += x
            if s > m: return 0
        l, r = 0, m
        maxv = minv = -1
        # 二分找到树上可能的最大猴子的数量
        while l <= r:
            mid = l + r >> 1
            if self.check1(mid, m, a):
                maxv = mid
                l = mid + 1
            else:
                r = mid - 1
        l, r = 0, m
        # 二分找到树上可能的最小的猴子的数量
        while l <= r:
            mid = l + r >> 1
            if self.check2(mid, a):
                minv = mid
                r = mid - 1
            else:
                l = mid + 1
        return max(0, maxv - minv + 1)


if __name__ == '__main__':
    print(Solution().process())

求解区间的交集：

class Solution:
    def process(self):
        n, w = map(int, input().split())
        a = list(map(int, input().split()))
        # s记录前缀和
        s = 0
        l, r = 0, w
        for x in a:
            s += x
            l = max(l, -s)
            r = min(r, w - s)
        return max(0, r - l + 1)


if __name__ == '__main__':
    print(Solution().process())