結果

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

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using V = vector<int>;
#define rep(i, n) for(int i=0; i<int(n); ++i)
#define all(a) a.begin(), a.end()


struct UnionFind {
  vector<int> d;
  UnionFind(int n=0) : d(n, -1) {}
  int find(int x) {
    if (d[x] < 0) return x;
    return d[x] = find(d[x]);
  }
  bool unite(int x, int y) {
    x = find(x); y = find(y);
    if (x == y) return false;
    if (d[x] > d[y]) swap(x, y);
    d[x] += d[y];
    d[y] = x;
    return true;
  }
  bool same(int x, int y) { return find(x) == find(y); }
  int size(int x) { return -d[find(x)]; }
};


int main() {
  int n, a, b; cin >> n >> a >> b;
  V x(n);
  rep(i, n) cin >> x[i];

// assert //
  assert(n >= 2 && n <= int(2e5));
  assert(1 <= a && a <= b && b <= int(1e9));
  assert(0 <= x[0] && x[n-1] <= int(1e9));
  rep(i, n-1) assert(x[i] < x[i+1]);
/////////// 

  UnionFind uf(n);
  V y(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.unite(i, l);
    ++y[l];
    --y[r];
  }
  rep(i, n-1) y[i+1] += y[i];
  rep(i, n-1) if (y[i] > 0) uf.unite(i, i+1);

  rep(i, n) cout << uf.size(i) << "\n";
  return 0;
}
0