結果

問題 No.2319 Friends+
コンテスト
ユーザー magurofly
提出日時 2023-05-27 14:01:07
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 772 ms / 3,000 ms
コード長 806 bytes
コンパイル時間 2,016 ms
コンパイル使用メモリ 197,232 KB
最終ジャッジ日時 2025-02-13 09:15:07
ジャッジサーバーID
（参考情報） 		judge5 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 45
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
const int N_MAX = 20000;
int main() {
  int N, M; cin >> N >> M;
  vector<int> world_of(N);
  vector<bitset<N_MAX>> in_world(N);
  for (int i = 0; i < N; i++) {
    cin >> world_of[i]; world_of[i]--;
    in_world[world_of[i]][i] = true;
  }
  vector<bitset<N_MAX>> friend_of(N);
  for (int i = 0; i < M; i++) {
    int A, B; cin >> A >> B; A--; B--;
    friend_of[A][B] = friend_of[B][A] = true;
  }

  int Q;
  cin >> Q;
  for (int t = 0; t < Q; t++) {
    int X, Y; cin >> X >> Y; X--; Y--;
    if (world_of[X] != world_of[Y] && (friend_of[X] & in_world[world_of[Y]]).any()) {
      cout << "Yes\n";
      in_world[world_of[X]][X] = false;
      world_of[X] = world_of[Y];
      in_world[world_of[X]][X] = true;
    } else {
      cout << "No\n";
    }
  }
}
0