本博文为《代码随想录》学习笔记，原文链接：代码随想录
题目
原题链接：58. 区间和（第九期模拟笔试）
题目描述
给定一个整数数组 Array，请计算该数组在每个指定区间内元素的总和。
输入描述
第一行输入为整数数组 Array 的长度 n，接下来 n 行，每行一个整数，表示数组的元素。随后的输入为需要计算总和的区间下标：a，b （b > = a），直至文件结束。
输出描述
输出每个指定区间内元素的总和。
输入示例
5
1
2
3
4
5
0 1
1 3输出示例
3
9提示信息
数据范围：
0 < n <= 100000
题解
方法一：暴力解法
注意**处代码的写法，之前没有这样写过
#include <bits/stdc++.h>
using namespace std;
int main()
{
int n,a,b;
cin>>n;
int nums[n];
for(int i=0;i<n;i++)
{
cin>>nums[i];
}
while(cin>>a>>b)//** 注意这里的写法
{
int sum=0;
for(int i=a;i<=b;i++)sum+=nums[i];
cout<<sum<<endl;
}
return 0;
}vector初始化
这里纠正之前关于vector的一个误区，vector初始化的方式有以下几种
1、定义空向量
vector<int> a; //相当于空数组
2、定义具有10个整型元素的向量
vector<int> a(10); //相当于a[10]
3、定义具有10个整型元素的向量，且赋予每个元素初值为1
vector<int> a(10,1); //相当于a[10] = {1}
4、定义与向量b具有相同值的向量a
vector<int> a(b); //将向量b赋值给向量a，即向量a等于向量b
5、将向量b中下标0-2（共三个）的元素赋值给a
//第一个数据是起始地址，第二个数据是结束地址（不包括），第二个数据就是你要截取的长度加1
vector<int> a(b.begin(), b.begin()+3);6、从数组中获得初值
int b[7] = {1,2,3,4,5,6,7}; //定义整形数组
vector<int> a(b, b+7）; //将数组b赋值给a，第一个数据是起始地址，第二个数据是结束地址（不包括）
7、二维数组初始化
vector<vector<int>> a; //初始化为int型二维数组，相当于int a[][]
下标只能获取已存在的元素，所以以下赋值方法对第一种初始化方法是错误的
for(int i=0; i<10; i++){
a[i] = i; //应使用a.push_back(i)
}但当数组中已经存在元素则可以使用上述方法进行赋值，因此方法一还可以写成如下形式：
#include <iostream>
#include <vector>
using namespace std;
int main() {
int n, a, b;
cin >> n;
vector<int> vec(n);
for (int i = 0; i < n; i++) cin >> vec[i];
while (cin >> a >> b) {
int sum = 0;
// 累加区间 a 到 b 的和
for (int i = a; i <= b; i++) sum += vec[i];
cout << sum << endl;
}
} 提交代码，发现超时了
当查询范围和查询次数较大时，这种暴力解法的时间复杂度时非常大的。
方法二：前缀和方法
对前缀和的思路进行举例说明，例如我们要统计vec[i]这个数组上的区间和。
我们先做累加，即p[i]表示下标0到i的vec[i]累加之和。如图：
如果，我们想统计，在vec数组上 下标 2 到下标 5 之间的累加和，那就用 p[5] - p[1] 就可以了。
p[1] = vec[0] + vec[1];
p[5] = vec[0] + vec[1] + vec[2] + vec[3] + vec[4] + vec[5];
p[5] - p[1] = vec[2] + vec[3] + vec[4] + vec[5];
p[5] - p[1] 就是 红色部分的区间和。
而 p 数组是我们之前就计算好的累加和，所以后面每次求区间和的之后 我们只需要 O(1) 的操作。
特别注意： 在使用前缀和求解的时候，要特别注意 求解区间。
如上图，如果我们要求 区间下标 [2, 5] 的区间和，那么应该是 p[5] - p[1]，而不是 p[5] - p[2]。在解题时可以通过画图来帮助理解。
编写代码如下：
#include <bits/stdc++.h>
using namespace std;
int main()
{
int n,a,b;
cin>>n;
int nums[n];
int p[n];//前缀和数组
for(int i=0;i<n;i++)
{
cin>>nums[i];
p[i]=p[i-1]+nums[i];
}
while(cin>>a>>b)
{
int sum=p[b]-p[a-1];
cout<<sum<<endl;
}
return 0;
}
《代码随想录》中给出的代码如下：
#include <iostream>
#include <vector>
using namespace std;
int main() {
int n, a, b;
cin >> n;
vector<int> vec(n);
vector<int> p(n);
int presum = 0;
for (int i = 0; i < n; i++) {
cin >> vec[i];
presum += vec[i];
p[i] = presum;
}
while (cin >> a >> b) {
int sum;
if (a == 0) sum = p[b];
else sum = p[b] - p[a - 1];
cout << sum << endl;
}
}
C++ 代码 面对大量数据 读取 输出操作，最好用scanf 和 printf，耗时会小很多：
#include <iostream>
#include <vector>
using namespace std;
int main() {
int n, a, b;
cin >> 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);
}
}
