088_食物链问题(并查集)

来源:互联网 发布:上传网站数据库是啥 编辑:程序博客网 时间:2024/06/09 21:50

 食物链问题:有N个动物,每个动物只能为A,B,C三种动物中的一种,已知 A吃B, B吃C, C吃A。

 现在给出K条信息,每条信息值可能为以下两种中的一种:

 1. x,y属于同一种动物;

 2. x吃y。

 要求找出这K 条信息中不正确的条数。 题源来自《挑战程序竞赛》第二版88页。

 此题使用并查集来实现应该比较方便。

 先初始化并查集,对于每一个动物x,创建三个元素:x,x+N,x+2N,分别代表x为动物 A, B, C。

 对于每一号动物,根据其可能性,可以归到如下三个集合中:

 1. x为A以及其捕食关系;如1号动物为A,2号动物也为A,(1,2浩动物属于同种) 3号动物为B(1号吃3号)

 2. x为B以及其捕食关系;

 3. x为C以及其捕食关系;

 以上三个并查集的集合,对于某一个来说,要么集合内全部成立,要么集合内全部不成立。

 对于某一条信息,根据如下方式来维护并查集:

 1. 如果x和y是同种,那么合并x-A和y-A,x-B和y-B,x-C和y-C, 即 (x,y)合并到第一个集合,(x+N,y+N)合并到第二个集合,(x+2N,y+2N)合并到第三个集合;

 2. 如果x吃y,那么合并x-A和y-B,x-B和y-C,x-C和y-A,即(x,y+N)合并到第一个集合,(x+N,y+2N)合并到第二个集合,(x+2N,y)合并到第三个集合;

 如果某个集合内的元素关系不冲突,那么这个集合就是可行的一个食物链关系。 可以看到由于我们维护集合关系的轮换性,如果第一个集合可行,那么第二,第三个集合同样可  行。同样,如果对于某一条信息,x-A的情况下在第一个集合中冲突,必有x-B在第二个集合中冲突,x-C在第三个集合中冲突。

 所以,检测冲突时只需检测x-A的情况。

 那么,如何检测冲突呢?

 1.如果遇到第一种关系,即x,y属于同种,那么冲突的话就是有x吃y或者y吃x。

 只需考虑x-A的情况,即发生了y-B,y-C的情况,具体为:(x-A,y-B)存在第一集合(x吃y已经存在),(x-A,y-C)存在第三集合(y吃x  已经存在)

 2.如果遇到第二种关系,即x吃y,那么冲突的情况就是y吃x后者x与y同种。

 同样只需考虑x-A的情况,即发生了y-A,y-C的情况,具体为(x-A,y-A)存在第一集合(x,y同种已经存在), (x-A,y-C)存在第三集合(y吃x已经存在)

 使用并查集的same方法就能做出以上判断。


////  088_food chain.cpp//  changlle////  Created by user on 12/30/15.//  Copyright (c) 2015 user. All rights reserved.//#include <iostream>using namespace std;int par[500];int _rank[500];//初始化元素void init(int n) {    for (int i=0;i<n;i++) {        par[i]=i;        _rank[i]=0;    }  }//查询树的根int find(int x) {    if (par[x]==x)        return x;    else        return par[x]=find(par[x]);   }//合并x,y所属的集合void unite(int x, int y) {    x=find(x);    y=find(y);    if (x==y) return;        if (_rank[x]<_rank[y])        par[x]=y;    else {        par[y]=x;        if(_rank[x]==_rank[y])            _rank[x]++;      }}//判断x和y是否属于同一个集合bool same(int x, int y) {        return find(x)==find(y);}//以上为并查集的实现int N=100;int K=7;int T[7]={1,2,2,2,1,2,1};int X[7]={101,1,2,3,1,3,5};int Y[7]={1,2,3,3,3,1,5};int main() {        init(N*3);    int count=0;        for (int i=0;i<K;i++){                int t=T[i];        int x=X[i]-1;        int y=Y[i]-1;                if (x<0 || N<=x || y<0 || N<=y){            count++;            continue;        }                if (t==1){ //属于同一类,只需针对x=A来进行判断                        if (same(x,y+N) || same(x,y+2*N))                count++;                        else {                unite(x,y);                unite(x+N,y+N);                unite (x+2*N, y+2*N);            }                    }                if (t==2) { //捕食关系,同样只需针对x=A来进行判断                        if (same(x,y) || same(x, y+2*N))               count++;            else {                unite(x,y+N);                unite(x+N,y+2*N);                unite(x+2*N,y);            }                    } }            cout<<count<<endl;                                    return 0;}













  

0 0
原创粉丝点击