Fast Matrix Operations UVA 11992-线段树

这个题属于线段树的一个入门题，对线段树了解之后可以用这个题练一练

题意：

有一个初始状态全为0的矩阵，一共有三个操作

1 x1 y1 x2 y2 v：子矩阵(x1,y1,x2,y2)所有元素增加v                                          add

2 x1 y1 x2 y2 v：子矩阵(x1,y1,x2,y2)所有元素设为v                                          set

3 x1 y1 x2 y2 v：查询子矩阵(x1,y1,x2,y2)的元素和，最大值和最小值               query
 

解析

题目的意思也非常的明确，知道线段树的看到这道题应该都能直接想到线段树，理解起来确实没什么难度

题目可能又两个地方理解起来稍微有点困难

1.我们常见的线段树都是一维的，这里给定的是一个矩阵，怎么办

2.我们都知道线段树又延迟更新策略，但是对于这道题来说，两种操作之间，延迟更新会相互影响，应该怎么办？

 

第一个问题：

这个有两个方法来进行解决。第一种，每一行构建一个线段树，然后进行操作。第二种，把矩阵转换为一元组，直接构建一个线段树。

第二个问题:

我们要理解清楚set和add两个操作之间的相互影响关系，简单分析一下。对于一个区间来说，当前被标记了add延迟更新，后进标记了set延迟。那么前一个add操作来说，他其实算是一个无效的操作了，我们需要取消这个add延迟标记，这一点很好理解。如果先进行了set延迟标记，后进行了add延迟标记，那么这个set标记就不能被忽略了，因为add操作不能让前一个set操作无效。

那么在标记传递的过程中，两个标记的关系应该如何处理呢？

首先我们应该知道如果某一个区间节点存在set延迟标记和add延迟标记，那么可以肯定的是，add延迟标记一定是在set延迟标记之后被标记上的，所以我们在标记传递的过程中，应该先去处理set标记，然后再处理add标记。且在处理set延迟标记的时候，如果当前区间的两个子区间存在add延迟标记更新的话，这个add延迟标记也应该被取消，因为我们可以肯定这个add延迟标记是在set延迟标记之前被标记上的。你问为什么能肯定？因为我们在add的过程中也会去进行标记的传递，所以如果他是后面被标记上的，那么他的父节点不可能会存在set延迟标记，因为该标记一定会被传递。

 

代码实现，不得不说线段树的题代码真的是太长了，但其实理解线段树之后看起来就没那么困难了

#include <stdio.h>
#define lson 2*i
#define rson 2*i+1
#define LS l,mid,lson
#define RS mid+1,r,rson
#define N 1000005
#define INF 0x3f3f3f3f
#define max(x,y) x > y ? x : y
#define min(x,y) x < y ? x : y
#define mid(l,r) l + ((r - l)>>1)

int ans_sum,ans_max,ans_min;
struct node
{
    int l,r;
    int sum,max,min;
    int add,set;
} a[N<<2];

void pushdown(int i)
{
    if(a[i].set!=-1)
    {
        a[lson].set = a[rson].set = a[i].set;
        a[lson].add = a[rson].add = 0;
        a[lson].min = a[rson].min = a[i].set;
        a[lson].max = a[rson].max = a[i].set;
        a[lson].sum = (a[lson].r-a[lson].l+1)*a[i].set;
        a[rson].sum = (a[rson].r-a[rson].l+1)*a[i].set;
        a[i].set = -1;
    }
    if(a[i].add>0)
    {
        a[lson].add+=a[i].add;
        a[rson].add+=a[i].add;
        a[lson].min+=a[i].add;
        a[rson].min+=a[i].add;
        a[lson].max+=a[i].add;
        a[rson].max+=a[i].add;
        a[lson].sum+=a[i].add*(a[lson].r-a[lson].l+1);
        a[rson].sum+=a[i].add*(a[rson].r-a[rson].l+1);
        a[i].add = 0;
    }
}

void pushup(int i)
{
    a[i].sum=a[lson].sum+a[rson].sum;
    a[i].max=max(a[lson].max,a[rson].max);
    a[i].min=min(a[lson].min,a[rson].min);
}

void build(int l,int r,int i)
{
    a[i].l = l;
    a[i].r = r;
    a[i].sum = 0;
    a[i].max = 0;
    a[i].min = 0;
    a[i].add = 0;
    a[i].set = -1;
    if(l == r) return;
    int mid = mid(l,r);
    build(LS);
    build(RS);
}

void set_data(int l,int r,int i,int val)
{
    if(a[i].l==l&&a[i].r==r)
    {
        a[i].sum = val*(r-l+1);
        a[i].min = val;
        a[i].max = val;
        a[i].set = val;
        a[i].add = 0;
        return;
    }
    pushdown(i);
    int mid = mid(a[i].l,a[i].r);
    if(r<=mid)
        set_data(l,r,lson,val);
    else if(l>mid)
        set_data(l,r,rson,val);
    else
    {
        set_data(LS,val);
        set_data(RS,val);
    }
    pushup(i);
}

void add_data(int l,int r,int i,int val)
{
    if(a[i].l==l&&a[i].r==r)
    {
        a[i].sum += val*(r-l+1);
        a[i].min += val;
        a[i].max += val;
        a[i].add += val;
        return;
    }
    pushdown(i);
    int mid = mid(a[i].l,a[i].r);
    if(r<=mid)
        add_data(l,r,lson,val);
    else if(l>mid)
        add_data(l,r,rson,val);
    else
    {
        add_data(LS,val);
        add_data(RS,val);
    }
    pushup(i);
}

void query(int l,int r,int i)
{
    if(l == a[i].l && a[i].r == r)
    {
        ans_sum += a[i].sum;
        ans_max = max(ans_max,a[i].max);
        ans_min = min(ans_min,a[i].min);
        return ;
    }
    pushdown(i);
    int mid = mid(a[i].l,a[i].r);
    if(r<=mid)
        query(l,r,lson);
    else if(l>mid)
        query(l,r,rson);
    else
    {
        query(LS);
        query(RS);
    }
    pushup(i);
}

int main()
{
    int op,i,j,x1,x2,y1,y2,v;
    int r,c,m;
    while(~scanf("%d%d%d",&r,&c,&m))
    {
        build(1,r*c,1);
        while(m--)
        {
            scanf("%d",&op);
            if(op==1)
            {
                scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&v);
                for(i = x1; i<=x2; i++)
                    add_data((i-1)*c+y1,(i-1)*c+y2,1,v);
            }
            else if(op==2)
            {
                scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&v);
                for(i = x1; i<=x2; i++)
                    set_data((i-1)*c+y1,(i-1)*c+y2,1,v);
            }
            else
            {
                scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                ans_sum = 0;
                ans_max = -INF;
                ans_min = INF;
                for(i = x1; i<=x2; i++)
                    query((i-1)*c+y1,(i-1)*c+y2,1);
                printf("%d %d %d\n",ans_sum,ans_min,ans_max);
            }
        }
    }
    return 0;
}