AcWing860-染色法判定二分图（java实现）

AcWing860-染色法判定二分图

题目描述

给定一个 n 个点 m 条边的无向图，图中可能存在重边和自环。

请你判断这个图是否是二分图。

输入格式

第一行包含两个整数 n 和 m。

接下来 m 行，每行包含两个整数 u 和 v，表示点 u 和点 v 之间存在一条边。

输出格式

如果给定图是二分图，则输出 Yes，否则输出 No。

数据范围

1≤n,m≤10^5

输入样例：

4 4
1 3
1 4
2 3
2 4

输出样例：

Yes

题解

package acWing860;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {
    static int n,m;
    static int N = 100010, M = 200010;
    static int h[] = new int[N],e[] = new int[M],ne[] = new int[M],idx = 0;
    static int[] color = new int[N];	//	共1和2两种不同的颜色
    static boolean[] st = new boolean[N];
    public static void add(int a,int b){
        e[idx] = b;ne[idx] = h[a];h[a] = idx ++;
    }
    //	dfs(u,c)表示把u号点染色成c颜色，并且判断从u号点开始染其他相连的点是否成功
    public static boolean dfs(int u,int c){
        color[u] = c;
        for(int i = h[u];i != -1;i = ne[i]){
            int j = e[i];
            if(color[j] == 0){
                if(!dfs(j,3 - c)) return false;
            }else if(color[j] == c) return false;//颜色重复
        }
        return true;
    }
    public static void main(String[] args) throws IOException {
        BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
        String[] str = reader.readLine().split(" ");
        n = Integer.parseInt(str[0]);m = Integer.parseInt(str[1]);
        Arrays.fill(h, -1);
        while(m-->0){
            str = reader.readLine().split(" ");
            int a = Integer.parseInt(str[0]);
            int b = Integer.parseInt(str[1]);
            add(a,b);add(b,a);
        }
        boolean flag = true;	//	标记是否染色成功
        for(int i = 1;i <= n;i++){
            //	若未染色
            if(color[i] == 0){
                if(!dfs(i,1)){
                    flag = false;
                    break;
                }
            }
        }
        System.out.println(flag ? "Yes" : "No");
    }
}