Codeforces 1144F

题目链接

http://codeforces.com/contest/1144/problem/F

题意

输入一张有向图，无自回路和重边，判断能否将它变为有向图，使得图中任意一条路径长度都小于2。

如果可以，按照输入的边的顺序输出构造的每条边的方向，构造的边与输入的方向一致就输出1，否则输出0。

思路

构造的有向图中，对于每个顶点，要么所有边都是出，要么所有边都是入。

可以转化成用两种颜色染色。

从任意一点开始dfs，把和它相邻的点全部染成不同的颜色，遇到相邻的点颜色相同（存在奇数环）就输出NO。

AC代码

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <set>
#include <vector>
#include <queue>
#include <stack>
#include <map>
using namespace std;
#define ll long long 
const int maxn = 200010;

vector<int> g[maxn], a; //a是输入的第i条边的起点
bool vis[maxn];
int c[maxn];    //第i个点的颜色
bool isok = true;

void dfs(int x, int cur) {  //第x个点，涂cur颜色
    vis[x] = true;
    c[x] = cur;
    for(int i = 0; i < g[x].size(); i++) {
        if(vis[g[x][i]]) {
            if(c[x] == c[g[x][i]])
                isok = false;
            else
                continue;
        }
        else {
            if(cur == 0)
                dfs(g[x][i], 1);
            else
                dfs(g[x][i], 0);
        }
    }
}

int main() {
    // freopen("in.txt", "r", stdin);
    // freopen("out.txt", "w", stdout);
    int n, m, x, y;
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= m; i++) {
        scanf("%d%d", &x, &y);
        g[x].push_back(y);
        g[y].push_back(x);
        a.push_back(x);
    }
    dfs(1, 0);
    if(!isok)
        printf("NO\n");
    else {
        printf("YES\n");
        for(int i = 0; i < a.size(); i++) {
            printf("%d", c[a[i]]);
        }
    }
    return 0;
}