UVA 11992 Fast MatrixOperations(线段树:区间查询+区间赋值+区间增值）

最新推荐文章于 2023-10-30 00:13:16 发布

pxlsdz

最新推荐文章于 2023-10-30 00:13:16 发布

阅读量391

点赞数

分类专栏：数据结构－－线段树

本文链接：https://blog.csdn.net/sdz20172133/article/details/86418401

版权

数据结构－－线段树专栏收录该内容

49 篇文章 0 订阅

订阅专栏

题目描述

PDF

输入输出格式

输入格式：

输出格式：

输入输出样例

输入样例#1： 复制

4 4 8
1 1 2 4 4 5
3 2 1 4 4
1 1 1 3 4 2
3 1 2 4 4
3 1 1 3 4
2 2 1 4 4 2
3 1 2 4 4
1 1 1 4 3 3

输出样例#1： 复制

题意：

题意翻译

有一个r行c列的全0矩阵，有以下三种操作。

1 X1 Y1 X2 Y2 v 子矩阵（X1,Y1,X2,Y2）的元素加v

2 X1 Y1 X2 Y2 v 子矩阵（X1,Y1,X2,Y2）的元素变为v

3 X1 Y1 X2 Y2 查询子矩阵（X1,Y1,X2,Y2）的和，最大值，最小值

子矩阵（X1,Y1,X2,Y2）满足X1<=X<=X2 Y1<=Y<=Y2的所有元素（X1,Y2）。

输入保证和不超过10^9

感谢@Himself65 提供的翻译

分析：线段树区间修改的模板题目，每一行建立线段树，但其实就是把所有元素都放到一起就行，只有一个问题注意：当我们打add（元素加值）标记时是不会影响set（修改值）标记的，但是我们在打set标记时无论你前面add标记是多少，此时就相当于作废，所以直接将add标记赋为0就好了,然后直接修改sum,mi和mx(分别记录该区间的和，最小值，最大值)。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <math.h>
#include <bitset>
#include <algorithm>
#include <climits>
using namespace std;
 
#define lson i<<1
#define rson i<<1|1
#define LS l,mid,lson
#define RS mid+1,r,rson
#define LL long long
#define N 1000005
#define MOD 1000000007
#define INF 0x3f3f3f3f
#define EXP 1e-8
#define lowbit(x) (x&-x)
 
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;
    }
    
        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].add = 0;
    a[i].set = -1;
    if(l == r)     
	{
	a[i].sum = 0;
    a[i].max = 0;
    a[i].min = 0;
    
		return;
	}
    int mid = (l+r)>>1;
    build(LS);
    build(RS);
    pushup(i);
}
//a[l,r]都变为val
void set_data(int l,int r,int i,int val) 
{
    if(a[i].l>=l&&a[i].r<=r)
    {
        a[i].sum = val*(a[i].r-a[i].l+1);
        a[i].min = val;
        a[i].max = val;
        a[i].set = val;
        a[i].add = 0;
        return;
    }
    pushdown(i);         //标记下传
    int mid = (a[i].l+a[i].r)>>1;
    if(l<=mid) set_data(l,r,lson,val);
    if(r>mid) set_data(l,r,rson,val);
    pushup(i);
}
//a[l,r]+=val 
void add_data(int l,int r,int i,int val)
{
    if(a[i].l>=l&&a[i].r<=r)
    {
        a[i].sum += val*(a[i].r-a[i].l+1);
        a[i].min += val;
        a[i].max += val;
        a[i].add += val;
        return;
    }
    pushdown(i);//标记下传
    int mid = (a[i].l+a[i].r)>>1;
    if(l<=mid) add_data(l,r,lson,val);
    if(r>mid) add_data(l,r,rson,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 = (a[i].l+a[i].r)>>1;
    if(l<=mid) query(l,r,lson);
    if(r>mid) query(l,r,rson);
    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);
                    ///修改第i行的对应区间的元素，转变在线段树所对应的元素
            }
            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;
}