【思维题】Codeforce 1217D Coloring Edges

这段时间要沉迷刷题一段时间了，就让CSDN陪我一起吧！

一、题目大意

这道题目的大致意思就是，给定一个无自环、无多重边的有向图，并且给定好染色图的定义：如果一个有向图每条边都有颜色，并且一个环中不止有一种颜色，那么这个图称为一个好染色图。现在要你求最少多少种颜色，可以把给定的有向图染成好染色图。

二、题目思路以及AC代码

本以为题很复杂，一直在琢磨如何DFS，也是自己比较菜，没想出来，就去查题解了。结果题解两句话解决问题：

• 如果有向图无环，那么结果为1
• 如果有向图有环，那么结果为2

我当时就懵逼了，第一个我是明白的，无环，那全染一个颜色就完事了。后来仔细琢磨了一下，如果有环，那构成这个环的边，肯定是一部分由小顶点连向大顶点，一部分由大顶点连向小顶点，那么只要把这两种边分别染不同的颜色，不就可以了？

虽然道理简单，但真的不容易想到，还需要历练呀！然后这道题就简化成了一道用拓扑排序判断有无环的一道题。

下面给出AC代码：

#include <iostream>
#include <vector>
#include <queue>
#define MAXN 5010
using namespace std;

struct Edge {
    int u, v;
};
Edge e[MAXN];

int n, m;
int du[MAXN];
vector<int> edges[MAXN];
bool vis[MAXN];

void init() {
    for (int i=0;i<MAXN;i++) {
        edges[i].clear();
        vis[i] = false;
        du[i] = 0;
    }
}

bool Tuopu() {
    queue<int> q;
    for (int i=1;i<=n;i++) {
        if (!du[i]) {
            q.push(i);
            vis[i] = true;
        }
    }

    while (!q.empty()) {
        int cur = q.front(); q.pop();

        int size = edges[cur].size();
        for (int i=0;i<size;i++) {
            int v = edges[cur][i];

            du[v]--;
            if (!du[v]) {
                q.push(v);
                vis[v] = true;
            }
        }
    }

    for (int i=1;i<=n;i++) {
        if (!vis[i]) return true;           // 有环
    }

    return false;
}

int main() {

    init();

    cin >> n >> m;

    for (int i=0;i<m;i++) {
        int u, v;
        cin >> u >> v;
        e[i].u = u;
        e[i].v = v;
        edges[u].push_back(v);
        du[v]++;
    }

    if (Tuopu()) {
        cout << "2" << endl;
        for (int i=0;i<m;i++) {
            if (e[i].u < e[i].v) cout << 1 << " ";
            else cout << 2 << " ";
        }
        cout << endl;
    }
    else {
        cout << "1" << endl;
        for (int i=0;i<m;i++) {
            cout << 1 << " ";
        }
        cout << endl;
    }

    // system("pause");

    return 0;
}

如果有问题，欢迎大家指正！！！