UVA 11992 Fast Matrix Operations 线段树

最新推荐文章于 2023-01-14 18:51:17 发布

原创最新推荐文章于 2023-01-14 18:51:17 发布 · 310 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

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

72 篇文章

订阅专栏

本文介绍了解决UVA11992问题的一种方法，该问题涉及在一个r行c列的矩阵上进行m次操作，包括子矩阵元素的增加、设置及查询等操作。通过将二维矩阵一维化，并使用线段树进行区间更新和查询，实现了高效的解题策略。

UVA 11992

有一个r行c列的全0矩阵，查询个数是m，支持一下3种操作，op x1 y1 x2 y2 v，op=1时子矩阵全部加v，op=2时子矩阵全部设为v，op=3时求子矩阵的元素和以及最小值，最大值。

因为题目说了矩阵不超过20行且元素总和不超过1e6，可以把二维化一维，x1 y1到x1 y2可以换成 (x1-1)*c+y1到 (x1-1)*c+y2接下来就是套用刘汝佳的模板了，因为一个细节没注意，wa的我怀疑人生，初始化setv数组要为0 ！！！

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=3*1000000;
const int inf=999999999;
int maxv[maxn],minv[maxn],sumv[maxn],addv[maxn],setv[maxn];
int _max,_min,_sum;
int r,c,q;
void build()
{
	for(int i=0;i<maxn;i++)
	{
		maxv[i]=minv[i]=sumv[i]=addv[i]=0;
		setv[i]=0;  //千万不要再写setv[i]=-1了！！！！！！
	}
}
void pushdown(int o)
{
	int lc=o*2,rc=o*2+1;
	if(setv[o]>=0)
	{
		setv[lc]=setv[rc]=setv[o];
		addv[lc]=addv[rc]=0;
		setv[o]=-1;
	}
	if(addv[o]>0)
	{
		addv[lc]+=addv[o];
		addv[rc]+=addv[o];
		addv[o]=0;
	}
}
void maintain(int o,int L,int R)
{
	int lc=o*2,rc=o*2+1;
	if(L<R)
	{
		sumv[o]=sumv[lc]+sumv[rc];
		minv[o]=min(minv[lc],minv[rc]);
		maxv[o]=max(maxv[lc],maxv[rc]);
	}
	if(setv[o]>=0)
	{
		minv[o]=maxv[o]=setv[o];
		sumv[o]=(R-L+1)*setv[o];
	}
	if(addv[o]>0)
	{
		sumv[o]+=(R-L+1)*addv[o];
		minv[o]+=addv[o];
		maxv[o]+=addv[o];	
	}

}
void updateadd(int L,int R,int o,int ql,int qr,int v)
{
	int lc=o*2,rc=o*2+1;	
	if(L>=ql&&R<=qr)
	addv[o]+=v;
	else
	{
		pushdown(o);
		int m=(L+R)/2;
		if(ql<=m)
		updateadd(L,m,lc,ql,qr,v);
		else 
		maintain(lc,L,m);
		if(qr>m)
		updateadd(m+1,R,rc,ql,qr,v);
		else
		maintain(rc,m+1,R);
	}
	maintain(o,L,R);
}
void updateset(int L,int R,int o,int ql,int qr,int v)
{
	int lc=o*2,rc=o*2+1;
	if(L>=ql&&R<=qr)
	{
		setv[o]=v;
		addv[o]=0;
	}
	else
	{
		pushdown(o);
		int m=(L+R)/2;
		if(ql<=m)
		updateset(L,m,lc,ql,qr,v);
		else
		maintain(lc,L,m);
		if(qr>m)
		updateset(m+1,R,rc,ql,qr,v);
		else
		maintain(rc,m+1,R);
	}
	maintain(o,L,R);
}
void query(int L,int R,int o,int ql,int qr,int add)
{
	if(setv[o]>=0)
	{
		_min=min(_min,setv[o]+add+addv[o]);
		_max=max(_max,setv[o]+add+addv[o]);
		_sum+=(add+addv[o]+setv[o])*(min(R,qr)-max(L,ql)+1);
	}
	else if(L>=ql&&R<=qr)
	{
		_min=min(_min,minv[o]+add);
		_max=max(_max,maxv[o]+add);
		_sum+=(sumv[o]+add*(R-L+1));
	}
	else
	{
		int m=(L+R)/2;
		if(ql<=m)
		query(L,m,o*2,ql,qr,add+addv[o]);
		if(qr>m)
		query(m+1,R,o*2+1,ql,qr,add+addv[o]);
	}
}
int main()
{
	while(~scanf("%d%d%d",&r,&c,&q))
	{
		int i,j,op,x1,y1,x2,y2,v;
		build();
		while(q--)
		{
			scanf("%d",&op);
			if(op==1)
			{
				scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&v);
				for(j=x1;j<=x2;j++)
				updateadd(1,r*c,1,(j-1)*c+y1,(j-1)*c+y2,v);
			}
			else if(op==2)
			{
				scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&v);
				for(j=x1;j<=x2;j++)
				updateset(1,r*c,1,(j-1)*c+y1,(j-1)*c+y2,v);				
			}
			else
			{
				scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
				_min=inf,_max=0,_sum=0;
				for(j=x1;j<=x2;j++)
				query(1,r*c,1,(j-1)*c+y1,(j-1)*c+y2,0);
				printf("%d %d %d\n",_sum,_min,_max);
			}
		}
	}
}

接下来贴一个更简洁的模板，留着备用

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=4e5+10;
const int inf=1000000000;
int _min,_max,_sum;
int ql,qr;
struct node
{
	int addv[maxn],setv[maxn],minv[maxn],maxv[maxn],sumv[maxn];
	void build()
	{
		for(int i=0;i<maxn;i++)
		{
			addv[i]=minv[i]=maxv[i]=sumv[i]=0;
			setv[i]=0;
		}
	}
	void pushdown(int o)
	{
		int lc=o*2,rc=o*2+1;
		if(setv[o]>=0)
		{
			setv[lc]=setv[rc]=setv[o];
			addv[lc]=addv[rc]=0;
			setv[o]=-1;
		}
		if(addv[o]>0)
		{
			addv[lc]+=addv[o];
			addv[rc]+=addv[o];
			addv[o]=0;
		}
	}
	void maintain(int o,int L,int R)
	{
		int lc=o*2,rc=o*2+1;
		if(L<R)
		{
			sumv[o]=sumv[lc]+sumv[rc];
			minv[o]=min(minv[lc],minv[rc]);
			maxv[o]=max(maxv[lc],maxv[rc]);
		}
		if(setv[o]>=0)
		{
			sumv[o]=setv[o]*(R-L+1);
			minv[o]=maxv[o]=setv[o];
		}
		if(addv[o]>0)
		{
			sumv[o]+=(addv[o]*(R-L+1));
			maxv[o]+=addv[o];
			minv[o]+=addv[o];
		}
	}
	void adddata(int L,int R,int o,int v)
	{
		int lc=o*2,rc=o*2+1;
		if(L>=ql&&R<=qr)
		addv[o]+=v;
		else
		{
			pushdown(o);
			int m=L+(R-L)/2;
			if(ql<=m)
			adddata(L,m,lc,v);
			else
			maintain(lc,L,m);
			if(qr>m)
			adddata(m+1,R,rc,v);
			else
			maintain(rc,m+1,R);
		}
		maintain(o,L,R);
	}
	void setdata(int L,int R,int o,int v)
	{
		int lc=o*2,rc=o*2+1;
		if(L>=ql&&R<=qr)
		{
			setv[o]=v;
			addv[o]=0;
		}
		else
		{
			pushdown(o);
			int m=L+(R-L)/2;
			if(ql<=m)
			setdata(L,m,lc,v);
			else
			maintain(lc,L,m);
			if(qr>m)
			setdata(m+1,R,rc,v);
			else
			maintain(rc,m+1,R);
		}
		maintain(o,L,R);
	}
	void query(int L,int R,int o,int add)
	{
		if(setv[o]>=0)
		{
			int v=add+addv[o]+setv[o];
			_min=min(_min,v);
			_max=max(_max,v);
			_sum+=(v*(min(R,qr)-max(L,ql)+1));
		}
		else if(L>=ql&&R<=qr)
		{
			_min=min(_min,minv[o]+add);
			_max=max(_max,maxv[o]+add);
			_sum+=(sumv[o]+add*(R-L+1));
		}
		else
		{
			int m=L+(R-L)/2;
			int lc=o*2,rc=o*2+1;
			if(ql<=m)
			query(L,m,lc,add+addv[o]);
			if(qr>m)
			query(m+1,R,rc,add+addv[o]);
		}
	}	
}a[25];
int main()
{
	int r,c,m;
	while(~scanf("%d%d%d",&r,&c,&m))
	{
		int i,x1,x2,v,op;
        for(i = 1; i <= r;++i)
        a[i].build();
		while(m--)
		{
			scanf("%d%d%d%d%d",&op,&x1,&ql,&x2,&qr);
			if(op==1||op==2)
			{
				scanf("%d",&v);
				for(i=x1;i<=x2;i++)
				if(op==1)
				a[i].adddata(1,c,1,v);
				else
				a[i].setdata(1,c,1,v);
			}
			else
			{
				_min=inf,_max=0,_sum=0;
				for(i=x1;i<=x2;i++)
				a[i].query(1,c,1,0);
				printf("%d %d %d\n",_sum,_min,_max);
			}
		}
	}
}