HDU - 1556: Color the ball

题目链接：​​点击打开链接​​

题目大意：略。

解题思路：求区间和的树状数组。

Ps：至于为什么最后用 query(i) 来代表最终结果，用数学可以解释（前提理解了 x+=lowbit(x) or x-=lowbit(x)），只能说很凑巧，但用语言无法形容其意。

AC 代码（向上更新，向下求和）

#include<bits/stdc++.h>
#include<cmath>

#define mem(a,b) memset(a,b,sizeof a);
#define INF 0x3f3f3f3f

using namespace std;

typedef long long ll;

const int maxn=100005;

int n;
int rs[maxn];

int lowbit(int x)
{
    return x & -x;
}

void update(int x,int val)
{
    while(x<=n)
    {
        rs[x]+=val;
        x+=lowbit(x);
    }
}

int query(int x)
{
    int res=0;
    while(x>0)
    {
        res+=rs[x];
        x-=lowbit(x);
    }

    return res;
}

int main()
{
    int l,r;
    while(~scanf("%d",&n) && n)
    {
        mem(rs,0);
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&l,&r);
            update(l,1);
            update(r+1,-1);
        }

        for(int i=1;i<n;i++)
            printf("%d ",query(i));
        printf("%d\n",query(n));
    }

    return 0;
}

AC 代码（向下更新，向上求和）

#include<bits/stdc++.h>
#include<cmath>

#define mem(a,b) memset(a,b,sizeof a);
#define INF 0x3f3f3f3f

using namespace std;

typedef long long ll;

const int maxn=100005;

int n;
int rs[maxn];

int lowbit(int x)
{
    return x & -x;
}

void update(int x,int val)
{
    while(x>0)
    {
        rs[x]+=val;
        x-=lowbit(x);
    }
}

int query(int x)
{
    int res=0;
    while(x<=n)
    {
        res+=rs[x];
        x+=lowbit(x);
    }

    return res;
}

int main()
{
    int l,r;
    while(~scanf("%d",&n) && n)
    {
        mem(rs,0);
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&l,&r);
            update(r,1); // [1,r] 全部涂上，+1次
            update(l-1,-1); // [1,l-1] 全部擦掉，-1次
        }

        for(int i=1;i<n;i++)
            printf("%d ",query(i));
        printf("%d\n",query(n));
    }

    return 0;
}