DFS写法

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
#include<iostream>
#include<cstring>

using namespace std;

const int N = 100000 + 10;
int h[N], e[2 * N], idx, ne[2 * N];
int n, m;
int st[N];

void add(int a, int b)
{
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx ++;
}

bool dfs(int x, int c)
{
  	// c表示的是当前的颜色
    st[x] = c;
    for(int i = h[x]; i != -1; i = ne[i])
    {
        int d = e[i];
        if(st[d] == 0)		// 如果没有被染色
        {
            if(!dfs(d,3 - c)) return false;
        }
        if(st[d] == c)		// 如果连接的点的颜色等于当前点的颜色,则不是二分图
            return false;
    }
    
    return true;
}

int main()
{
    memset(h, -1, sizeof h);
    cin >> n >> m;
    while(m--)
    {
        int a, b;
        cin >> a >> b;
        add(a,b);
        add(b,a);
    }
    
  	// 分别遍历所有没有遍历过的点,有可能二分图不是连通图
    bool ret = true;
    for(int i = 1; i <= n; i++)
    {
        if(st[i] == 0)
        {
            if(!dfs(i,1)) 
            {
                ret = false;
                break;
            }
        }
    }
  
  
    if(ret) cout << "Yes";
    else cout << "No";
    return 0;
}

BFS写法

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int N = 200010;

int h[N], e[N], ne[N], st[N], idx;
int n, m;

void add(int a, int b)
{
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx++;
}

bool bfs(int o)
{
    queue<int> q;
    q.push(o);
    int flag = 1;
    st[o] = flag; 
    while(q.size())
    {
      	// 每次修改flag状态
        int t = q.size();
        flag = 3 - flag;
      	// 按层遍历
        for(int i = 0 ; i < t; i++)
        {
            auto p = q.front();
            q.pop();
            
            for(int j = h[p]; j != -1; j = ne[j])
            {
                int p2 = e[j];
                if(!st[p2])
                {
                    st[p2] = flag;
                    q.push(p2);
                }
              	// 重边也要判断
                else if(p == p2 || st[p2] != flag) {
                    return false;
                }
            }
        }
    }
    return true;
}

int main()
{
    cin >> n >> m;
    
    memset(h, -1, sizeof h);
    for(int i = 0 ; i< m ; i++)
    {
        int a, b;
        cin >> a >> b;
        add(a,b);
        add(b,a);
    }
    
    bool flag = bfs(1);
    for(int i = 1; i <= n; i++)
    {
        if(!st[i]) flag = flag && bfs(i);
    }
    
    
    if(flag)cout << "Yes";
    else cout << "No";
    return 0;
}