結果

問題 No.2536 同値性と充足可能性
コンテスト
ユーザー Carpenters-Cat
提出日時 2023-11-10 22:12:13
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,498 bytes
コンパイル時間 1,404 ms
コンパイル使用メモリ 97,324 KB
最終ジャッジ日時 2025-02-17 20:55:05
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 22 WA * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <stack>
#include <vector>
using namespace std;
#include <atcoder/dsu>
using namespace atcoder;
int main()
{
    int N;
    cin >> N;
    using P = pair<int, int>;
    std::vector<P> Eq, Neq;
    int M;
    cin >> M;
    while (M--) {
        int a, b;
        string s;
        cin >> a >> s >> b;
        if (s == "<==>") {
            Eq.emplace_back(--a, --b);
        } else {
            Neq.emplace_back(--a, --b);
        }
    }
    dsu D(N);
    for (auto [a, b] : Eq) {
        D.merge(a, b);
    }
    auto grp = D.groups();
    vector<int> ID(N);
    for (int i = 0; i < grp.size(); i++) {
        for (auto v : grp[i]) {
            ID[v] = i;
        }
    }
    vector<int> gr[200020];
    for (auto [u, v] : Neq) {
        u = ID[u];
        v = ID[v];
        gr[u].push_back(v);
        gr[v].push_back(u);
    }
    vector<int> did(N, -1);
    vector<int> ans;
    for (int i = 0; i < N; i++) {
        if (did[i] != -1) {
            continue;
        }
        did[i] = 0;
        vector<int> X[2];
        for (auto v : grp[i]) {
            X[0].push_back(v);
        }
        stack<int> st;
        st.push(0);
        bool ok = true;
        while (!st.empty()) {
            int u = st.top();
            st.pop();
            for (auto v : gr[u]) {
                if (did[v] == did[u]) {
                    ok = false;
                    break;
                }
                if (did[v] == -1) {
                    did[v] = did[u] ^ 1;
                    st.push(v);
                    for (auto x : grp[v]) {
                        X[did[v]].push_back(x);
                    }
                }
            }
        }
        if (!ok) {
            puts("No");
            return 0;
        }
        // for (int x = 0; x < 2; x++) {
        //     printf("%d : ", x);
        //     for (auto a : X[x]) {
        //         cout << a << " ";
        //     }
        //     cout << endl;
        // }
        auto& vc = (X[0].size() > X[1].size() ? X[0] : X[1]);
        for (auto v : vc) {
            ans.push_back(v + 1);
        }
    }
    // for (auto a : ans) {
    //     cout << a << " ";
    // }
    // cout << endl;
    if (ans.size() * 2 < N) {
        puts("No");
        return 0;
    } else {
        puts("Yes");
        cout << ans.size() << endl;
        sort(ans.begin(), ans.end());
        for (auto a : ans) {
            cout << a << (a == ans.back() ? "" : " ");
        }
        cout << endl;
    }
}
0