前缀和处理数组区间之和问题

1.什么是区间和问题

“区间和问题”通常指的是涉及计算或处理数组或数列某个子区间(即一段连续元素)的总和的类型问题。这类问题可能有多种变体和不同的复杂度,但基本思想都是在给定的区间内快速计算总和或处理与区间和相关的操作。

2.例题1

题目描述

给定一个整数数组 Array,请计算该数组在每个指定区间内元素的总和。

输入描述

第一行输入为整数数组 Array 的长度 n,接下来 n 行,每行一个整数,表示数组的元素。随后的输入为需要计算总和的区间下标:a,b (b > = a),直至文件结束。

输出描述

输出每个指定区间内元素的总和。

3.解题思路

在看到题目的第一反应,大多数同学认为这是一道非常简单的数组求和问题,只需要将区间中对应下标的元素依次相加即可,但通过这种方法提交问题往往会编译超时。why?取极端情况:如果我次次查询的范围都是n-1到0,那么算法的复杂度是O(n * m) m,也就是查询的次数。可见查询次数非常多,而且时间也很长。
所以,为了加速区间和的查询,可以创建一个前缀和数组。前缀和数组 P[i] 表示原数组 A 从第1个元素到第 i 个元素的和。利用前缀和数组可以在 O(1) 时间内计算任意区间的和:
在这里插入图片描述

通过上面的图示可以看出,Sum(a,b)=P[b]−P[a−1]
注意这里a不可以为0,a为0要单独考虑:Sum(0,b)=P[b]
计算前缀和数组的时间复杂度是 O(n),每个查询时间复杂度是 O(1)。
在这里插入图片描述

有了前缀和数组之后,我们就可以在输入原数组元素时,设定一个累加变量,输入一次累加变量就增加一次。而前缀和数组的每个元素大小就是对应的累加变量的大小。
代码如下:

# include <iostream>
# include <vector>
using namespace std;
int main(){
    int n,a,b;
    scanf("%d",&n);
    vector<int> vec(n);
    vector<int> p(n);
    int presum=0;
    for(int i=0;i < n;i++){
        scanf("%d",&vec[i]);
        presum+=vec[i];
        p[i]=presum;
    }
    while(~scanf("%d%d", &a, &b)){
        int sum;
        if(a==0) sum=p[b];
        else sum = p[b]-p[a-1];
        printf("%d\n",sum);
    }
}

---------8.9更新---------

4.例题2

题目描述

在一个城市区域内,被划分成了n * m个连续的区块,每个区块都拥有不同的权值,代表着其土地价值。目前,有两家开发公司,A 公司和 B 公司,希望购买这个城市区域的土地。 现在,需要将这个城市区域的所有区块分配给 A 公司和 B 公司。
然而,由于城市规划的限制,只允许将区域按横向或纵向划分成两个子区域,而且每个子区域都必须包含一个或多个区块。 为了确保公平竞争,你需要找到一种分配方式,使得 A 公司和 B 公司各自的子区域内的土地总价值之差最小。
注意:区块不可再分。

输入描述

第一行输入两个正整数,代表 n 和 m。接下来的 n 行,每行输出 m 个正整数。

输出描述

请输出一个整数,代表两个子区域内土地总价值之间的最小差距。

5.解题思路

嵌套两个for去累加二维数组区间里的和,分为横向前缀和与纵向前缀和。一个for循环来枚举分割线可能在的位置。
本题也可以使用前缀和的思路来求解,先将行方向和列方向的和求出来,这样可以方便知道划分的两个区间的和。
代码如下:

#include <iostream>
#include <vector>
#include <climits>
using namespace std;
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    int sum=0;
    vector<vector<int>> vec(n,vector<int>(m,0));
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            cin>>vec[i][j];
            sum+=vec[i][j];
        }
    }
    //横向
    vector<int> horizontal(n,0);
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            horizontal[i]+=vec[i][j];
        }
    }
    //纵向
    vector<int> vertical(m,0);
    for(int j=0;j<m;j++){
        for(int i=0;i<n;i++){
            vertical[j]+=vec[i][j];
        }
    }
    int results=INT_MAX;
    int horizontalCut=0;
    int verticalCut=0;
    for(int i=0;i<n;i++){
        horizontalCut += horizontal[i];
        results=min(results,abs(sum-horizontalCut-horizontalCut));
    }
    for (int j=0; j<m;j++) {
        verticalCut += vertical[j];
        results=min(results,abs(sum-verticalCut-verticalCut));
    }
    cout << results << endl;
}

6.小节

1.scanf 函数返回成功读取的输入项的数量(如成功读取两个整数则返回 2),如果遇到输入错误或到达文件末尾,scanf 会返回 EOF(通常为 -1)。while (~scanf("%d%d", &a, &b)) 这行代码的意思是对 scanf 的返回值进行按位取反,然后将结果用于 while 循环的条件判断。
2.当对 scanf 的返回值使用 ~ 时,如果 scanf 成功读取了输入项,返回值将不为 0,而取反后的结果通常为一个负数;如果 scanf 返回 -1(即 EOF),则 ~(-1) 将得到 0,这时循环会停止。
3.C++ 代码中面对大量数据的读取、输出操作时,最好用scanf 和 printf,而不是cin和cout。因为scanf和printf耗时会小很多
4.区间和问题是许多编程竞赛和算法面试中常见的题型,掌握这些问题的高效解法对提升算法和数据结构的能力非常重要。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值