結果

問題 No.2319 Friends+
ユーザー hliuser1hliuser1
提出日時 2023-12-25 16:33:04
言語 C++17(clang)
(17.0.6 + boost 1.83.0)
結果
MLE  
実行時間 -
コード長 2,414 bytes
コンパイル時間 3,457 ms
コンパイル使用メモリ 167,296 KB
実行使用メモリ 814,592 KB
最終ジャッジ日時 2024-09-27 14:28:48
合計ジャッジ時間 10,132 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 MLE -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
testcase_42 -- -
testcase_43 -- -
testcase_44 -- -
testcase_45 -- -
testcase_46 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#pragma GCC optimize("O3")
#pragma GCC target("sse4")
using namespace std;

using HashSet = unordered_set<int>;
using FewCounter = unordered_map<int, vector<int>>;

signed main() {
    int N, M, Q;
    cin >> N >> M;
    int K = sqrt(M);
    vector<int>  world(N+1);      // world[i] = the world that person i is in
    vector<bool> is_many(N+1);    // is_many[i] = does person i have >= K friends
    FewCounter   num_few_of;  // num_few_of[i].at(w) = # friends of i in world j with < K friends
    vector<int>  many;            // many = [those with >= K friends]
    vector<HashSet> friends(N+1);

    for (int i = 1; i <= N; ++i) {
        cin >> world[i];
        num_few_of[i].assign(N+1, 0);
    }

    for (int i = 1; i <= M; ++i) {
        int a, b;
        cin >> a >> b;
        friends[a].insert(b);
        friends[b].insert(a);
    }

    for (int i = 1; i <= N; ++i) {
        int num_friends = friends[i].size();
        is_many[i] = (num_friends >= K);
        if (is_many[i])
            many.push_back(i);
    }

    for (int i = 1; i <= N; ++i) {
        if (is_many[i])
            for (int f : friends[i])
                if (!is_many[f]) {
                    int w = world[f];
                    ++num_few_of[i].at(w);
                }
    }

    auto solve_many = [&] (int x, int y) {
        int new_world = world[y];
        if (num_few_of[x].at(new_world) > 0)
            goto success;

        for (int m : many)
            if (friends[x].count(m) && world[m] == new_world)
                goto success;

        return "No\n";

        success:
        world[x] = new_world;
        return "Yes\n";
    };

    auto solve_few = [&] (int x, int y) {
        for (int f : friends[x])
            if (world[f] == world[y])
                goto success;

        return "No\n";

        success:
        int old_world = world[x];
        int new_world = world[y];
        for (int f : friends[x]) {
            if (!is_many[f]) continue;
            --num_few_of[f].at(old_world);
            ++num_few_of[f].at(new_world);
        }
        world[x] = new_world;
        return "Yes\n";
    };

    cin >> Q;
    while (Q--) {
        int x, y;
        cin >> x >> y;
        if (world[x] == world[y]) {
            cout << "No\n";
            continue;
        }
        auto ans = is_many[x] ? solve_many(x,y) : solve_few(x,y);
        cout << ans;
    }
}
0