Zjnu Stadium(hdu3047带权并查集)

题意:一个300列的无限行的循环场地,a b d代表a,b顺时针相距d的距离,现在给你一些距离,判断是否有冲突,如果有冲突计算冲突的次数

思路:带权并查集

a,b的距离等于b到根节点的距离 - a到根节点的距离

1.当a,b在同一集合的时候就用b到根节点的距离 - a到根节点的距离和当前输入的距离进行对比,看是否满足条件

2.当a,b不在同一集合的时候合并两个节点,更新距离

向量法,方向一定要搞清楚,父亲指向儿子

如果x的父亲rootx ,y的父亲是rooty

rootx --> x, rooty --> y合并的方向是rootx的父亲是rooty 即rooty --> rootx

x -->  y的距离是d

rooty --> rootx = rootx --> x + x --> y + y --> rooty = rootx --> x +(-( y --> x ) ) + (-( rooty -->y)) 

合并的方向不同式子也不同

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int flag = 0;
struct bian
{
    int father;
    int dis;
}p[50050];

void Make_set(int n)
{
    int i;
    for(i = 1; i <= n; i++)
    {
        p[i].father = i;
        p[i].dis = 0;
    }
}

int Find_set(int x)
{
    if(x != p[x].father)
    {
        int temp = p[x].father;
        p[x].father = Find_set(p[x].father);
        p[x].dis = ( p[x].dis + p[temp].dis) % 300;
    }
    return p[x].father;
}

void Union(int a,int b,int d)
{
    int x = Find_set(a);
    int y = Find_set(b);
    if(x == y)
    {
        if(( (p[b].dis-p[a].dis+300) % 300 ) != d)
        {
            flag = 1;
            return ;
        }
    }
    else
    {
        p[x].father = y;
        p[x].dis = (p[b].dis+300-d-p[a].dis) % 300;
    }
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m) != EOF)
    {
        int i,sum = 0;
        Make_set(n);
        for(i = 0; i < m; i++)
        {
            int a,b,d;
            flag = 0;
            scanf("%d%d%d",&a,&b,&d);
            Union(a,b,d);
            if(flag)
                sum++;
        }
        printf("%d\n",sum);
    } 
    return 0;
}


 


 

你可能感兴趣的:(HDU)