結果

問題 No.2319 Friends+
ユーザー t98slider
提出日時 2023-05-26 23:18:21
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 257 ms / 3,000 ms
コード長 1,540 bytes
コンパイル時間 1,685 ms
コンパイル使用メモリ 175,056 KB
実行使用メモリ 29,440 KB
最終ジャッジ日時 2024-12-25 10:49:16
合計ジャッジ時間 9,492 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 45
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n, m, u, v;
    cin >> n >> m;
    int sqN = 640;
    vector<int> pos(n);
    //vector<set<int>> world(n);
    vector<vector<int>> fri(n);
    vector<vector<int>> g(n);
    vector<vector<int>> G(n);
    for(int i = 0; i < n; i++){
        cin >> pos[i];
        pos[i]--;
    }
    for(int i = 0; i < m; i++){
        cin >> u >> v;
        u--, v--;
        g[u].emplace_back(v);
        g[v].emplace_back(u);
    }
    for(int i = 0; i < n; i++){
        if(g[i].size() >= sqN){
            fri[i].resize(n, 0);
        }
    }
    for(int i = 0; i < n; i++){
        for(auto &&j : g[i]){
            if(g[j].size() >= sqN){
                G[i].emplace_back(j);
                fri[j][pos[i]]++;
            }
        }
    }
    int Q;
    cin >> Q;
    while(Q--){
        cin >> u >> v;
        u--, v--;
        if(pos[u] == pos[v]){
            cout << "No" << '\n';
            continue;
        }
        bool flag = false;
        if(g[u].size() < sqN){
            for(auto &&c : g[u]){
                if(pos[c] == pos[v]){
                    flag = true;
                    break;
                }
            }
        }else{
            flag = fri[u][pos[v]] >= 1;
        }
        cout << (flag ? "Yes" : "No") << '\n';
        if(!flag) continue;
        for(auto &&c : G[u]){
            fri[c][pos[u]]--;
            fri[c][pos[v]]++;
        }
        pos[u] = pos[v];
    }
}
0