結果

問題 No.2780 The Bottle Imp
コンテスト
ユーザー Today03
提出日時 2024-06-08 00:38:24
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,974 bytes
コンパイル時間 2,680 ms
コンパイル使用メモリ 213,208 KB
最終ジャッジ日時 2025-02-21 20:43:02
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 38 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

// #include "kyopro_library/graph/toposort.hpp"

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};
}

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 (count(member[0].begin(), member[0].end(), 0) == 0) {
        cout << "No" << endl;
        return 0;
    }

    queue<int> Q;
    Q.push(0);
    vector<bool> seen(N, false);
    seen[0] = true;

    int end_point = 0;

    while (!Q.empty()) {
        int now = Q.front();
        Q.pop();

        if (G[now].size() == 0) {
            end_point++;
        }

        for (int nxt : G[now]) {
            if (!seen[nxt]) {
                seen[nxt] = true;
                Q.push(nxt);
            }
        }
    }

    bool ans = true;
    for (bool x : seen) {
        if (!x) {
            ans = false;
        }
    }

    if (ans && end_point <= 1) {
        cout << "Yes" << endl;
    } else {
        cout << "No" << endl;
    }
}
0