結果

問題 No.1170 Never Want to Walk
コンテスト
ユーザー hayaten
提出日時 2020-08-15 15:35:44
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 449 ms / 2,000 ms
コード長 1,533 bytes
コンパイル時間 2,705 ms
コンパイル使用メモリ 196,084 KB
最終ジャッジ日時 2025-01-13 01:31:21
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 0; i < (n); i++)
#define rrep(i, n) for (int i = (n - 1); i >= 0; i--)
#define ALL(v) v.begin(), v.end()
using namespace std;
using P = pair<int, int>;
typedef long long ll;

struct UnionFind {
  vector<int> par; //par[i]:iの親の番号

  //最初は全てが根であるとする。-1のときに根である。
  UnionFind(int N) : par(N, -1){}

  // データxが属する木の根を再帰で得る：root(x) = {xの木の根}
  int root(int x) {
    if(par[x] < 0) return x;
    else return par[x] = root(par[x]);
  }
  // データxが根かどうか
  bool cond_root(int x){
    if(par[x] < 0) return true;
    else return false;
  }
  //xとyが同じ集合に属するか
  bool issame(int x, int y) {
    return root(x) == root(y);
  }

  bool merge(int x, int y) {
    x = root(x); y = root(y);
    if(x == y) return false;
    if(par[x] > par[y]) swap(x,y);
    par[x] += par[y];
    par[y] = x;
    return true;
  }

  int size(int x) {
    return -par[root(x)];
  }
};

int main() {
  int n, a, b;
  cin >> n >> a >> b;
  vector<int> x(n);
  vector<int> imos(n);
  rep(i, n) cin >> x[i];
  UnionFind uf(n);
  rep(i, n) {
    int l = lower_bound(ALL(x), x[i] + a) - x.begin();
    int r = upper_bound(ALL(x), x[i] + b) - x.begin() - 1;
    if(l>r)continue;
    uf.merge(i, l);
    imos[l]++;
    imos[r]--;
  }
  rep(i, n - 1) imos[i + 1] += imos[i];
  rep(i, n - 1) {
    if(imos[i] > 0)uf.merge(i, i+1);
  }
  rep(i, n) {
    cout << uf.size(i) << endl;
  }
}
0