码迷,mamicode.com
首页 > 编程语言 > 详细

【树状数组二维区间加+区间查询模板】bzoj3132

时间:2017-08-19 23:35:23      阅读:215      评论:0      收藏:0      [点我收藏+]

标签:str   style   个数   维护   main   fine   can   iostream   pre   

新知识,其实和之前讲过的一维差不多,只要维护四个数组就行了,不过还是参考了别人的代码,还是要好好练练才行

#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 2050
using namespace std;
int a[maxn][maxn],b[maxn][maxn],c[maxn][maxn],d[maxn][maxn];
int n,m;
int lowbit(int x){return(x&-x);}
void update(int x,int y,int v,int g[][maxn])
{
  for (int i=x;i<=n;i+=lowbit(i)) for (int j=y;j<=m;j+=lowbit(j)) 
  g[i][j]+=v;
}
int query(int x,int y,int g[][maxn])
{
  int sum=0;
  for (int i=x;i>0;i-=lowbit(i))
  for (int j=y;j>0;j-=lowbit(j))
  sum+=g[i][j];
  return sum;
}
int calc(int x,int y)
{
  int sum=query(x,y,a)+query(x,y,b)*x+query(x,y,c)*y+query(x,y,d)*x*y;
  return sum;
}
void add(int x,int y,int v)
{
  if (!x||!y) return;
  update(x,y,x*y*v,a);
  update(x,y,y*-v,b); update(1,y,y*v,b);
  update(x,y,x*-v,c); update(x,1,x*v,c);
  update(x,y,v,d);update(1,1,v,d);
  update(x,1,-v,d);update(1,y,-v,d);
//这里是重点,在访问到当前的时候,会加上一个值,过了边界的时候再减去这个值 }
int main() { char ch; scanf("%c %d%d",&ch,&n,&m); int x,y,xx,yy,val; while (scanf("\n%c ",&ch)!=EOF) { if (ch==L) { scanf("%d%d%d%d%d",&x,&y,&xx,&yy,&val); add(x-1,y-1,val);add(xx,yy,val);add(xx,y-1,-val);add(x-1,yy,-val); } else { scanf("%d%d%d%d",&x,&y,&xx,&yy);x--;y--; int ans=calc(xx,yy)-calc(xx,y)-calc(x,yy)+calc(x,y); printf("%d\n",ans); } } return 0; }

 

【树状数组二维区间加+区间查询模板】bzoj3132

标签:str   style   个数   维护   main   fine   can   iostream   pre   

原文地址:http://www.cnblogs.com/2014nhc/p/7398164.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!