UVA - 11992——Fast Matrix Operations（线段树）

题目链接：https://cn.vjudge.net/problem/UVA-11992

思路：把二维当成一维，每次查询或者操作x1,y1,x2,y3区域的时候都分成处理x2-x1+1个小段，每一段处理y1~y2这个区间。

#include<string.h>
#include<stdio.h>
#include<algorithm>
using namespace std;
struct node
{
    int l,r,v,f,se;
    int minn,maxx;
} A[4000010];
int r,c,m,ans,ansn,ansx;
void down(int k)
{
    if(A[k].se)
    {
        A[2*k].maxx=A[k].se;
        A[2*k].minn=A[k].se;
        A[2*k+1].maxx=A[k].se;
        A[2*k+1].minn=A[k].se;
        A[2*k].v=(A[2*k].r-A[2*k].l+1)*A[k].se;
        A[2*k+1].v=(A[2*k+1].r-A[2*k+1].l+1)*A[k].se;
        A[2*k].se=A[2*k+1].se=A[k].se;
        A[2*k].f=A[2*k+1].f=0;
        A[k].se=0;
    }
    if(A[k].f)
    {
        A[2*k].maxx+=A[k].f;
        A[2*k].minn+=A[k].f;
        A[2*k+1].minn+=A[k].f;
        A[2*k+1].maxx+=A[k].f;
        A[2*k+1].v+=(A[2*k+1].r-A[2*k+1].l+1)*A[k].f;
        A[2*k].v+=(A[2*k].r-A[2*k].l+1)*A[k].f;
        A[2*k].f+=A[k].f;
        A[2*k+1].f+=A[k].f;
        A[k].f=0;
    }
}
void init_tire(int l,int r,int k)
{
    A[k].l=l,A[k].r=r,A[k].f=0,A[k].se=0;
    if(l==r)
    {
        A[k].v=0;
        A[k].minn=0;
        A[k].maxx=0;
        return ;
    }
    int mid=(l+r)>>1;
    init_tire(l,mid,k*2);
    init_tire(mid+1,r,k*2+1);
    A[k].v=A[2*k].v+A[2*k+1].v;
    A[k].minn=min(A[2*k].minn,A[2*k+1].minn);
    A[k].maxx=max(A[2*k].maxx,A[2*k+1].maxx);
}
void add(int l,int r,int x,int q,int k)
{
    if(A[k].l>=l&&A[k].r<=r)
    {
        if(q==1)
        {
            A[k].v+=x*(A[k].r-A[k].l+1);
            A[k].f+=x;
            A[k].maxx+=x;
            A[k].minn+=x;
            return ;
        }
        if(q==2)
        {

            A[k].f=0;
            A[k].se=x;
            A[k].maxx=x;
            A[k].minn=x;
            A[k].v=x*(A[k].r-A[k].l+1);
            return ;
        }
    }
    down(k);
    int mid=(A[k].l+A[k].r)>>1;
    if(r<=mid) add(l,r,x,q,2*k);
    else if(l>mid) add(l,r,x,q,2*k+1);
    else
    {
        add(l,r,x,q,2*k);
        add(l,r,x,q,2*k+1);
    }
    A[k].v=A[2*k].v+A[2*k+1].v;
    A[k].minn=min(A[2*k].minn,A[2*k+1].minn);
    A[k].maxx=max(A[2*k].maxx,A[2*k+1].maxx);
}
void find_tire(int l,int r,int k)
{
    if(l<=A[k].l&&A[k].r<=r)
    {
//        printf("***%d %d %d\n",A[k].l,A[k].r,A[k].maxx);
        ans+=A[k].v;
        ansn=min(ansn,A[k].minn);
        ansx=max(ansx,A[k].maxx);
        return ;
    }
    down(k);
    int mid=(A[k].l+A[k].r)>>1;
    if(r<=mid) find_tire(l,r,2*k);
    else if(l>mid) find_tire(l,r,2*k+1);
    else
    {
        find_tire(l,r,2*k);
        find_tire(l,r,2*k+1);
    }
}
int main()
{
    int q,x1,x2,y1,y2,x;
    while(~scanf("%d%d%d",&r,&c,&m))
    {
        init_tire(1,r*c,1);
        while(m--)
        {
            scanf("%d",&q);
            if(q==1||q==2)
            {
                scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&x);
                for(int i=x1; i<=x2; i++)
                {
                    add((i-1)*c+y1,(i-1)*c+y2,x,q,1);
                }
            }
            else
            {
                scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                ans=ansx=0;
                ansn=0x3f3f3f3f;
                for(int i=x1; i<=x2; i++)
                {
                    find_tire((i-1)*c+y1,(i-1)*c+y2,1);
                }
                printf("%d %d %d\n",ans,ansn,ansx);
            }
        }
    }
}