一维前缀和的应用
一维前缀和顾名思义是对一个一维的数组进行操作。
前缀和是一种重要的预处理,能大大降低查询的时间复杂度。
用前缀和来处理问题可以很好得避免循环的嵌套,大大缩短了程序的运行时间。
我们可以通过一道例题来体会一下一维前缀和的思想。
问题描述:
输入多个区间的左右下标,每次进行加减操作,输出该区间内每个数字的值(初始值均为0)。
输入:
包括多行,第一行:两个整数 n , m。
接下来的 m 行,每行3个整数 L , R , K ,表示区间的左边界 L ,右边界 R, 以及对该区间进行加减操作的数 K 。
L <= R <= 1000000 。
输出:
输出包括 n 行,每行一个整数,表示数组每一个变量的值。
接下来附上C++代码:
#include <iostream>
#include <stdio.h>
using namespace std;
int x[1000005]; // 数据较大,应定义为全局变量,放在堆中(堆会默认初始化变量为 0 )
int main()
{
int n, m, i, l, r, k;
long long int count = 0;
cin >> n >> m;
for (i = 0; i < m; i++)
{
scanf("%d%d%d", &l, &r, &k); // scanf 输入较 cin 速度快
x[r + 1] -= k;
x[l] += k;
}
for (i = 1; i <= n; i++)
{
count += x[i];
printf("%lld\n", count);// printf 输出较 cout 速度快
}
return 0;
}