题目
教室外有 N 棵树,根据不同的位置和树种,学校要对其上不同的药。 因为树的排列成线性,且非常长,我们可以将它们看作一条直线给他们编号。 树的编号从 0-N-1 且 N<1e6。 对于树的药是成区间分布,比如 3 - 5 号的树靠近下水道,所以他们要用驱蚊虫的药, 20 - 26 号的树,他们排水不好,容易涝所以要给他们用点促进根系的药。 诸如此类,每种不同的药要花不同的钱。 现在已知共有 M 个这样的区间,并且给你每个区间花的钱,请问最后,这些树木花了多少药费。
输入
输入描述:
每组输入的第一行有两个整数 N(1 <= N<= 1000000)和 M(1 <= M <= 100000)。
N 代表马路的共计多少棵树,M代表区间的数目,N 和 M 之间用一个空格隔开。
接下来的 M 行每行包含三个不同的整数,用一个空格隔开,表示一个区域的起始点 L 和终止点 R 的坐标,以及花费。
输入样例:
500 3
150 300 4
100 200 20
470 471 19
输出
输出描述:
输出包括一行,这一行只包含一个整数,所有的花费。
输出样例:
2662
解答
-
利用b[i]=a[i]-a[i-1]b[i]=a[i]−a[i−1] 差分式。
这里由于开始时都是 0,可以用,但是用完都还是 0,所以没有意义,所以直接跳过即可。
-
依次读入区间的值,然后将对于区间的操作转化为对于区间端点操作加减。 由于我们从1开始,所以数目整体区间要右移1位。
对于每个 [l,r] 区间的加减操作都转化为对端点 l,r+1 的操作。
-
差分还原(前缀和)。
for (int i = 1; i < n; i++) b[i] = a[i] - a[i - 1]
差分算法解决区间加减问题通用框架如下:
//读入原始数据 n,m,a
输入n,m
for(int i=1;i<=n;i++){
输入a[i]
}
//差分
for(int i=1;i<=n;i++)
b[i]=a[i]-a[i-1]
//区间操作
while(m--)
{
输入l,r,value
b[l]+value
b[r+1]-value
}
//前缀和还原
for(int i=1;i<n;i++)
b[i]=b[i]+b[i-1]
ac代码
#include <iostream>
using namespace std;
int b[100005];
int main()
{
int n; //n层
int m; // m个区间
cin >> n >> m;
while (m--)
{
int l, r, value;
cin >> l >> r >> value;
b[l+1] += value;
b[r + 1+1] -= value;
}
for (int i = 1; i <= n; i++)
b[i] = b[i] + b[i - 1];
int sum = 0;
for (int i = 1; i <= n; i++)
sum += b[i];
/*
也可以一次性搞定
int sum=b[1];
for(int i=1; i<=n; i++){
b[i]=b[i]+b[i-1];
sum+=b[i]
}
*/
cout << sum << endl;
}