UVA - 11992 Fast Matrix Operations(线段树)

对于一个r*c的矩阵，建r个线段树，分别update和query。

操作都是简单的操作，只不过是叠加在了一个题里，对于add和set的更新要有一定的顺序。

进行set操作，直接把add标记清空。

在pushdown里面，要先下推set标记，在推add。

剩下的就是填代码就好了。

/*************************************
*hardbird's Personal Training        *
*Segment Tree                        *
*UVA - 11992 Fast Matrix Operations  *
*2014.11.22                          *
*************************************/

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long LL;
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
const int maxn=55000;
struct ST
{
    int sum[maxn<<2], add[maxn<<2], col[maxn<<2], max_val[maxn<<2], min_val[maxn<<2];
    void pushup(int rt)
    {
        sum[rt]=sum[rt<<1]+sum[rt<<1|1];
        max_val[rt]=max(max_val[rt<<1], max_val[rt<<1|1]);
        min_val[rt]=min(min_val[rt<<1], min_val[rt<<1|1]);
        add[rt]=0;
        col[rt]=-1;
    }
    void build(int l, int r, int rt)
    {
        sum[rt]=max_val[rt]=min_val[rt]=add[rt]=0;
        col[rt]=-1;
        if(l==r)
            return ;
        int mid=(l+r)>>1;
        build(lson);
        build(rson);
        pushup(rt);
    }
    void pushdown(int rt, int m)
    {
        if(col[rt]!=-1)
        {
            col[rt<<1]=col[rt<<1|1]=col[rt];
            add[rt<<1]=add[rt<<1|1]=0;
            max_val[rt<<1]=min_val[rt<<1]=max_val[rt<<1|1]=min_val[rt<<1|1]=col[rt];
            sum[rt<<1]=(m-(m>>1))*col[rt];
            sum[rt<<1|1]=(m>>1)*col[rt];
            col[rt]=-1;
        }
        if(add[rt]>0)
        {
            add[rt<<1]+=add[rt];add[rt<<1|1]+=add[rt];
            max_val[rt<<1]+=add[rt];min_val[rt<<1]+=add[rt];
            max_val[rt<<1|1]+=add[rt];min_val[rt<<1|1]+=add[rt];
            sum[rt<<1]+=(m-(m>>1))*add[rt];sum[rt<<1|1]+=(m>>1)*add[rt];
            add[rt]=0;
        }
    }
    void update_set(int L, int R, int v, int l, int r, int rt)
    {
        if(L<=l && R>=r)
        {
            col[rt]=v;
            add[rt]=0;
            sum[rt]=v*(r-l+1);
            max_val[rt]=min_val[rt]=v;
            return ;
        }
        pushdown(rt, r-l+1);
        int mid=(l+r)>>1;
        if(L<=mid)update_set(L, R, v, lson);
        if(R>mid)update_set(L, R, v, rson);
        pushup(rt);
    }
    void update_add(int L, int R, int v, int l, int r, int rt)
    {
        if(L<=l && R>=r)
        {
            add[rt]+=v;
            sum[rt]+=v*(r-l+1);
            max_val[rt]+=v;
            min_val[rt]+=v;
            return ;
        }
        pushdown(rt, r-l+1);
        int mid=(l+r)>>1;
        if(L<=mid)update_add(L, R, v, lson);
        if(R>mid)update_add(L, R, v, rson);
        pushup(rt);
    }
    int query_sum(int L, int R, int l, int r, int rt)
    {
        if(L<=l && R>=r)
            return sum[rt];
        pushdown(rt, r-l+1);
        int mid=(l+r)>>1;
        int ans=0;
        if(L<=mid)ans+=query_sum(L, R, lson);
        if(R>mid)ans+=query_sum(L, R, rson);
        return ans;
    }
    int query_max_val(int L, int R, int l, int r, int rt)
    {
        if(L<=l && R>=r)
            return max_val[rt];
        pushdown(rt, r-l+1);
        int mid=(l+r)>>1;
        int ans=0;
        if(L<=mid)ans=max(ans, query_max_val(L, R, lson));
        if(R>mid)ans=max(ans, query_max_val(L, R, rson));
        return ans;
    }
    int query_min_val(int L, int R, int l, int r, int rt)
    {
        if(L<=l && R>=r)
            return min_val[rt];
        pushdown(rt, r-l+1);
        int mid=(l+r)>>1;
        int ans=1e9+10;
        if(L<=mid)ans=min(ans, query_min_val(L, R, lson));
        if(R>mid)ans=min(ans, query_min_val(L, R, rson));
        return ans;
    }
}st[22];

int main()
{
//    freopen("in.txt", "r", stdin);
    int n, m, q;
    while(~scanf("%d%d%d", &m, &n, &q))
    {
        for(int i=1; i<=m; i++)
            st[i].build(1, n, 1);
        while(q--)
        {
            int op, x1, y1, x2, y2;
            int val;
            scanf("%d%d%d%d%d", &op, &x1, &y1, &x2, &y2);
            if(op==1)
            {
                scanf("%d", &val);
                for(int i=x1; i<=x2; i++)
                    st[i].update_add(y1, y2, val, 1, n, 1);
            }
            else if(op==2)
            {
                scanf("%d", &val);
                for(int i=x1; i<=x2; i++)
                    st[i].update_set(y1, y2, val, 1, n, 1);
            }
            else
            {
                int sum=0, max_val=0, min_val=1e9+10;
                for(int i=x1; i<=x2; i++)
                    sum+=st[i].query_sum(y1, y2, 1, n, 1);
                for(int i=x1; i<=x2; i++)
                    max_val=max(max_val, st[i].query_max_val(y1, y2, 1, n, 1));
                for(int i=x1; i<=x2; i++)
                    min_val=min(min_val, st[i].query_min_val(y1, y2, 1, n, 1));
                printf("%d %d %d\n", sum, min_val, max_val);
            }
        }
    }
    return 0;
}