HDU 3255 Farming(线段树:扫描线)
http://acm.hdu.edu.cn/showproblem.php?pid=3255
题意:多个可能重合的长方体求体积.
分析:
该题只要把种子的价值转化为Z轴坐标[0,pi],就可以看成了重叠长方体就总体积的问题.解法类似HDU3642:
http://blog.csdn.net/u013480600/article/details/22625139
由于这里x与y的坐标范围都很大,所以离散化处理x坐标,然后扫描线以y为高.
Z轴可以离散化,也可以不离散化,这里我们离散化.总体的流程就是:读取输入数据,把每个农田转化为一个立方体,从小到大枚举Z轴坐标,然后遍历一遍所有包含当前Z轴坐标的立方体生成对应的两条扫描线.然后根据这些扫描线求出Z[i]到Z[i+1]之间的矩形面积,乘以Z[i+1]-Z[i]就是这部分的体积了.
线段树采用线信息保留型线段树,维护的信息有:
cnt:当前下位边-上位边数目的结果
sum:当前被覆盖的总长度.
只需要PushUp和build就行.
AC代码:1734ms,一次AC
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int MAXN=60000+1000;
#define lson i*2,l,m
#define rson i*2+1,m+1,r
struct seg
{
int l,r,h,d;
seg(){};
seg(int l,int r,int h,int d):l(l),r(r),h(h),d(d){}
bool operator <(const seg &b)const
{
return h<b.h;
}
}ss[MAXN];
int cnt[MAXN*4],sum[MAXN*4];
int X[MAXN],p[10];
struct point
{
int x,y,z;
point(){}
point(int x,int y,int z):x(x),y(y),z(z){}
};
struct cub
{
point a,b;
}cube[MAXN/2];
void PushUp(int i,int l,int r)
{
if(cnt[i])
{
sum[i]=X[r+1]-X[l];
}
else if(l==r)
sum[i]=0;
else
sum[i]=sum[i*2]+sum[i*2+1];
}
void update(int ql,int qr,int v,int i,int l,int r)
{
if(ql<=l && r<=qr)
{
cnt[i]+=v;
PushUp(i,l,r);
return ;
}
int m=(r+l)>>1;
if(ql<=m) update(ql,qr,v,lson);
if(m<qr) update(ql,qr,v,rson);
PushUp(i,l,r);
}
int main()
{
int T;
scanf("%d",&T);
for(int kase=1;kase<=T;kase++)
{
int n,m;
scanf("%d%d",&n,&m);
p[0]=0;//表示Z轴的第0个不同的坐标
for(int i=1;i<=m;i++)
scanf("%d",&p[i]);
int cntx=0;
for(int i=1;i<=n;i++)
{
int x1,y1,x2,y2,s;
scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&s);
cube[i].a=point(x1,y1,0);
cube[i].b=point(x2,y2,p[s]);
X[cntx++]=x1;
X[cntx++]=x2;
}
sort(p,p+m+1);
sort(X,X+cntx);
cntx=unique(X,X+cntx)-X;
long long ans=0;
for(int k=0;k<m;k++)
{
int cnt_ss=0;//扫描线的条数
for(int i=1;i<=n;i++)if(cube[i].a.z<=p[k] && cube[i].b.z>p[k])
{
ss[cnt_ss++]=seg(cube[i].a.x,cube[i].b.x,cube[i].a.y,1);
ss[cnt_ss++]=seg(cube[i].a.x,cube[i].b.x,cube[i].b.y,-1);
}
sort(ss,ss+cnt_ss);
memset(cnt,0,sizeof(cnt));
memset(sum,0,sizeof(sum));
long long res=0;
for(int i=0;i<cnt_ss-1;i++)
{
int ql=lower_bound(X,X+cntx,ss[i].l)-X;
int qr=lower_bound(X,X+cntx,ss[i].r)-X-1;
if(ql<=qr) update(ql,qr,ss[i].d,1,0,cntx-1);
res += (long long)sum[1]*(ss[i+1].h-ss[i].h);//求面积
}
ans += res *(p[k+1]-p[k]);//求体积
}
printf("Case %d: %I64d\n",kase,ans);
}
return 0;
}