简单前缀和问题

前缀和问题

力扣第70场双周赛第二题
2145、统计隐藏数组数目
题目描述
给你一个下标从 0 开始且长度为 n 的整数数组 differences ,它表示一个长度为 n + 1 的 隐藏 数组 相邻 元素之间的 差值 。更正式的表述为:我们将隐藏数组记作 hidden ,那么 differences[i] = hidden[i + 1] - hidden[i] 。
同时给你两个整数 lower 和 upper ,它们表示隐藏数组中所有数字的值都在 闭 区间 [lower, upper] 之间。
比方说,differences = [1, -3, 4] ,lower = 1 ,upper = 6 ,那么隐藏数组是一个长度为 4 且所有值都在 1 和 6 (包含两者)之间的数组。
[3, 4, 1, 5] 和 [4, 5, 2, 6] 都是符合要求的隐藏数组。
[5, 6, 3, 7] 不符合要求,因为它包含大于 6 的元素。
[1, 2, 3, 4] 不符合要求,因为相邻元素的差值不符合给定数据。
请你返回 符合 要求的隐藏数组的数目。如果没有符合要求的隐藏数组,请返回 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 。

题目分析
初看题目,隐藏数组的元素可以通过整数数组推导出来,但是每一个位置都会有很多种可能,而且无法准确判断是否符合题目要求。再一次次筛选是若数据变大,便会出现超出时间限制的情况。
之前在学校有听别人讲到过前缀和的思想,这个题在推导隐藏数组的过程又和数列比较像,于是就想到用反推的思想来确定可行数组的个数。

	已知:
	differences[i]=hidden[i+1]-hidden[i]
	由数学知识可推导知:
	hidden[i]=difference[0]+difference[1]+....difference[i-1]+hidden[0]
	要使数组满足题目条件,则:
	lower<hidden[i]<upper
	则lower-difference[0]+difference[1]+....difference[i-1]<hidden[0]<upper-difference[0]+difference[1]+....difference[i-1]

我们将difference[0]+difference[1]+…difference[i-1]的数值放在一个数组中,求出每一次a[i]满足条件时a[0]的取值范围,并不断更新a[0]的数值范围(不断缩小),直到所有a[i]都满足条件时a[0]的取值个数即为题目答案。

class Solution {
public:
    int numberOfArrays(vector<int>& differences, int lower, int upper) {
        int n=differences.size();
       // vector<int>a(n);
        vector<long long>h(n);//difference的前缀和数组
        h[0]=differences[0];
        for(int i=1;i<n;i++){
            h[i]=h[i-1]+differences[i];
        }
        int low=lower;
        int up=upper;
        int result=0;
        for(int i=0;i<n;i++){//不断缩小可行范围
            int left=lower-h[i];
            int right=upper-h[i];
            low=max(low,left);
            up=min(up,right);
        }
        if(low>up)return 0;
        result=up-low+1;//最后所有a[i]满足条件时a[0]的取值范围
        return result;
    }
};

这是一个简单的前缀和思想,但是在题目中涉及了反推思想,在平时比较不容易想到。前缀和还有二维数组的前缀和,大家有兴趣可以去了解了解。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值