关闭

树状数组(二维)

157人阅读 评论(0) 收藏 举报
分类:

今天学习了一维的数组数组,就又学了学二维的树状数组,

其实,二维树状数组和一维的套路差不多,就是在处理的时候有一点点不同而已。。

一维树状数组更新是这样的:

void add(int x,int val)  
{  
    for(;x<=n;x+=lowbit(x))  
    {  
        num[x]+=val;  
    }  
}  

二维树状数组更新是这样的:

void  add(int x,int y,int val)  
{  
    for(int i=x;i<=n;i+=lowbit(i))  //n就是二维数组的长度
    {  
        for(int j=y;j<=n;j+=lowbit(j))  
        {  
            num[i][j]+=val;  
        }  
    }  
}  

一维树状数组查询是这样的:

int query(int x)  
{  
    int ans=0;  
    for(;x>0;x-=lowbit(x))  
    {  
        ans+=c[i];  
    }  
    return ans;  
}  

查询的是1到x的数总和。

二维树状数组查询是这样的

int query(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+=num[i][j];  
        }  
    }  
    return ans;  
}  

查询的是(1,1)到(x,y)的数的总和。(比如二维数组a【10】【10】,(1,1)到(3,3)=(1,1)+(1,2)+(1,3)+(2,1)+(2,2)+(2,3)+(3,1)+(3,2)+(3,3),即(1,1)是矩形的左上角,(3,3)矩形的右下角,围成的矩形的和)

例题:poj-1195

题意:
给出一个全0的矩阵,然后一些操作
0 S:初始化矩阵,维数是S*S,值全为0,这个操作只有最开始出现一次
1 X Y A:对于矩阵的X,Y坐标增加A
2 L B R T:询问(L,B)到(R,T)区间内值的总和
3:结束对这个矩阵的操作

思路:

使用二维树状数组进行操作

代码:

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <vector>
#include <set>
#include <map>
#include <string>
#include <cstring>

using namespace std;
int a[1030][1030],c[1030][1030];
int n;
void read(int x,int y,int num)
{
    for(int i=x;i<=n;i+=i&-i)
    {
        for(int j=y;j<=n;j+=j&-j)
        {
            c[i][j]+=num;
        }
    }
}
int sum(int x,int y)
{
    int ans=0;
    for(int i=x;i>0;i-=i&-i)
    {
        for(int j=y;j>0;j-=j&-j)
        {
            ans+=c[i][j];
        }
    }
    return ans;
}
int main()
{
    int m;
    memset(c,0,sizeof(c));
    while(scanf("%d",&m))
    {
        if(m==0)
        {
            //int n;
            //cin>>n;
            scanf("%d",&n);
            for(int i=1;i<=n;i++)
                for(int j=1;j<=n;j++)
                   a[i][j]=0;
        }
        else if(m==1)
        {
            int x,y,s;
            scanf("%d%d%d",&x,&y,&s);
            x++;
            y++;
            //s++;
            read(x,y,s);
        }
        else if(m==2)
        {
            int x1,y1,x2,y2;
            scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
            x1++;y1++;x2++;y2++;
            //cout<<sum(x2,y2)-sum(x1-1,y1-1)<<endl;
            printf("%d\n",sum(x2,y2)-sum(x2,y1-1)+sum(x1-1,y1-1)-sum(x1-1,y2));
        }
        else
        {
            break;
        }
    }
    return 0;
}
0
0
查看评论

一维 + 二维树状数组 + 单点更新 + 区间更新 详解

树状数组详解: 假设一维数组为A[i](i=1,2,...n),则与它对应的树状数组C[i](i=1,2,...n)是这样定义的:  C1 = A1  C2 = A1 + A2  C3 = A3  C4 = A1 + A2 + A3 + A4  C5...
  • qq_18661257
  • qq_18661257
  • 2015-08-08 00:13
  • 3622

POJ2155 - Matrix (二维树状数组)

题目链接:http://poj.org/problem?id=2155题目大意:给定一个矩阵,初始化为0,现在可以进行两种操作,一种是查询某个点的值是 0 还是 1。另一种是让这个矩阵的一个子矩阵内的值取反。解题过程:省赛选拔赛的题,太难了直接没看……… 后来补起来,有模板还是挺容易的。题目分析...
  • ACM_Fish
  • ACM_Fish
  • 2017-04-09 17:00
  • 248

POJ 2155 Matrix(二维树状数组,绝对详细)

Matrix Time Limit: 3000MS   Memory Limit: 65536K Total Submissions: 20599   Accepted: 7673 De...
  • acm_BaiHuzi
  • acm_BaiHuzi
  • 2015-07-09 16:49
  • 2886

[TYVJ P1716/BZOJ 3132 上帝造题的七分钟] 二维树状数组区间修改、区间查询

[TYVJ P1716/BZOJ 3132 上帝造题的七分钟] 二维树状数组区间修改、区间查询知识点:data structure binary index tree1. 题目链接[TYVJ P1716 上帝造题的七分钟] 类似题目还有[POJ 2155 Matrix]。只不过这个题目就是一个二维...
  • ACMore_Xiong
  • ACMore_Xiong
  • 2017-01-03 03:09
  • 706

【算法】逆序对问题的四种解法(归并排序,BST,树状数组,线段树)及变形

引发我对逆序对这个问题思考的源自这道题:315. Count of Smaller Numbers After Self 以前我以为求一个序列逆序对个数的解法就是归并排序的改版
  • haolexiao
  • haolexiao
  • 2017-02-12 17:18
  • 1188

POJ 2155 Matrix(二维树状数组+数组数组区间更新+单点查询)

Given an N*N matrix A, whose elements are either 0 or 1. A[i, j] means the number in the i-th row and j-th column. Initially we have A[i, j] = 0 (1 &l...
  • qq_37497322
  • qq_37497322
  • 2017-07-26 11:11
  • 299

poj 1195 二维树状数组 及二维树状数组模板

http://poj.org/problem?id=1195 求矩阵和的时候,下标弄错WA了一次... 求矩形(x1,y1) (x2,y2)的sum |sum=sum(x2,y2)-sum(x1-1,y2)-sum(x2,y1-1)+sum(x1-1,y1-1) 二维树状数组讲解:htt...
  • u011026968
  • u011026968
  • 2014-08-13 10:45
  • 824

树状数组求区间最大值

讲这个的博文已经不少了,但感觉不够详细不够通俗易懂,所以我尝试着更详细更通俗易懂的说一下我的理解。   这个算法只支持单点修改和区间查询最值。每一次维护和查询的时间复杂度都是O((logn)^2),但这是满打满算的时间复杂度。 假设是要维护和查询区间的最大值(最小值将max改成min 就好...
  • u010598215
  • u010598215
  • 2015-09-04 10:42
  • 5160

hdu1892 See you~ 二维树状数组

首先,这个时二维数组数组的用法。。。 直接套模版的说。。。 See you~ Time Limit: 5000/3000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others) ...
  • kyoma
  • kyoma
  • 2016-10-22 22:42
  • 307

bzoj2120树状数组套主席树解法

这题是分块和莫队的板,不过听说可以树状主席套板,就屁颠屁颠地跑过来打板了。(然而并没有想象得那么简单) 首先把颜色去重排序种种 然后,记录每个颜色的前一种相同颜色的位置,假设为head数组。 l~r中有多少种不同的颜色,就是有多少种l~r中的颜色的前一种颜色的位置在l之前(证明:如果一种颜色有在这段...
  • lvzelong2014
  • lvzelong2014
  • 2017-08-02 14:22
  • 249
    个人资料
    • 访问:13882次
    • 积分:884
    • 等级:
    • 排名:千里之外
    • 原创:76篇
    • 转载:2篇
    • 译文:0篇
    • 评论:3条