给你一个下标从 0 开始且长度为 n 的整数数组 differences
,它表示一个长度为 n + 1
的 隐藏 数组 相邻 元素之间的 差值 。
更正式的表述为:我们将隐藏数组记作 hidden ,那么 differences[i] = hidden[i + 1] - hidden[i]
。
同时给你两个整数 lower 和 upper ,它们表示隐藏数组中所有数字的值都在 闭 区间 [lower, upper] 之间。
请你返回 符合 要求的隐藏数组的数目。 如果没有符合要求的隐藏数组,请返回 0 。
示例 1:
输入:differences = [1,-3,4], lower = 1, upper = 6
输出:2
解释:符合要求的隐藏数组为:
- [3, 4, 1, 5]
- [4, 5, 2, 6]
所以返回 2 。
示例 2:
输入:differences = [3,-4,5,1,-2], lower = -4, upper = 5
输出:4
解释:符合要求的隐藏数组为:
- [-3, 0, -4, 1, 2, 0]
- [-2, 1, -3, 2, 3, 1]
- [-1, 2, -2, 3, 4, 2]
- [0, 3, -1, 4, 5, 3]
所以返回 4 。
示例 3:
输入:differences = [4,-7,2], lower = 3, upper = 6
输出:0
解释:没有符合要求的隐藏数组,所以返回 0 。
提示:
n == differences.length
1 <= n <= 10^5
-10^5 <= differences[i] <= 10^5
-10^5 <= lower <= upper <= 10^5
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/count-the-hidden-sequences 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
public:
int numberOfArrays(vector<int>& differences, int lower, int upper) {
// 求前缀和
vector<long long> presum(differences.size());
long long n = differences.size(), maxprevsum = LONG_LONG_MIN, minprevsum = LONG_LONG_MAX;
for(int i = 0; i < n; ++i)
{
presum[i] = (i>0 ? presum[i-1] : 0)+differences[i];
maxprevsum = max(maxprevsum, presum[i]);
minprevsum = min(minprevsum, presum[i]);
}
// a0 是第一个数,那么数组内最大的数是 a0+maxprevsum, 最小的数字是 a0+minprevsum
// a0+maxprevsum <= upper and a0+minprevsum >= lower
// lower-minprevsum <= a0 <= upper-maxprevsum
// 加上题目的条件 lower <= a0 <= upper
long long ans = min(1LL*upper, upper-maxprevsum) - max(1LL*lower, lower-minprevsum)+1;
return ans >= 0 ? ans : 0;
}
};
204 ms 113 MB C++