poj 1195 Mobile phones(树状数组)

刚接触树状数组,感觉是个很简练的数据结构,虽然我还没怎么明白它的原理。本来想用浙大模版里的树状数组,不过看半天不知道代码什么意思,而且求和的时候坐标向后错了一位,只能在网上找一个了,速度比浙大模版要快一点,不过浙大模版的简练程度让我真心佩服……

#include <cstdio>
#include <iostream>
#include <string.h>
using namespace std;
#define lowbit(x) ((x)&(-x))
#define MAXN 1050
typedef int elem_t;
int n,m,t;
struct sum
{
    elem_t a[MAXN][MAXN],c[MAXN][MAXN]ret;
    int m,n,t;
    void init(int i,int j)
    {
        memset(a,0,sizeof(a));
        memset(c,0,sizeof(c));
        m=i,n=j;
    }
    //浙大模版,简练的半天没看懂……
    /*void update(int i,int j,elem_t v){
		for (v-=a[i][j],a[i++][j++]+=v,t=j;i<=m;i+=lowbit(i))
			for (j=t;j<=n;c[i-1][j-1]+=v,j+=lowbit(j));
	}
	elem_t query(int i,int j){
		for (ret=0,t=j;i;i^=lowbit(i))
			for (j=t;j;ret+=c[i-1][j-1],j^=lowbit(j));
		return ret;
	}*/
    void modify(int i, int j, int w)
    {
        while(i<=m)
        {
            t=j;
            while(t<=n)
            {
                a[i][t]+=w;
                t+=lowbit(t);
            }
            i+=lowbit(i);
        }
    }
    elem_t query(int i,int j)
    {
        int ans=0;
        while(i>0)
        {
            t=j;
            while(t>0)
            {
                ans+=a[i][t];
                t-=lowbit(t);
            }
            i-=lowbit(i);
        }
        return ans;
    }
}tree;
int main()
{
    //freopen("input.txt","r",stdin);
    int op;
    while(scanf("%d",&op)!=EOF)
    {
        if(op==0)
        {
            scanf("%d",&n);
            tree.init(n,n);
        }
        else if(op==1)
        {
            int x,y,v;
            scanf("%d%d%d",&x,&y,&v);
            modify(x+1,y+1,v);
        }
        else if(op==2)
        {
            int l,b,r,t;
            scanf("%d%d%d%d",&l,&b,&r,&t);
            printf("%d\n",query(r+1,t+1)-query(r+1,b)-query(l,t+1)+query(l,b));
        }
        else continue;
    }
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值