poj 1182 食物链

来源:互联网 发布:数控g94编程实例带图 编辑:程序博客网 时间:2024/06/05 14:59


 带权并查集的题--》  

本题详细讲解博文--》 大牛的博客


第一次接触带权并查集,集训队的人都不太会,表示这道题折腾了很长时间。。。


代码:


#include<stdio.h>#include<string.h>int rank[50005]= {0};int fath[50005]= {0};int n,k;int cnt=0;void init(){    for(int i=0; i<=n; i++)    {        fath[i]=i;        rank[i]=0;    }}int find(int a){    if(a==fath[a])        return a;    int old=fath[a];    //  这里坑了我好多次       fath[a]=find(fath[a]);    rank[a]=(rank[a]+rank[old])%3;    return fath[a];}/*****   错误代码:  find()int find(int a){    if(a==fath[a])        return a;    fath[a]=find(fath[a]);    rank[a]=(rank[a]+rank[fath[a]])%3;    这里被坑了   在上一步中,fath【a】的值已经改变了   上一步fath【a】的更新又必须在前面,所以我们应当定义一个变量old来记录他的原始值。    return fath[a];}*//// 并查集的合并  包括权值转化 void uion(int d,int x,int y){    int fx=find(x);    int fy=find(y);    if(fx==fy)        return ;    fath[fx]=fy;    rank[fx]=(rank[y]+d-rank[x]+3)%3;   ///这里的权值转化表示暂时还没懂,只是从大牛们的博客中套用的。}void sove(int d,int a,int b)     // 不符合要求的三种情况的判断{    if(a>n||b>n)    {        cnt++;        return ;    }    if(d==2&&a==b)    {        cnt++;        return ;    }    int x=find(a);    int y=find(b);    if(x!=y)        uion(d-1,a,b);    else    {        if(rank[a]==((d-1)+rank[b])%3)            uion(d-1,a,b);        else            cnt++;    }}int main(){    //freopen("a.txt","r",stdin);    scanf("%d%d",&n,&k);    init();    for(int i=0; i<k; i++)    {        int d,a,b;        scanf("%d%d%d",&d,&a,&b);        sove(d,a,b);    }    printf("%d\n",cnt);}


0 0
原创粉丝点击