【并查集】:poj1182,食物链


关键是向量化的关系调整。


下面能过。

注意几点:

FindParent、用scanf别用cin。

# include<iostream>
# include<string.h>
# include<stdio.h>
using namespace std;

# define N 50010
# define K 100005


struct NODE
{
    int parent;
    int relation;
    //0表示和父节点同类,1表示被父节点吃(规定,让y指向x),2表示吃父节点
    //(必须和d-1对应上)
    //若D=1,则表示X和Y是同类。
    //若D=2,则表示X吃Y。
};


NODE node[N];


void MakeSet(int n)
{
    for(int i=1;i<=n;i++)
    {
        node[i].parent=i;
        node[i].relation=0;
    }
}

int FindParent(int x)
{
    if(node[x].parent==x)
    {
        return x;
    }
    else
    {
        /*
        node[x].parent=FindParent(node[x].parent);
        node[x].relation=(node[x].relation+node[node[x].parent].relation)%3; //x->xp->xpp
        */
        //can't use above codes, will WA
        int temp = node[x].parent; //路径压缩
        node[x].parent = FindParent(temp);
        node[x].relation = (node[x].relation + node[temp].relation) % 3; //关系域更新

        return node[x].parent;
    }
}

int Judge(int x, int y, int d)
{
    int xp=FindParent(x);
    int yp=FindParent(y);

    if(xp==yp) //不需要合并,直接判断是否冲突
    {
        if( d==1 && node[x].relation!=node[y].relation )
        {
            return 1;
        }
        if( d==2 && ( (3+node[y].relation-node[x].relation)%3!=(d-1) ) )
        {
            return 1;
        }
    }
    else
    {
        //先合并
        node[yp].parent=xp;
        //再调整关系
        //node[yp].relation=(-node[y].relation-(d-1)+node[x].relation+3)%3;
        node[yp].relation=(3-node[y].relation+(d-1)+node[x].relation)%3;
        //注意是,+(d-1),因为d-1之后,已经对应y->x的关系了(d表示x->y的关系)
        return 0;
    }
}

int main()
{
    int sum=0;
    int n,k,d,x,y,i;
    cin>>n>>k;

    MakeSet(n);

    for(i=1;i<=k;i++)
    {
        //cin>>d>>x>>y;
        scanf("%d%d%d",&d,&x,&y); //or TLE

        if( (x>n||y>n) || (x==y&&d==2) )
        {
            sum++;
        }
        else
        {
            //sum+=Judge(x,y,d);
            int xp=FindParent(x);
            int yp=FindParent(y);

            if(xp==yp) //不需要合并,直接判断是否冲突
            {
                if( d==1 && node[x].relation!=node[y].relation )
                {
                    //return 1;
                    sum++;
                }
                if( d==2 && ( (3+node[y].relation-node[x].relation)%3!=(d-1) ) )
                {
                    //return 1;
                    sum++;
                }
            }
            else
            {
                //先合并
                node[yp].parent=xp;
                //再调整关系
                //node[yp].relation=(-node[y].relation-(d-1)+node[x].relation+3)%3;
                node[yp].relation=(3-node[y].relation+(d-1)+node[x].relation)%3;
                //注意是,+(d-1),因为d-1之后,已经对应y->x的关系了(d表示x->y的关系)
                //return 0;
            }
        }
    }

    cout<<sum<<endl;

    return 0;
}



没看出啊下面有什么问题,但是就是不AC。大神看到了指导一下。

# include<iostream>
# include<string.h>
# include<stdio.h>
using namespace std;

# define N 50010
# define K 100005


struct NODE
{
    int parent;
    int relation;
    //0表示和父节点同类,1表示被父节点吃(规定,让y指向x),2表示吃父节点
    //(必须和d-1对应上)
    //若D=1,则表示X和Y是同类。
    //若D=2,则表示X吃Y。
};


NODE node[N];


void MakeSet(int n)
{
    for(int i=1;i<=n;i++)
    {
        node[i].parent=i;
        node[i].relation=0;
    }
}

int FindParent(int x)
{
    if(node[x].parent==x)
    {
        return x;
    }
    else
    {
        /*
        node[x].parent=FindParent(node[x].parent);
        node[x].relation=(node[x].relation+node[node[x].parent].relation)%3; //x->xp->xpp
        */
        //can't use above codes, will WA
        int temp = node[x].parent; //路径压缩
        node[x].parent = FindParent(temp);
        node[x].relation = (node[x].relation + node[temp].relation) % 3; //关系域更新

        return node[x].parent;
    }
}

int Judge(int x, int y, int d)
{
    int xp=FindParent(x);
    int yp=FindParent(y);

    if(xp==yp) //不需要合并,直接判断是否冲突
    {
        if( d==1 && node[x].relation!=node[y].relation )
        {
            return 1;
        }
        if( d==2 && ( (3+node[y].relation-node[x].relation)%3!=(d-1) ) )
        {
            return 1;
        }
    }
    else
    {
        //先合并
        node[yp].parent=xp;
        //再调整关系
        //node[yp].relation=(-node[y].relation-(d-1)+node[x].relation+3)%3;
        node[yp].relation=(3-node[y].relation+(d-1)+node[x].relation)%3;
        //注意是,+(d-1),因为d-1之后,已经对应y->x的关系了(d表示x->y的关系)
        return 0;
    }
}

int main()
{
    int sum=0;
    int n,k,d,x,y,i;
    cin>>n>>k;

    MakeSet(n);

    for(i=1;i<=k;i++)
    {
        //cin>>d>>x>>y;
        scanf("%d%d%d",&d,&x,&y); //or TLE

        if( (x>n||y>n) || (x==y&&d==2) )
        {
            sum++;
        }
        else
        {
            sum+=Judge(x,y,d);
            /*
            int xp=FindParent(x);
            int yp=FindParent(y);

            if(xp==yp) //不需要合并,直接判断是否冲突
            {
                if( d==1 && node[x].relation!=node[y].relation )
                {
                    //return 1;
                    sum++;
                }
                if( d==2 && ( (3+node[y].relation-node[x].relation)%3!=(d-1) ) )
                {
                    //return 1;
                    sum++;
                }
            }
            else
            {
                //先合并
                node[yp].parent=xp;
                //再调整关系
                //node[yp].relation=(-node[y].relation-(d-1)+node[x].relation+3)%3;
                node[yp].relation=(3-node[y].relation+(d-1)+node[x].relation)%3;
                //注意是,+(d-1),因为d-1之后,已经对应y->x的关系了(d表示x->y的关系)
                //return 0;
            }
            */
        }
    }

    cout<<sum<<endl;

    return 0;
}


又做了一遍,过了:

#include <iostream>
#include <stdio.h>
using namespace std;

#define N 50005

struct NODE
{
    int par;
    int rel;
    //d=1,x和y同类
    //d=2,x吃y
    //转换为d-1:
    //d-1=0,x和y同类
    //d-1=1,父亲吃儿子(令y是x的儿子)
    //d-1=2,儿子吃父亲(令y是x的儿子)
};

NODE node[N];

int FindPar(int x)
{
    if(node[x].par==x)
    {
        return x;
    }
    else
    {
        int temp=node[x].par;
        node[x].par=FindPar(temp);
        node[x].rel=(node[x].rel+node[temp].rel)%3;
        return node[x].par;
    }
}


int main()
{
    int n,k,d,x,y,i,sum=0;

    cin>>n>>k;

    for(i=1;i<=n;i++)
    {
        node[i].par=i;
        node[i].rel=0;
    }

    for(i=1;i<=k;i++)
    {
        scanf("%d%d%d",&d,&x,&y);

        if( (x>n || y>n) || (d==2 && x==y) )
        {
            sum++;
        }
        else
        {
            int xp=FindPar(x);
            int yp=FindPar(y);
            if(xp==yp)
            {
                if( d==1 && (node[x].rel!=node[y].rel) )
                {
                    sum++;
                }
                if( d==2 && (3+node[y].rel-node[x].rel)%3 != (d-1) )
                {
                    sum++;
                }
            }
            else
            {
                node[yp].par=xp;
                node[yp].rel=(3-node[y].rel+(d-1)+node[x].rel)%3;
            }
        }
    }

    cout<<sum<<endl;

    return 0;
}


  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值