poj 1984(带权并查集)
题目链接：http://poj.org/problem?id=1984
题意：输入N表示村庄，M表示命令条数。接下来有M条命令，输入x，y，z，e
分别表示村庄x到村庄y的距离为z，e表示东西南北的方向。
接下来输入k，表示有k个询问。输入k行命令，分别有x，y，z。
求村庄x与村庄y在第z条命令后的曼哈顿距离。
经典的带权并查集的向量转移操作。
关键是要离线操作。
emmm，刚开始的时候方向搞错了。用自己的代码a题就是爽。。
1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 using namespace std; 5 const int maxn=40006; 6 int fa[maxn],north[maxn],east[maxn]; 7 int ans[maxn]; 8 struct node{ 9 int a,b,c,d; 10 }seek[maxn]; 11 struct edge{ 12 int id,x,y,z; 13 }innum[maxn]; 14 bool cmp(edge a, edge b) 15 { 16 return a.z<b.z; 17 } 18 void init() 19 { 20 for(int i=0;i<maxn;i++){ 21 fa[i]=i; 22 north[i]=0; 23 east[i]=0; 24 } 25 } 26 int find_fa(int x) 27 { 28 if(fa[x]==x) return x; 29 int tmp=fa[x]; 30 fa[x]=find_fa(tmp); 31 north[x]=north[x]+north[tmp]; 32 east[x]=east[x]+east[tmp]; 33 return fa[x]; 34 } 35 int sqrt(int x) 36 { 37 if(x<0) return -x; 38 return x; 39 } 40 41 int main() 42 { 43 int n,m; 44 while( ~scanf("%d%d",&n,&m)){ 45 init(); 46 for(int i=1;i<=m;i++){ 47 char str[10]; 48 scanf("%d%d%d%s",&seek[i].a,&seek[i].b,&seek[i].c,str); 49 if(str[0]=='E') seek[i].d=0; 50 if(str[0]=='N') seek[i].d=1; 51 if(str[0]=='W') seek[i].d=2; 52 if(str[0]=='S') seek[i].d=3; 53 } 54 int k; 55 scanf("%d",&k); 56 for(int i=1;i<=k;i++){ 57 scanf("%d%d%d",&innum[i].x,&innum[i].y,&innum[i].z); 58 innum[i].id=i; 59 } 60 sort( innum+1, innum+1+k, cmp); 61 int now=1; 62 for(int i=1;i<=k;i++){ 63 while(now<=innum[i].z){ 64 int iop,kop,tmp,dic; 65 iop=seek[now].a; 66 kop=seek[now].b; 67 tmp=seek[now].c; 68 dic=seek[now].d; 69 int roota=find_fa(iop); 70 int rootb=find_fa(kop); 71 if(roota!=rootb){ 72 fa[rootb]=roota; 73 if(dic==0){ 74 east[rootb]=east[iop]-east[kop]+tmp; 75 north[rootb]=north[iop]-north[kop]; 76 } else if(dic==1){ 77 east[rootb]=east[iop]-east[kop]; 78 north[rootb]=north[iop]-north[kop]+tmp; 79 } else if(dic==2){ 80 east[rootb]=east[iop]-east[kop]-tmp; 81 north[rootb]=north[iop]-north[kop]; 82 } else if(dic==3){ 83 east[rootb]=east[iop]-east[kop]; 84 north[rootb]=north[iop]-north[kop]-tmp; 85 } 86 } 87 now++; 88 } 89 int iop,kop; 90 iop=innum[i].x; kop=innum[i].y; 91 int roota=find_fa(iop); 92 int rootb=find_fa(kop); 93 if(roota!=rootb){ 94 ans[innum[i].id]=-1; 95 } else{ 96 int x1,y1; 97 x1=sqrt(north[iop]-north[kop]); 98 y1=sqrt(east[iop]-east[kop]); 99 ans[innum[i].id]=x1+y1; 100 } 101 } 102 103 for(int i=1;i<=k;i++){ 104 printf("%d\n",ans[i]); 105 } 106 } 107 return 0; 108 }
