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;
}