树状数组＆二维树状数组模板

最新推荐文章于 2021-07-05 14:53:15 发布

WilliamCode

最新推荐文章于 2021-07-05 14:53:15 发布

阅读量786

点赞数

文章标签： c++ poj 编程语言

本文链接：https://blog.csdn.net/WilliamCode/article/details/51134283

版权

树状数组模板

int lowbit(int x)
{       return (x&(-x));}
int add(int x,int val)
{
        for (;x<=40000;x+=lowbit(x))
                c[x]+=val;
}
int que(int x)
{
        int ans=0;
        for (;x>0;x-=lowbit(x))
                ans+=c[x];
        return ans;
}

模板题：http://poj.org/problem?id=2352　
题解：

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
int c[300000],rank[300000];
int n;
int lowbit(int x)
{       return (x&(-x));}
int add(int x,int val)
{
        for (;x<=40000;x+=lowbit(x))
                c[x]+=val;
}
int que(int x)
{
        int ans=0;
        for (;x>0;x-=lowbit(x))
                ans+=c[x];
        return ans;
}


int main()
{
        cin>>n;
        for (int i=1;i<=n;i++)
        {
                int x,y;
                scanf("%d%d",&x,&y);
                x++;y++;
                add(x);
                rank[que(x)]++;
        }
        for (int i=1;i<=n;i++)
                printf("%d\n",rank[i]);
}

二维树状数组模板：

int lowbit(int x)
{
        return (x&(-x));
}
int add(int x,int y,int val)
{
        for (int i=x;i<=s;i+=lowbit(i))
                for (int j=y;j<=s;j+=lowbit(j))
                        c[i][j]+=val;
}
int que(int x,int y)
{
        int ans=0;
        for (int i=x;i>0;i-=lowbit(i))
                for (int j=y;j>0;j-=lowbit(j))
                        ans+=c[i][j];
        return ans;
}

同样很简单，注意数组尽量开大，树状数组会产生未知原因的溢出
模板题：【poj 1195】Mobile phones　http://poj.org/problem?id=1195
（题意即：每行第一个数，如果为０，后面一个数字ｓ，表示给出一个ｓ*ｓ的矩阵，初始化为均为0，如果为　1　：(X Y A)对位于坐标（X Y）的值加A；如果为　2：（L B R T）求出位于L<=x<=R，B<=y<=T的值的和；如果为　3：结束程序。）

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
int s;
int c[2050][2050];
int lowbit(int x)
{
        return (x&(-x));
}
int add(int x,int y,int val)
{
        for (int i=x;i<=s;i+=lowbit(i))
                for (int j=y;j<=s;j+=lowbit(j))
                        c[i][j]+=val;
}
int que(int x,int y)
{
        int ans=0;
        for (int i=x;i>0;i-=lowbit(i))
                for (int j=y;j>0;j-=lowbit(j))
                        ans+=c[i][j];
        return ans;
}

int main()
{
        int p,n,m,a,b;
        while(cin>>p)
        {
                if (p==3)       return 0;
                if (p==0)       {scanf("%d",&s);
                                        for (int i=1;i<=s+100;i++)
                                                for (int j=1;j<=s+100;j++)
                                                        c[i][j]=0;
                                        }
                if (p==1)
                {
                        scanf("%d%d%d",&a,&b,&n);
                        a++;b++;
                        add(a,b,n);
                }
                else if (p==2)
                {
                        scanf("%d%d%d%d",&a,&b,&n,&m);
                        a++;b++;n++;m++;
                        int ans;
                        ans=que(n,m)-que(a-1,m)-que(n,b-1)+que(a-1,b-1);
                        printf("%d\n",ans);

                }
        }
}