【差分模版】Color the ball
题意
N个气球排成一排,从左到右依次编号为1,2,3….N.每次给定2个整数a b(a <= b),lele便为骑上他的“小飞鸽”牌电动车从气球a开始到气球b依次给每个气球涂一次颜色。但是N次以后lele已经忘记了第I个气球已经涂过几次颜色了,你能帮他算出每个气球被涂过几次颜色吗?
Input
每个测试实例第一行为一个整数N,(N <= 100000).接下来的N行,每行包括2个整数a b(1 <= a <= b <= N)。
当N = 0,输入结束。
Output
每个测试实例输出一行,包括N个整数,第I个数代表第I个气球总共被涂色的次数
Example
3
1 1
2 2
3 3
3
1 1
1 2
1 3
0
1 1 1
3 2 1
分析:本题需要多次修改某段区间的值,故考虑用线段树或者差分,由于差分实现更为简单,故此处用差分实现。差分为前缀和逆运算,常用于多次区间修改。
#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
using namespace std;
int n, l, r;
const int maxn = 1e5+10;
int a[maxn], diff[maxn];
//初始化原数组
void Init()
{
//一般是置0或者手动输入
//一般a[0]=0,方便后续前缀和求原数组
memset(a, 0, sizeof(a));
}
//通过原数组构建差分数组
void getDiff()
{
// 知道差分数组的时候可通过O(n)的时间复杂度求出原数组
// a[i]=a[i-1]+diff[i];a[0]=0
for (int i = 1; i <= n;i++)
diff[i] = a[i] - a[i - 1];//常用的构造方法
}
//区间修改
//l,r为区间左右端点,c为区间元素增量
void modify(int l,int r,int c)
{
diff[l] += c;//使原数组a[l]及右边每个元素都+c
diff[r + 1] -= c;//上一行把超过区间右边界r的值也+c,故得-c调整
}
//计算原数组
void getAns()
{
for (int i = 1; i <= n;i++)
{
a[i] = a[i - 1] + diff[i];
//可在这里做输出或者其他一些操作
printf(" %d" + (i == 1), a[i]);
}
cout << endl;
}
//差分模版
int main()
{
while(cin>>n&&n)
{
Init();//初始化原数组
getDiff();//构建差分数组
for (int i = 0; i < n;i++)
{
cin >> l >> r;
modify(l, r, 1);//题意中增量为1
}
getAns();
}
return 0;
}