codeforces 390E Inna and Large Sweet Matrix

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

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

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

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

详情请见博客。

#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

 

转载于:https://www.cnblogs.com/yours1103/p/3562537.html