暑期嗨十之A - A CodeForces - 1217D（拓扑排序）

CodeForces - 1217D

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

第一个好理解，第二个要怎么想呢，如果有环，那构成这个环的边，肯定是一部分由小顶点连向大顶点，一部分由大顶点连向小顶点，那么只要把这两种边分别染不同的颜色，然后这个题就转换成拓扑排序判断有环无环

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <stack>
#include <algorithm>
#include <iomanip>
#include <map>
#include <queue>
#include <vector>
#include <set>
typedef long long ll;
using namespace std;
#define MAXN 5010
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;
    }

}