题目链接:http://hihocoder.com/problemset/problem/1515
#1515 : 分数调查
时间限制:
10000ms
单点时限:
1000ms
内存限制:
256MB
-
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
描述
小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高的分数。
解析:带权并查集(自己百度模板)
代码:
#include<bits/stdc++.h>
#define N 100009
using namespace std;
const int INF = 0x3f3f3f3f;
int pre[N], d[N];
inline void q_read(int &num)
{
char ch; int f = 1;
while(true)
{
ch = getchar();
if(ch == '-') f = -1;
if(isdigit(ch))
{
num = ch - '0';
break;
}
}
while(ch = getchar(), isdigit(ch)) num = num*10+ch-'0';
num *= f;
}
int get_pre(int x)
{
if(x==pre[x]) return x;
int t = pre[x];
pre[x] = get_pre(pre[x]);
d[x] += d[t];
return pre[x];
}
void Union(int u, int v, int w)
{
int x = get_pre(u);
int y = get_pre(v);
if(x != y)
{
pre[x] = y;
d[x] = d[v] + w - d[u];
//d[u] = w+d[v];
}
}
void init(int n)
{
for(int i = 1; i <= n; i++) pre[i] = i;
memset(d, 0, sizeof(d));
}
int main()
{
int n, m, q, u, v, w;
q_read(n); q_read(m); q_read(q);
init(n);
for(int i = 1; i <= m; i++)
{
q_read(u); q_read(v); q_read(w);
Union(u, v, w);
}
while(q--)
{
q_read(u); q_read(v);
int x = get_pre(u);
int y = get_pre(v);
if(x != y) puts("-1");
else printf("%d\n", d[u] - d[v]);
}
return 0;
}