HDU - 1556: Color the ball

最新推荐文章于 2021-08-20 18:18:38 发布

陆克和他的代码

最新推荐文章于 2021-08-20 18:18:38 发布

阅读量346

点赞数 1

CC 4.0 BY-SA版权

分类专栏： # OJ 文章标签： ACM 树状数组 1556 Color the ball BIT

本文链接：https://blog.youkuaiyun.com/Dream_Weave/article/details/81414282

OJ 专栏收录该内容

66 篇文章

订阅专栏

本文介绍了一种使用树状数组求解区间和的方法，提供了两种实现方式：向上更新和向下更新。通过实例代码详细展示了如何利用树状数组解决实际问题，并解释了关键步骤。

题目链接：点击打开链接

题目大意：略。

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

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;
}