hdu(3047)——Zjnu Stadium带权并查集

hdu(3047)——Zjnu Stadium

带权并查集

hdu(3047)——Zjnu Stadium

题目传送门

题意：输入A、B、X 表示B要坐在A大X的座位，由于体育馆的行数可以视为无穷大，所以可以不考虑环形的循环，有n个人，m条数据，要求输出错误数据的条数。当B没有坐在比A大x 的位置时，该条数据为错误。

这题是2009年杭电多校的题目，一开始看这题，觉得不是很容易理解，还是有点难想的，也不知道当时这题算什么难度，因为一条数据表示了两个人的相对位置，所以我们考虑用带权并查集来处理，dis[i]表示第i个人离这个集合根节点的距离，当新输出一条数据，两个人属于不同的集合，此时不会出现错误，设fx为x的根节点，fy为y的根节点，我们将x集合合并到y集合，这时候y集合的距离不变，但x集合中的dis需要更新，要使y与x的距离保持s，如下图所示：

这边的dis数组可由find数组得到，当输入的两个x、y为同一集合时，我们只要判断dis[y]-dis[x]是不是等于s，如果不是，则不满足条件，ans++；
下边贴下代码：

#include<iostream>
#include<vector>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<set>
#include<algorithm>
#include<queue>
#include <map>
#include<stack>
#include<list>
#define inf 0x3f3f3f3f
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll> P;
const int maxn=5e4+5;
const ll mod=1e9+7;
int n,m,pre[maxn],dis[maxn],ans;
void init(){
    ans=0;
    for(int i=0;i<maxn;i++){
        pre[i]=i;
        dis[i]=0;
    }
}
int find(int x){
    if(pre[x]==x) return x;
    int temp=pre[x];
    pre[x]=find(temp);
    dis[x]+=dis[temp];
    return pre[x];
}
int join(int x,int y,int s){
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy){
        pre[fy]=fx;
        dis[fy]=dis[x]+s-dis[y];
    }
    else{
        if(dis[y]-dis[x]!=s) ans++;
    }
}
int main()
{
    while(cin>>n>>m){
        init();
        while(m--){
            int u,v,s;
            scanf("%d %d %d",&u,&v,&s);
            join(u,v,s);
        }
        cout<<ans<<'\n';
    }
    return 0;
}