bzoj 3362: [Usaco2004 Feb]Navigation Nightmare 导航噩梦 带权并查集

→题目链接←

【想说的话】

一道带权并查集好(水)题

以前觉得带权并查集挺屌的...

但是写过才发现就是正常的并查集顺便维护点什么东西...

好像我写的有点暴力啊...跑的好慢啊

【题解】

做并查集的过程中找根时不做路径压缩

然后对于每个点保存它到它的父亲节点需要在x轴和y轴上移动多少

每次合并x、y时找到y的根并记录从y移动到跟需要在x轴和y轴上移动多少

然后再把y的根的父亲设为x，将y的根的需要移动多少的值更新

最后询问就搞一搞就行了，我相信大家都会写=.=

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>

using namespace std;

struct node1{
int x,y,len;
char way[2];
}edge[40040];

struct node2{
int x,y,ti,num;
friend bool operator < (node2 a,node2 b){
return a.ti<b.ti;
}
}que[40040];

int n,m,k;
int ans[40040];
int fa[40040];
int tox[40040],toy[40040];

int getroot(int x){
if(fa[x]==x)return x;
return getroot(fa[x]);
}

int getx(int x){
if(fa[x]==x)return 0;
return tox[x]+getx(fa[x]);
}

int gety(int x){
if(fa[x]==x)return 0;
return toy[x]+gety(fa[x]);
}

void un(int x,int y,int len,int to){
int xx=getx(y);
int yy=gety(y);
y=getroot(y);
fa[y]=x;
tox[y]=-xx;
toy[y]=-yy;
if(to==1)toy[y]-=len;
if(to==2)toy[y]+=len;
if(to==3)tox[y]+=len;
if(to==4)tox[y]-=len;
}

int main(){
scanf("%d%d",&n,&m);
for(int i=1; i<=n; i++)fa[i]=i;
for(int i=1; i<=m; i++){
scanf("%d%d%d%s",&edge[i].x,&edge[i].y,&edge[i].len,edge[i].way);
}
scanf("%d",&k);
for(int i=0; i<k; i++){
scanf("%d%d%d",&que[i].x,&que[i].y,&que[i].ti);
que[i].num=i;
}
sort(que,que+k);
int now=1;
for(int i=0; i<k; i++){
while(now<=que[i].ti && now<=m){
if(edge[now].way[0]=='N')un(edge[now].x,edge[now].y,edge[now].len,1);
if(edge[now].way[0]=='S')un(edge[now].x,edge[now].y,edge[now].len,2);
if(edge[now].way[0]=='W')un(edge[now].x,edge[now].y,edge[now].len,3);
if(edge[now].way[0]=='E')un(edge[now].x,edge[now].y,edge[now].len,4);
now++;
}
if(getroot(que[i].x)!=getroot(que[i].y))ans[que[i].num]=-1;
else ans[que[i].num]=abs(getx(que[i].x)-getx(que[i].y))+abs(gety(que[i].x)-gety(que[i].y));
}
for(int i=0; i<k; i++)printf("%d\n",ans[i]);

return 0;
}