您的位置:首页 > 大数据 > 人工智能

HDU-4819-Mosaic

2014-11-11 16:11 309 查看
这个题需要求二维区间的最大值最小值,并完成单点更新,这就应该说二维线段树进行解决,对二维线段树的处理方面还不够熟练吧,写的有点复杂~

代码:

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=810;
struct SNode
{
int l;
int r;
int smax;
int smin;
int getmid()
{
return (l+r)>>1;
}
};
struct Node
{
int l;
int r;
SNode t[maxn*4];
int getmid()
{
return (l+r)>>1;
}
}t[maxn*4];
int n,m,a[maxn][maxn];
void BuildY(int l,int r,int index,int p,int flag)
{
t[p].t[index].l=l;
t[p].t[index].r=r;
if(l==r)
{
if(flag)
{
t[p].t[index].smax=max(t[p<<1].t[index].smax,t[p<<1|1].t[index].smax);
t[p].t[index].smin=min(t[p<<1].t[index].smin,t[p<<1|1].t[index].smin);
}
else
t[p].t[index].smin=t[p].t[index].smax=a[t[p].l][l];
return;
}
int mid=(l+r)>>1;
BuildY(l,mid,index<<1,p,flag);
BuildY(mid+1,r,index<<1|1,p,flag);
t[p].t[index].smax=max(t[p].t[index<<1].smax,t[p].t[index<<1|1].smax);
t[p].t[index].smin=min(t[p].t[index<<1].smin,t[p].t[index<<1|1].smin);
}
void BuildX(int l,int r,int index)
{
t[index].l=l;
t[index].r=r;
if(l==r)
{
BuildY(1,n,1,index,0);
return;
}
int mid=t[index].getmid();
BuildX(l,mid,index<<1);
BuildX(mid+1,r,index<<1|1);
BuildY(1,n,1,index,1);
}
int QueryMaxY(int l,int r,int index,int p)
{
if(t[p].t[index].l==l&&t[p].t[index].r==r)
return t[p].t[index].smax;
int mid=t[p].t[index].getmid();
if(r<=mid)
return QueryMaxY(l,r,index<<1,p);
else if(l>mid)
return QueryMaxY(l,r,index<<1|1,p);
else
return max(QueryMaxY(l,mid,index<<1,p),QueryMaxY(mid+1,r,index<<1|1,p));
}
int QueryMaxX(int l,int r,int x,int y,int index)
{
if(t[index].l==l&&t[index].r==r)
return QueryMaxY(x,y,1,index);
int mid=(t[index].l+t[index].r)>>1;
if(r<=mid)
return QueryMaxX(l,r,x,y,index<<1);
else if(l>mid)
return QueryMaxX(l,r,x,y,index<<1|1);
else
return max(QueryMaxX(l,mid,x,y,index<<1),QueryMaxX(mid+1,r,x,y,index<<1|1));
}
int QueryMinY(int l,int r,int index,int p)
{
if(t[index].l==l&&t[index].r==r)
return t[p].t[index].smin;
int mid=(t[p].t[index].l+t[p].t[index].r)>>1;
if(r<=mid)
return QueryMinY(l,r,index<<1,p);
else if(l>mid)
return QueryMinY(l,r,index<<1|1,p);
else
return min(QueryMinY(l,mid,index<<1,p),QueryMinY(mid+1,r,index<<1|1,p));
}
int QueryMinX(int l,int r,int x,int y,int index)
{
if(t[index].l==l&&t[index].r==r)
return QueryMinY(x,y,1,index);
int mid=(t[index].l+t[index].r)>>1;
if(r<=mid)
return QueryMinX(l,r,x,y,index<<1);
else if(l>mid)
return QueryMinX(l,r,x,y,index<<1|1);
else
return min(QueryMinX(l,mid,x,y,index<<1),QueryMinX(mid+1,r,x,y,index<<1|1));
}
void ModifyY(int y,int index,int val,int p,int flag)
{
if(t[p].t[index].l==t[p].t[index].r)
{
if(!flag)
t[p].t[index].smin=t[p].t[index].smax=val;
else
{
t[p].t[index].smin=min(t[p<<1].t[index].smin,t[p<<1|1].t[index].smin);
t[p].t[index].smax=max(t[p<<1].t[index].smax,t[p<<1|1].t[index].smax);
}
return;
}
int mid=(t[p].t[index].l+t[p].t[index].r)>>1;
if(y<=mid)
ModifyY(y,index<<1,val,p,flag);
else
ModifyY(y,index<<1|1,val,p,flag);
t[p].t[index].smax=max(t[p].t[index<<1].smax,t[p].t[index<<1|1].smax);
t[p].t[index].smin=min(t[p].t[index<<1].smin,t[p].t[index<<1|1].smin);
}
void ModifyX(int x,int y,int index,int val)
{
if(t[index].l==t[index].r)
{
ModifyY(y,1,val,index,0);
return;
}
int mid=(t[index].l+t[index].r)>>1;
if(x<=mid)
ModifyX(x,y,index<<1,val);
else
ModifyX(x,y,index<<1|1,val);
ModifyY(y,1,val,index,1);
}
int main()
{
int T,cas=1;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
scanf("%d",&a[i][j]);
BuildX(1,n,1);
scanf("%d",&m);
printf("Case #%d:\n",cas++);
while(m--)
{
int x,y,l;
scanf("%d%d%d",&x,&y,&l);
int sx=max(1,x-l/2);
int sy=max(1,y-l/2);
int mx=min(n,x+l/2);
int my=min(n,y+l/2);
int smax=QueryMaxX(sx,mx,sy,my,1);
int smin=QueryMinX(sx,mx,sy,my,1);
printf("%d\n",(smax+smin)>>1);
ModifyX(x,y,1,(smax+smin)>>1);
}
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: