小Hi的学校总共有N名学生,编号1-N。学校刚刚进行了一场全校的古诗文水平测验。
学校没有公布测验的成绩,所以小Hi只能得到一些小道消息,例如X号同学的分数比Y号同学的分数高S分。
小Hi想知道利用这些消息,能不能判断出某两位同学之间的分数高低?
Input
第一行包含三个整数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
数据保证没有矛盾。
Output
对于每个询问,如果不能判断出X比Y高几分输出-1。否则输出X比Y高的分数。
Sample 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
Sample Output
-1
20
0
跟hdu的一个题特别像,代码几乎一样。
dis数组代表着距离父亲节点的距离。如果这两个节点的父亲节点不一样,就不能判断大小。如果父亲节点一样的话,就直接做差比较就好了。
代码如下:
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxx=1e5+100;
int f[maxx],dis[maxx];
int n,m,k;
inline int getf(int u)
{
int x;
if(u!=f[u])
{
x=getf(f[u]);
dis[u]+=dis[f[u]];
f[u]=x;
}
return f[u];
}
inline void merge(int u,int v,int val)
{
int t1=getf(u);
int t2=getf(v);
if(t1!=t2)
{
f[t1]=t2;
dis[t1]=dis[v]-dis[u]+val;
}
}
int main()
{
int x,y,v;
while(~scanf("%d%d%d",&n,&m,&k))
{
for(int i=0;i<=n;i++) f[i]=i,dis[i]=0;
while(m--)
{
scanf("%d%d%d",&x,&y,&v);
if(v<0) swap(x,y),v=-v;
merge(x,y,v);
}
while(k--)
{
scanf("%d%d",&x,&y);
int t1=getf(x);
int t2=getf(y);
if(t1!=t2) puts("-1");
else printf("%d\n",dis[x]-dis[y]);
}
}
return 0;
}
努力加油a啊,(o)/~