結果

問題 No.2780 The Bottle Imp
ユーザー Today03Today03
提出日時 2024-06-08 00:46:53
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 116 ms / 2,000 ms
コード長 3,651 bytes
コンパイル時間 2,730 ms
コンパイル使用メモリ 221,528 KB
実行使用メモリ 47,032 KB
最終ジャッジ日時 2024-06-08 10:40:33
合計ジャッジ時間 5,270 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 1 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 60 ms
16,128 KB
testcase_08 AC 56 ms
16,128 KB
testcase_09 AC 58 ms
16,128 KB
testcase_10 AC 56 ms
16,000 KB
testcase_11 AC 56 ms
16,128 KB
testcase_12 AC 111 ms
33,056 KB
testcase_13 AC 116 ms
33,052 KB
testcase_14 AC 35 ms
6,144 KB
testcase_15 AC 35 ms
6,272 KB
testcase_16 AC 33 ms
6,400 KB
testcase_17 AC 33 ms
6,272 KB
testcase_18 AC 35 ms
6,272 KB
testcase_19 AC 34 ms
6,272 KB
testcase_20 AC 35 ms
6,400 KB
testcase_21 AC 36 ms
6,400 KB
testcase_22 AC 21 ms
7,040 KB
testcase_23 AC 29 ms
6,528 KB
testcase_24 AC 47 ms
13,824 KB
testcase_25 AC 82 ms
23,220 KB
testcase_26 AC 42 ms
9,856 KB
testcase_27 AC 29 ms
8,808 KB
testcase_28 AC 29 ms
8,804 KB
testcase_29 AC 42 ms
14,848 KB
testcase_30 AC 34 ms
15,968 KB
testcase_31 AC 54 ms
15,524 KB
testcase_32 AC 19 ms
5,376 KB
testcase_33 AC 68 ms
25,916 KB
testcase_34 AC 106 ms
47,032 KB
testcase_35 AC 16 ms
5,376 KB
testcase_36 AC 2 ms
5,376 KB
testcase_37 AC 1 ms
5,376 KB
testcase_38 AC 16 ms
5,376 KB
testcase_39 AC 52 ms
15,524 KB
testcase_40 AC 51 ms
15,652 KB
testcase_41 AC 49 ms
15,524 KB
testcase_42 AC 2 ms
5,376 KB
testcase_43 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int INF = 1e9 + 10;
const ll INFL = 4e18;

struct SCC {
    vector<vector<int>> members, graph_decomposed;
    vector<int> group;
};

SCC scc_decomposition(const vector<vector<int>> &G) {
    int n = G.size();
    vector<vector<int>> G2(n);
    for (int i = 0; i < n; i++) {
        for (int nxt : G[i]) {
            G2[nxt].push_back(i);
        }
    }
    vector<int> order(n), component(n, -1);
    vector<bool> vst(n);
    auto F = [&](auto F, int now) -> void {
        vst[now] = true;
        for (int nxt : G[now]) {
            if (!vst[nxt]) {
                F(F, nxt);
            }
        }
        order.push_back(now);
    };
    auto F2 = [&](auto F2, int now, int idx) -> void {
        component[now] = idx;
        for (int nxt : G2[now]) {
            if (component[nxt] == -1) {
                F2(F2, nxt, idx);
            }
        }
    };
    for (int i = 0; i < n; i++) {
        if (!vst[i]) {
            F(F, i);
        }
    }
    int idx = 0;
    reverse(order.begin(), order.end());
    for (int now : order) {
        if (component[now] == -1) {
            F2(F2, now, idx);
            idx++;
        }
    }
    int n_n = *max_element(component.begin(), component.end()) + 1;
    vector<vector<int>> ret(n_n);
    for (int i = 0; i < n; i++) {
        ret[component[i]].push_back(i);
    }
    vector<vector<int>> ret2(n_n);
    for (int i = 0; i < n; i++) {
        for (int j : G[i]) {
            if (component[i] != component[j]) {
                ret2[component[i]].push_back(component[j]);
            }
        }
    }
    for (int i = 0; i < n_n; i++) {
        sort(ret2[i].begin(), ret2[i].end());
        ret2[i].erase(unique(ret2[i].begin(), ret2[i].end()), ret2[i].end());
    }
    return {ret, ret2, component};
}

bool topological_sort(const vector<vector<int>> &G) {
    int n = G.size();
    vector<int> indeg(n);
    for (int i = 0; i < n; i++) {
        for (int nxt : G[i]) {
            indeg[nxt]++;
        }
    }
    vector<int> ret1, ret2;
    {
        priority_queue<int> Q;
        for (int i = 0; i < n; i++) {
            if (indeg[i] == 0) {
                Q.push(i);
            }
        }
        vector<int> ret;
        while (!Q.empty()) {
            int now = Q.top();
            Q.pop();
            for (int nxt : G[now]) {
                indeg[nxt]--;
                if (indeg[nxt] == 0) {
                    Q.push(nxt);
                }
            }
            ret.push_back(now);
        }
        ret1 = ret;
    }
    {
        priority_queue<int, vector<int>, greater<>> Q;
        for (int i = 0; i < n; i++) {
            if (indeg[i] == 0) {
                Q.push(i);
            }
        }
        vector<int> ret;
        while (!Q.empty()) {
            int now = Q.top();
            Q.pop();
            for (int nxt : G[now]) {
                indeg[nxt]--;
                if (indeg[nxt] == 0) {
                    Q.push(nxt);
                }
            }
            ret.push_back(now);
        }
        ret2 = ret;
    }
    return ret1 == ret2;
}

int main() {
    int N;
    cin >> N;
    vector<vector<int>> G(N);
    for (int i = 0; i < N; i++) {
        int m;
        cin >> m;
        for (int j = 0; j < m; j++) {
            int a;
            cin >> a;
            a--;
            G[i].push_back(a);
        }
    }

    auto [member, nG, group] = scc_decomposition(G);

    if (group[0] != 0) {
        cout << "No" << endl;
    } else if (!topological_sort(nG)) {
        cout << "No" << endl;
    } else {
        cout << "Yes" << endl;
    }
}
0