結果

問題 No.2319 Friends+
ユーザー t98slider
提出日時 2023-05-26 23:08:44
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,447 bytes
コンパイル時間 2,177 ms
コンパイル使用メモリ 177,108 KB
実行使用メモリ 26,880 KB
最終ジャッジ日時 2024-12-25 10:30:03
合計ジャッジ時間 35,331 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 34 WA * 9 TLE * 2
権限があれば一括ダウンロードができます

ソースコード

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 = 500;
    vector<int> pos(n);
    //vector<set<int>> world(n);
    vector<multiset<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++){
        for(auto &&j : g[i]){
            if(g[j].size() < sqN) fri[i].insert(pos[j]);
            else G[i].emplace_back(j);
        }
    }
    int Q;
    cin >> Q;
    while(Q--){
        cin >> u >> v;
        u--, v--;
        if(pos[u] == pos[v]){
            cout << "No" << '\n';
            continue;
        }
        bool flag = (fri[u].find(pos[v]) != fri[u].end());
        if(!flag){
            for(auto &&c : G[v]){
                if(pos[c] == pos[v]){
                    flag = true;
                    break;
                }
            }
        }
        cout << (flag ? "Yes" : "No") << '\n';
        if(!flag) continue;
        if(g[u].size() < sqN){
            for(auto &&c : g[u]){
                fri[c].erase(fri[c].find(pos[u]));
                fri[c].insert(pos[v]);
            }
        }
        pos[u] = pos[v];
    }
}
0