hihoCoder-1515 : 分数调查(带权并查集)
一、题目描述
小Hi的学校总共有N名学生,编号1-N。学校刚刚进行了一场全校的古诗文水平测验。
学校没有公布测验的成绩,所以小Hi只能得到一些小道消息,例如X号同学的分数比Y号同学的分数高S分。
小Hi想知道利用这些消息,能不能判断出某两位同学之间的分数高低?
输入
第一行包含三个整数N, M和Q。N表示学生总数,M表示小Hi知道消息的总数,Q表示小Hi想询问的数量。
以下M行每行三个整数,X, Y和S。表示X号同学的分数比Y号同学的分数高S分。
以下Q行每行两个整数,X和Y。表示小Hi想知道X号同学的分数比Y号同学的分数高几分。
对于50%的数据,1 <= N, M, Q <= 1000
对于100%的数据,1 <= N, M, Q<= 100000 1 <= X, Y <= N -1000 <= S <= 1000
数据保证没有矛盾。
输出
对于每个询问,如果不能判断出X比Y高几分输出-1。否则输出X比Y高的分数。
样例输入
10 5 3
1 2 10
2 3 10
4 5 -10
5 6 -10
2 5 10
1 10
1 5
3 5
样例输出
-1
20
0
二、解题思路
两个点之间存在数量关系,不断合并点于点信息的时候,被合并集合结点的关系也进行更新。这时候使用带权路径的并查集最为合适。合并时:输入a b s在本程序中给num[Find(b)]赋值、查找时:给所查找元素a的num[a]压缩路径的同时进行更新。经过一系列操作最中查询两点的关系时,当两点在一个连通分量时(同一个祖先)用后者的num置减去前者的num置即可,不在一个连通分量时输出-1即可。
三、参考代码
#include <iostream>
#include <string.h>
using namespace std;
#define MAXN 110000
int pre[MAXN],num[MAXN];
int Find(int x)
{
if(pre[x]!=x)
{
int temp=pre[x];
pre[x]=Find(pre[x]);
num[x]+=num[temp];
return pre[x];
}
return x;
}
int join(int x,int y,int s)
{
int fx=Find(x),fy=Find(y);
if(fx!=fy)
{
pre[fy]=fx;
num[fy]=num[x]+s-num[y];
}
return 0;
}
int main()
{
int N,M,Q;
cin>>N>>M>>Q;
for(int i=1; i<=N; i++)
{
pre[i]=i;
}
memset(num,0,sizeof(num));
while(M--)
{
int a,b,s;
cin>>a>>b>>s;
join(a,b,s);
}
while(Q--)
{
int a,b;
cin>>a>>b;
if(Find(a)!=Find(b))
{
cout<<-1<<endl;
}
else if(Find(a)==Find(b))
cout<<num[b]-num[a]<<endl;
}
return 0;
}
四、总结
并查集有关的题目,根据不同的题只需考虑题目所给元素之间的关系在合并和查找中添加一些算法完成题目要求即可,多加练习就好。