Color the ball HDU - 1556 （前缀和与差分）

最新推荐文章于 2021-08-03 10:53:42 发布

sunmaoxiang

最新推荐文章于 2021-08-03 10:53:42 发布

阅读量218

点赞数

本文链接：https://blog.csdn.net/sunmaoxiang/article/details/82458694

版权

题目

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个气球总共被涂色的次数。
Sample Input

Sample Output

1 1 1
3 2 1

思路

这道题可以用线段树来做，不过代码量大，新学一招前缀和差分求解数组区间修改问题很方便。
前缀和其实可以把它理解为数学上的数列的前n项和（对于一个一维数组的前缀和）。我们定义对于一个数组a的前缀和数组s，s[i] = a[1]+a[2]+...+a[i].
用差分实现区间操作给定一个长度为n的数列a，要求支持操作add(L,R,k)表示对a[L]~a[R]的每个数都加上k。并求修改后的序列a。我们考虑用差分的做法。这里需要一个辅助数组c，c用来记录某一个位置上的总改变量。c[i]表示的是i~n这些元素都加上c[i]这个数。我们对[L,R]区间进行加值操作，在c[L]处加一个k，在c[R+1]处就减去一个k。最后求序列的每个位置变成了多少，只需要求一下c数组的前缀和，然后和原数组按位相加就好。

代码

#include <bits/stdc++.h>
using namespace std;

const int maxn = 100000 + 100;

int sum_change[maxn];
int change[maxn];

void init()
{
    memset(sum_change, 0, sizeof(sum_change));
    memset(change, 0, sizeof(change));
}

int Change(int l, int r) {
    change[l]++;
    change[r+1]--;
}

int main()
{
    int T;
    int l, r;
    while(cin >> T && T)
    {
        init();
        int n = T;
        while(T--){
            scanf("%d%d", &l, &r);
            Change(l-1, r-1);
        }
        sum_change[0] = change[0];
        for(int i=1;i<n; i++) {
             sum_change[i] = change[i]+sum_change[i-1];
        }
        for(int i=0;i < n; i++) {
            if(i) {
                printf(" %d", sum_change[i]);
            }
            else printf("%d", sum_change[i]);
        }

        cout << endl;
    }
    return 0;
}