暑假集训中期测试 Problem D: 装箱问题2 （并查集）

Description

有很多个棱长为1的正方体货物整齐地堆在一堆。不过有一些是悬空的，大概是粘上去的吧。。。

给出这些货物的相邻关系，求最小的长方体（或正方体）能装下这些货物的集装箱的体积，（集装箱棱长方向与这些正方体三个棱方向平行）。

Input

每组数据第一行一个n，表示有n个货物。1 <= n <= 1000

接下来n行每行6个数，第i行表示以棱方向为轴的x轴正负、y轴正负、z轴正负方向与第i个货物相邻的货物编号（i为1~n），0表示无该位置信息。

Output

如果描述出现矛盾或者无法确定货物是堆在一起的，输出"What?"，否则输出集装箱体积。

Sample Input

0 0 0 0 3 2

0 0 0 0 1 0

0 0 0 0 0 1

Sample Output

View Code

#include <stdio.h>
#define N 1005
#define MIN(a,b) ((a)<(b)?(a):(b))
#define MAX(a,b) ((a)>(b)?(a):(b))
#define INF 0x3fffffff
int x[N],y[N],z[N];
int p[N];
int n;
void make_set()
{for(int i=1;i<=n;i++)   p[i]=i,x[i]=y[i]=z[i]=0;
}
int find_set(int i)
{int pi=p[i];if(i^p[i])  p[i]=find_set(p[i]);if(pi^p[i]){x[i]+=x[pi];y[i]+=y[pi];z[i]+=z[pi];}return p[i];
}
void union_set(int i,int j,int dir,int d)
{int pi,pj;pi=find_set(i);pj=find_set(j);p[pj]=pi;//dir为0 表示 i与j在x方向上相邻，d=1表示j在i右边if(dir==0){x[pj]=x[i]-x[j]+d;y[pj]=y[i]-y[j];z[pj]=z[i]-z[j];}else if(dir==1){x[pj]=x[i]-x[j];y[pj]=y[i]-y[j]+d;z[pj]=z[i]-z[j];}else{x[pj]=x[i]-x[j];y[pj]=y[i]-y[j];z[pj]=z[i]-z[j]+d;}
}
int main()
{int i,j;bool wa;while(~scanf("%d",&n)){make_set();wa=false;for(i=1;i<=n;i++){for(int dir=0;dir<3;dir++){for(int d=1;d>=-1;d-=2){scanf("%d",&j);if(j==0)    continue;if(find_set(i)^find_set(j)) union_set(i,j,dir,d);else{if(dir==0 && !(y[i]==y[j]&&z[i]==z[j]&&d==x[j]-x[i]))  wa=true;if(dir==1 && !(x[i]==x[j]&&z[i]==z[j]&&d==y[j]-y[i]))  wa=true;if(dir==2 && !(y[i]==y[j]&&x[i]==x[j]&&d==z[j]-z[i]))  wa=true;}}}}for(i=2;!wa && i<=n;i++)   if(find_set(i)!=find_set(1))    wa=true;int minx=INF,maxx=-INF;int miny=INF,maxy=-INF;int minz=INF,maxz=-INF;if(wa)  puts("What?");else{for(int i=1;i<=n;i++){minx=MIN(minx,x[i]);    maxx=MAX(maxx,x[i]);miny=MIN(miny,y[i]);    maxy=MAX(maxy,y[i]);minz=MIN(minz,z[i]);    maxz=MAX(maxz,z[i]);}printf("%d\n",(maxx-minx+1)*(maxy-miny+1)*(maxz-minz+1));}}return 0;
}