POJ 8594 Butterfly|二部图染色

题目描述

总时间限制: 

1000ms

单个测试点时间限制: 

100ms

内存限制: 

65536kB

描述

有一群旅行爱好者,有一天,他们带回了n只蝴蝶回来。他们相信每一只都属于两个不同种类中的一种,为了讨论方便,我们称它们为A与B。他们想把n只标本分成两组——一些属于A且一些属于B——但是直接标记任何一个标本对于他们是非常困难,因此他们决定采用下面的方法。

对每对标本i和j,他们细心地把它们放到一起研究。如果他们以自己的判断足以确信,那么他们把这对蝴蝶标记为“相同”(这意味着他们相信这两只来自同一类)或者是“不同”(这意味着他们相信这两只来自不同的类)。他们也可以对某些标本判断不出来而弃权,在这种情况下,我们称这对标本是不明确的。

现在他们有n只标本的集合,还有对那些没有弃权的标本对的m个判断的集合(“相同”或者“不同”)。他们想知道这个数据与每只蝴蝶来自A和B中的一个类的想法是否一致。更具体地说,如果对每对蝴蝶按照下述方式标记A或B是可能的,即对每个标为“相同”的(i,j)对,就是i与j有相同标记的情况;对每个标为“不同”的(i,j)对,就是i与j有不同标记的情况。那么我们可以说这m个判断是一致的。他们正在冥思苦想自己的判断是否是一致的。请你帮他们设计合理的算法解决该问题。
 

输入

输入包含多组数据,以文件结束符为终止。

每组数据第一行为两个整数,分别是n和m:
n为蝴蝶的数量,编号从0到n-1
m为关系的数量

接下来是m组关系数据,每组数据占一行,为三个整数,前两个整数表示蝴蝶的编号,第三个整数为关系的种类(相同或者不同):
0为相同,1为不同

1 < n <= 1000
1 < m <= 100000

输出

合理就输出YES
不合理就输出NO

样例输入

3 3
0 1 0
1 2 1
0 2 1
3 3
0 1 0
1 2 1
0 2 0

样例输出

YES
NO

提示

注意输出的大写和回车

解题思路

一般二部图只有一种边,就是两个点不同类则连一条边。本题多了一种边,就是相同的边。

具体思路是对于每个节点,都要检查一遍染色是否正确。每个节点使用BFS的方法,若边没有染色,则根据边的类型染上相同/不同的颜色,若边已经染色,则查看是否有冲突。一旦产生冲突,整个就返回false。

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

int n,m;
int map[1005][1005];
int color[1005];
int vis[1005];
queue<int> q;

bool BFS(int node){
    if(vis[node]) return true;
    vis[node]=1;
    if(!color[node]) color[node] = 1;
    for(int i=0;i<n;i++){
        //1是相同,2是不同
        if (map[node][i]==1){
            if(color[i]){
                if(color[i]!=color[node]){
                    return false;        
                }
            }else{
                color[i]=color[node];
            }
            if(!vis[i]) q.push(i);
        }else if (map[node][i]==2){
            if(color[i]){
                if(color[i]==color[node]){
                    return false;
                    break;        
                }
            }else{
                color[i]=3-color[node];
            }
            if(!vis[i]) q.push(i);
        }
    }
    return true;
}

int main(){
    while(~scanf("%d%d",&n,&m)){
        int a,b,val;
        while(!q.empty()) q.pop();
        memset(color,0,sizeof(color));
        memset(vis,0,sizeof(vis));
        memset(map,0,sizeof(map));
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&a,&b,&val);
            map[a][b]=val+1;
            map[b][a]=val+1;
        }
        bool res=true;
        for(int i=0;i<n;i++){
            q.push(i);
            while(!q.empty()){
                if(!BFS(q.front())){
                    res=false;
                    break;
                }
                q.pop();
            }
            if(!res) break;   
        }
        if(!res) printf("NO\n");
        else printf("YES\n");
    }
    return 0;
}

debug过程中发现一个大问题,老是TLE,原来是因为写成了while(scanf("%d%d",&n,&m)),一直没办法从循环里跳出来......

以下三种写法等效,都是正确的写法TUT:

  • while(cin>>n>>m))
  • while(scanf("%d%d",&n,&m)!=EOF)
  • while(~scanf("%d%d",&n,&m))
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值