Description
在实验室中,Nathan Wada作为助手的职责是测定两个样品的重量差异。当样品的差异很小时,使用天平能比使用
弹簧秤得到更精确的结果,所以他只使用天平来测得一些样品的重量差。他偶尔会被询问一些样品的重量差,而他
能否回答这些问题取决于在回答相应问题时他已经得到的测量结果。由于他所在处理的测量数据是巨大的,所以他
希望你能写个程序帮他处理数据和回答问题。
Input
输入包含多组测试数据。每组数据第一行包含两个整数 N 和 M ,其中 N 表示样品的数量,
样品从 1 到 N 标号,满足 2≤N≤100000 。
接下来 M 行,每行包括一个测量结果或者询问,按时间顺序给出,满足 1≤M≤100000 。
一个测量结果被格式化为 ! a b w ,表示第 a 个样品比第 b 个样品轻 w 个单位重量
满足 a≠b,0≤w≤1000000 ,且任意的测试结果互不矛盾。
一个询问被格式化为 ? a b ,表示询问第 a 个样品比第 b 个样品轻多少个单位重量,满足 a≠b 。
输入以两个零作为结束。
Output
对于每个询问输出一行,如果能回答问题,则输出问题的答案,你可以认为答案的绝对值不超过 1000000
否则输出 UNKNOWN ,表示不能回答问题。
题解:
一定是还没睡醒。
这题带权并查集就好了,表示每个点比它祖宗轻多少。
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=100010;
int n,m;
int fa[N],v[N];
char s[10],x,y;
int findfa(int x)
{
if(x==fa[x]) return x;
int t=findfa(fa[x]);
v[x]+=v[fa[x]];
fa[x]=t;
return fa[x];
}
int main()
{
while(1)
{
scanf("%d%d",&n,&m);
if(!n&&!m) break;
memset(v,0,sizeof(v));
for(int i=1;i<=n;i++) fa[i]=i;
int x,y,z;
for(int i=1;i<=m;i++)
{
scanf("%s%d%d",s,&x,&y);
if(s[0]=='!')
{
scanf("%d",&z);
int tx=findfa(x),ty=findfa(y);
if(tx!=ty)
{
fa[tx]=ty;
v[tx]=v[y]+z-v[x];
}
}
else
{
int tx=findfa(x),ty=findfa(y);
if(tx!=ty)
{
printf("UNKNOWN\n");
}
else printf("%d\n",v[x]-v[y]);
}
}
}
}