uva 11992 Fast Matrix Operations 线段树

题意:

给定一个r*c(r<=20,r*c<=1e6)的矩阵,其元素都是0,现在对其子矩阵进行操作。

1 x1 y1 x2 y2 val 表示将(x1,y1,x2,y2)(x1<=x2,y1<=y2)子矩阵中的所有元素加val;

2 x1 y1 x2 y2 val 表示将(x1,y1,x2,y2)(x1<=x2,y1<=y2)子矩阵中的所有元素变为val;

3 x1 y1 x2 y2 val 表示输出(x1,y1,x2,y2)(x1<=x2,y1<=y2)子矩阵中的所有元素的和,最小值和最大值。

共有m次查询(1<=m<=20000)

题解:

由于r<=20,r*c<=1e6,所以我们可以将矩阵展开成一维,每次询问进行(x2-x1)次查询操作即可。这样就转换成了线段树问题。

我们做addv标记加val操作,用setv标记变val操作;若遇到setv,我们需要清空addv,反之则不需要。若在操作的时候setv和addv都存在,那么先执行setv操作,之后执行addv操作。

对于一段区间执行setv操作或者addv操作,我们都能直接得到这段区间的元素和,最小值和最大值,不需要再向下遍历。

pushdown():用作标记的下移。

add():用于加val操作

update():用于变val操作

query():用于查询操作







代码:

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
#include <queue>
#include <stack>
using namespace std;

const int maxn=1e6+10;
struct node{
    int l,r,addv,setv,maxv,minv,sum;
}e[maxn*4];
void build(int a,int b,int c)
{
    e[c].l=a;
    e[c].r=b;
    e[c].addv=e[c].setv=e[c].maxv=e[c].minv=e[c].sum=0;
    if(a==b)return;
    int mid=(a+b)/2;
    build(a,mid,2*c);
    build(mid+1,b,2*c+1);
}
void pushdown(int c)
{
    if(e[c].setv!=0)
    {
        e[2*c].addv=e[2*c+1].addv=0;
        e[2*c].setv=e[2*c+1].setv=e[c].setv;
        e[2*c].maxv=e[2*c+1].maxv=e[c].setv;
        e[2*c].minv=e[2*c+1].minv=e[c].setv;
        e[2*c].sum=(e[2*c].r-e[2*c].l+1)*e[c].setv;
        e[2*c+1].sum=(e[2*c+1].r-e[2*c+1].l+1)*e[c].setv;
        e[c].setv=0;
    }
    if(e[c].addv!=0)
    {
        e[2*c].addv+=e[c].addv;e[2*c+1].addv+=e[c].addv;
        e[2*c].maxv+=e[c].addv;e[2*c+1].maxv+=e[c].addv;
        e[2*c].minv+=e[c].addv;e[2*c+1].minv+=e[c].addv;
        e[2*c].sum+=(e[2*c].r-e[2*c].l+1)*e[c].addv;
        e[2*c+1].sum+=(e[2*c+1].r-e[2*c+1].l+1)*e[c].addv;
        e[c].addv=0;
    }
}
void add(int a,int b,int c,int val)
{
    //printf("%d\n",c);
    if(e[c].l==a&&e[c].r==b)
    {
        e[c].addv+=val;
        e[c].maxv+=val;
        e[c].minv+=val;
        e[c].sum+=(b-a+1)*val;
        return ;
    }
    int mid=(e[c].l+e[c].r)/2;
    pushdown(c);
    if(b<=mid)add(a,b,2*c,val);
    else if(a>mid)add(a,b,2*c+1,val);
    else
    {
        add(a,mid,2*c,val);
        add(mid+1,b,2*c+1,val);
    }
    e[c].maxv=max(e[2*c].maxv,e[2*c+1].maxv);
    e[c].minv=min(e[2*c].minv,e[2*c+1].minv);
    e[c].sum=e[2*c].sum+e[2*c+1].sum;
}
void update(int a,int b,int c,int val)
{
    if(e[c].l==a&&e[c].r==b)
    {
        e[c].addv=0;
        e[c].setv=val;
        e[c].maxv=val;
        e[c].minv=val;
        e[c].sum=(b-a+1)*val;
        return ;
    }
    pushdown(c);
    int mid=(e[c].l+e[c].r)/2;
    if(b<=mid)update(a,b,2*c,val);
    else if(a>mid)update(a,b,2*c+1,val);
    else
    {
        update(a,mid,2*c,val);
        update(mid+1,b,2*c+1,val);
    }
    e[c].maxv=max(e[2*c].maxv,e[2*c+1].maxv);
    e[c].minv=min(e[2*c].minv,e[2*c+1].minv);
    e[c].sum=e[2*c].sum+e[2*c+1].sum;
}
node query(int a,int b,int c)
{
    if(e[c].l==a&&e[c].r==b)return e[c];
    pushdown(c);
    int mid=(e[c].l+e[c].r)/2;
    if(b<=mid)return query(a,b,2*c);
    else if(a>mid)return query(a,b,2*c+1);
    else
    {
        node o,p,q;
        p=query(a,mid,2*c);
        q=query(mid+1,b,2*c+1);
        o.sum=p.sum+q.sum;
        o.maxv=max(p.maxv,q.maxv);
        o.minv=min(p.minv,q.minv);
        return o;
    }
}
int main()
{
    int r,c,m;
    while(scanf("%d%d%d",&r,&c,&m)!=EOF)
    {
        int i,j,k,a,x1,y1,x2,y2,val;
        build(1,r*c,1);
        for(i=0;i<m;i++)
        {
            scanf("%d",&a);
            if(a==1)
            {
                scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&val);
                //printf("*\n");
                for(j=x1;j<=x2;j++)
                add((j-1)*c+y1,(j-1)*c+y2,1,val);
            }
            else if(a==2)
            {
                scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&val);
                for(j=x1;j<=x2;j++)
                update((j-1)*c+y1,(j-1)*c+y2,1,val);
            }
            else
            {
                node ans,p;
                scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                for(j=x1;j<=x2;j++)
                {
                    if(j==x1)ans=query((j-1)*c+y1,(j-1)*c+y2,1);
                    else
                    {
                        p=query((j-1)*c+y1,(j-1)*c+y2,1);
                        ans.sum+=p.sum;
                        ans.maxv=max(ans.maxv,p.maxv);
                        ans.minv=min(ans.minv,p.minv);
                    }
                }
                printf("%d %d %d\n",ans.sum,ans.minv,ans.maxv);
            }
        }
    }
    return 0;
}


  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值