题面:
样例输入输出:
input
10 5 3
1 2 10
2 3 10
4 5 -10
5 6 -10
2 5 10
1 10
1 5
3 5
output
-1
20
0
思路:
带权并查集,原始并查集是通过不断地找爹来看祖宗是谁,换句话说只知道两者有关系但不知道有什么关系,如果引入权值的话,也就能更细化地处理一些情况,比如我比我爷爷大几岁这样。而引入权值又有一个问题,我比我爸小3岁,我爸比我爷小三岁,那比较我和我爷爷时候,要一点一点这样查么,时间复杂度呢,所以引入一点新变化,也就是说fa数组不是自己爹了,而是直接指向祖宗,再引入一个val数组直接记录和祖宗差的权值,由此也就解决问题了。
实现:
1:fa数组直接是自己祖宗,这步其实不难,就是在普通并查集的基础上直接更新了一下fa数组。
2:val数组怎么算,直接有关系的话也就算了,隔代再揪着数关系太麻烦了,我引一张别的讲得好的人的截图。
下面贴实现代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<iomanip>
#include<cmath>
using namespace std;
const int maxn=1e5+10;
int n,m,q;//学生 消息 询问
int x,y,s;//x-y=s
int fa[maxn],val[maxn];
int findFa(int x){
if(x!=fa[x]){
int t=fa[x];
fa[x]=findFa(fa[x]);
val[x]+=val[t];
}
return fa[x];
}
void comb(int x,int y,int s){
int fax=findFa(x);
int fay=findFa(y);
if(fax!=fay){
fa[fax]=fay;
val[fax]=-val[x]+val[y]+s;
}
}
int main(){
scanf("%d%d%d",&n,&m,&q);
for(int i=1;i<=n;i++) fa[i]=i;
for(int i=0;i<m;i++){
scanf("%d%d%d",&x,&y,&s);
comb(x,y,s);
}
while(q--){
int a,b;
scanf("%d%d",&a,&b);
if(findFa(a)==findFa(b))cout<<val[a]-val[b]<<endl;
else cout<<"-1"<<endl;
}
return 0;
}
寒假加油啊!!!别继续废物了