結果

問題 No.1640 簡単な色塗り
ユーザー sten_sansten_san
提出日時 2021-08-06 22:59:32
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,427 bytes
コンパイル時間 2,870 ms
コンパイル使用メモリ 230,220 KB
実行使用メモリ 26,360 KB
最終ジャッジ日時 2024-06-29 16:08:27
合計ジャッジ時間 14,386 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 162 ms
15,232 KB
testcase_05 AC 215 ms
26,360 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 116 ms
10,040 KB
testcase_11 WA -
testcase_12 AC 87 ms
8,960 KB
testcase_13 AC 184 ms
15,020 KB
testcase_14 AC 208 ms
15,348 KB
testcase_15 AC 63 ms
7,040 KB
testcase_16 AC 76 ms
8,272 KB
testcase_17 AC 147 ms
12,888 KB
testcase_18 WA -
testcase_19 AC 80 ms
8,160 KB
testcase_20 AC 106 ms
9,480 KB
testcase_21 AC 89 ms
8,812 KB
testcase_22 WA -
testcase_23 AC 125 ms
10,788 KB
testcase_24 AC 31 ms
5,376 KB
testcase_25 WA -
testcase_26 AC 128 ms
10,720 KB
testcase_27 AC 56 ms
6,656 KB
testcase_28 AC 175 ms
14,992 KB
testcase_29 WA -
testcase_30 AC 9 ms
5,376 KB
testcase_31 AC 83 ms
15,004 KB
testcase_32 AC 66 ms
13,668 KB
testcase_33 AC 38 ms
10,080 KB
testcase_34 AC 58 ms
12,448 KB
testcase_35 AC 44 ms
10,192 KB
testcase_36 AC 11 ms
5,376 KB
testcase_37 AC 15 ms
5,888 KB
testcase_38 AC 67 ms
12,960 KB
testcase_39 AC 24 ms
7,724 KB
testcase_40 AC 18 ms
7,168 KB
testcase_41 AC 54 ms
11,084 KB
testcase_42 AC 23 ms
7,768 KB
testcase_43 AC 34 ms
9,176 KB
testcase_44 AC 30 ms
8,572 KB
testcase_45 AC 26 ms
7,632 KB
testcase_46 AC 8 ms
5,376 KB
testcase_47 AC 7 ms
5,376 KB
testcase_48 AC 74 ms
14,788 KB
testcase_49 AC 4 ms
5,376 KB
testcase_50 AC 2 ms
5,376 KB
testcase_51 AC 2 ms
5,376 KB
testcase_52 AC 205 ms
19,688 KB
testcase_53 WA -
07_evil_01.txt AC 515 ms
30,360 KB
07_evil_02.txt AC 710 ms
35,884 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#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