結果

問題 No.1640 簡単な色塗り
コンテスト
ユーザー sten_san
提出日時 2021-08-06 22:59:32
言語 C++17
(gcc 15.2.0 + boost 1.89.0)
結果
WA  
実行時間 -
コード長 3,427 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,038 ms
コンパイル使用メモリ 220,736 KB
最終ジャッジ日時 2025-01-23 16:08:40
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 47 WA * 6
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;

struct iofast_t {
    iofast_t() {
        ios::sync_with_stdio(false);
        cin.tie(nullptr);
    }
} iofast;

struct uns_t {} uns;
template <typename Element, typename Head, typename ...Args>
auto vec(Element init, Head arg, Args ...args) {
    if constexpr (sizeof...(Args) == 0) return std::vector(arg, init);
    else return std::vector(arg, vec(init, args...));
}
template <typename Element, typename Head, typename ...Args>
auto vec(uns_t, Head arg, Args ...args) {
    return vec(Element(), arg, args...);
}

template <typename T, typename Compare = less<T>>
T &chmin(T &l, T r, Compare &&f = less<T>()) { return l = min(l, r, f); }
template <typename T, typename Compare = less<T>>
T &chmax(T &l, T r, Compare &&f = less<T>()) { return l = max(l, r, f); }

#include <atcoder/dsu>

int main() {
    int n; cin >> n;

    auto g = vec<tuple<int, int>>(uns, n, 0);

    atcoder::dsu dsu1(n);
    for (int i = 0; i < n; ++i) {
        int a, b; cin >> a >> b; --a; --b;
        g[a].push_back({ b, i });
        g[b].push_back({ a, i });
        dsu1.merge(a, b);
    }

    auto groups = dsu1.groups();

    for (auto &group : groups) {
        set<int> edge;
        for (auto v : group) {
            for (auto [u, c] : g[v]) {
                edge.insert(c);
            }
        }

        if (size(group) < size(edge)) {
            cout << "No" << endl;
            return 0;
        }
    }

    auto ans = vec<int>(uns, n);
    auto used = vec<bool>(false, n);

    auto vis = vec<bool>(false, n);
    auto fin = vec<bool>(false, n);
    for (auto &group : groups) {
        stack<int> hist;

        int con = -1;
        auto dfs = [&](auto &&self, int v, int p = -1) -> bool {
            vis[v] = true;

            hist.push(v);
            for (auto [u, _] : g[v]) {
                if (p == _) {
                    continue;
                }

                if (fin[u]) {
                    continue;
                }

                if (vis[u] && !fin[u]) {
                    con = u;
                    return true;
                }

                if (self(self, u, _)) {
                    return true;
                }
            }
            hist.pop();

            fin[v] = true;
            return false;
        };

        if (!dfs(dfs, group[0])) {
            cout << "No" << endl;
            return 0;
        }

        set<int> cycle;

        int cur = con;
        while (!empty(hist)) {
            int u = hist.top(); hist.pop();

            for (auto [_, c] : g[cur]) {
                if (u != _) {
                    continue;
                }
                ans[c] = u;
                used[c] = true;
                break;
            }
            cur = u;

            cycle.insert(u);
            if (u == con) {
                break;
            }
        }

        for (auto v : cycle) {
            queue<int> que;

            que.push(v);
            while (!empty(que)) {
                int v = que.front(); que.pop();

                for (auto [u, c] : g[v]) {
                    if (used[c]) {
                        continue;
                    }
                    ans[c] = u;
                    used[c] = true;

                    que.push(u);
                }
            }
        }
    }

    cout << "Yes" << endl;
    for (auto v : ans) {
        cout << v + 1 << endl;
    }
}
0