結果

問題 No.5004 Room Assignment
ユーザー butsurizuki
提出日時 2021-11-29 17:48:00
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 176 ms / 5,000 ms
コード長 1,064 bytes
コンパイル時間 3,186 ms
実行使用メモリ 22,392 KB
スコア 137,193,942
平均クエリ数 7642.09
最終ジャッジ日時 2021-11-30 23:54:56
合計ジャッジ時間 25,878 ms
ジャッジサーバーID
(参考情報)
judge14 / judge13
純コード判定しない問題か言語
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 100
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>

using namespace std;

struct UnionFind {
  vector<int> data;
  UnionFind(int size) : data(size, -1) { }
  bool unionSet(int x, int y) {
    x = root(x); y = root(y);
    if (x != y) {
      if (data[y] < data[x]) swap(x, y);
      data[x] += data[y]; data[y] = x;
    }
    return x != y;
  }
  bool findSet(int x, int y) {
    return root(x) == root(y);
  }
  int root(int x) {
    return data[x] < 0 ? x : data[x] = root(data[x]);
  }
  int size(int x) {
    return -data[root(x)];
  }
};

int main(){
  int T,R;
  cin >> T >> R;
  int p=0;
  UnionFind uf(5555);
  vector<int> S(5555);
  for(int i=0;i<T;i++){
    int n;
    cin >> n;
    vector<pair<int,int>> act;
    for(int j=0;j<n;j++){
      p++;
      cin >> S[p];
      for(int k=p-1;k>=1;k--){
        if(abs(S[k]-S[p])<=2 && uf.size(k)+uf.size(p)<=R){
          act.push_back({k,p});
          uf.unionSet(k,p);
        }
      }
    }
    cout << act.size() << '\n';
    for(auto &nx : act){cout << nx.first << ' ' << nx.second << '\n';}
    fflush(stdout);
  }
  return 0;
}
0