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

二维树状数组模板(区间修改+区间查询)

时间:2018-02-22 19:23:48      阅读:199      评论:0      收藏:0      [点我收藏+]

标签:namespace   cal   mat   math   int   down   mes   while   using   

二维树状数组模板(区间修改+区间查询)

例题:JOIOI上帝造题的七分钟
一共两种操作:

\(L\ x_1\ y_1\ x_2\ y_2\ d\):把\((x_1,y_1)\)\((x_2,y_2)\)这个矩形内所有元素加\(d\)
\(k\ x_1\ y_1\ x_2\ y_2\):查询\((x_1,y_1)\)\((x_2,y_2)\)这个矩形内所有元素的和。

代码如下:

#include<bits/stdc++.h>
#define RG register
#define IL inline
#define _ 2050
#define ll long long
//#define ll int
using namespace std;

ll n , m , c1[_][_],c2[_][_],c3[_][_],c4[_][_] ;

IL void upt(ll x , ll y , ll dt){
    for(RG int i = x; i <= n; i += (i & -i) )
        for(RG ll j = y; j <= m; j += (j & -j)){
            c1[ i ][ j ] += dt ;
            c2[ i ][ j ] += dt * y;
            c3[ i ][ j ] += dt * x;
            c4[ i ][ j ] += dt * x * y;
        }
}
IL ll calc(ll x,ll y){
    RG ll res = 0;
    for(RG ll i = x; i > 0; i -= (i & -i))
        for(RG ll j = y; j > 0; j -= (j & -j)){
            res = res
                + (x + 1) * (y + 1) * c1[ i ][ j ]
                - (x + 1) * c2[ i ][ j ]
                - (y + 1) * c3[ i ][ j ]
                + c4[ i ][ j ] ; 
        }
    return res;
}

IL void add(ll X1,ll Y1,ll X2,ll Y2,ll dt){
    upt(X1 , Y1 , dt ) ;
    upt(X2 + 1 , Y1 , -dt ) ;
    upt(X1 , Y2 + 1, -dt ) ;
    upt(X2 + 1, Y2 + 1, dt ) ; 
}
IL ll query(ll X1,ll Y1,ll X2,ll Y2){
    return
        calc(X2 , Y2) + calc(X1 - 1, Y1 - 1) -
        calc(X1 - 1 , Y2) - calc(X2 , Y1 - 1) ;
}

int main(){
    ll X1 , X2 , Y1 , Y2 , z; char c[3]; 
    scanf("X %lld %lld",&n,&m);
    while(scanf("%s",c)!=EOF){
        scanf("%lld%lld%lld%lld", &X1, &Y1, &X2, &Y2);
        if(c[0]=='L'){
            scanf("%lld", &z);
            add(X1 , Y1 , X2 , Y2 , z);
        }
        else printf("%lld\n",query(X1 , Y1 , X2 , Y2 ));
    }return 0;return 0;
    
}

二维树状数组模板(区间修改+区间查询)

标签:namespace   cal   mat   math   int   down   mes   while   using   

原文地址:https://www.cnblogs.com/Guess2/p/8459568.html

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