首页 > 其他 > 详细

codeforces 390E Inna and Large Sweet Matrix

时间:2014-02-24 19:55:07      阅读:390      评论:0      收藏:0      [点我收藏+]

本题的主要算法就是区间更新和区间求和;

可以用线段树和树状数组来做;

感觉线段树写的太麻烦了,看到官方题解上说可以用树状数组做,觉得很神奇,以前用过的树状数组都是单点维护,区间求和的;

其实树状数组还可以区间维护,单点求值;和区间维护,区间求和的;

详情请见博客

bubuko.com,布布扣
#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#define maxn 4000010
#define ll long long
using namespace std;

ll a[2][maxn];
ll b[2][maxn];

void add_a(int flag,int x,ll value)
{
    while(x>0)
    {
        a[flag][x]+=value;
        x-=x&(-x);
    }
}

void add_b(int flag,int n,int x,ll value)
{
    for(int i=x;i<=n;i+=i&(-i))
        b[flag][i]+=x*value;
}

ll sum_a(int flag,int n,int x)
{
    ll sum=0;
    while(x<=n)
    {
        sum+=a[flag][x];
        x+=x&(-x);
    }
    return sum;
}

ll sum_b(int flag,int x)
{
    ll sum=0;
    while(x>0)
    {
        sum+=b[flag][x];
        x-=x&(-x);
    }
    return sum;
}

ll getsum(int flag,int n,int x)
{
    if(x)
        return sum_a(flag,n,x)*x+sum_b(flag,x-1);
    else
    {
        return 0;
    }
}

int main()
{
    int n,m,w;
    int comand;
    int x1,x2,y1,y2;
    int v;
    scanf("%d%d%d",&n,&m,&w);
    for(int i=0; i<w; i++)
    {
        scanf("%d",&comand);
        if(comand==0)
        {
            scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&v);
            ll vy=v*(y2-y1+1);
            ll vx=v*(x2-x1+1);
            add_a(0,x2,vy);
            add_b(0,n,x2,vy);
            if(x1>1)
            {
                add_a(0,x1-1,-vy);
                add_b(0,n,x1-1,-vy);
            }
            add_a(1,y2,vx);
            add_b(1,m,y2,vx);
            if(y1>1)
            {
                add_a(1,y1-1,-vx);
                add_b(1,m,y1-1,-vx);
            }
        }
        else
        {
            scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
            ll tmp1=getsum(0,n,x2)-getsum(0,n,x1-1);
            ll tmp2=getsum(1,m,y1-1);
            ll tmp3=getsum(1,m,m)-getsum(1,m,y2);
            cout<<tmp1-tmp2-tmp3<<endl;
        }
    }
    return 0;
}
View Code

codeforces 390E Inna and Large Sweet Matrix

原文:http://www.cnblogs.com/yours1103/p/3562537.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!