結果

問題 No.1170 Never Want to Walk
コンテスト
ユーザー ScottShelby
提出日時 2020-10-15 15:37:42
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 337 ms / 2,000 ms
コード長 1,174 bytes
コンパイル時間 1,526 ms
コンパイル使用メモリ 171,716 KB
実行使用メモリ 6,400 KB
最終ジャッジ日時 2024-07-20 19:45:43
合計ジャッジ時間 8,619 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#define rep(i,N) for(ll (i)=0;(i)<(N);(i)++)
#define chmax(x,y) x=max(x,y)
#define chmin(x,y) x=min(x,y)
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
const int mod = 1000000007;
const int INF = 1001001001;

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;
  vector<ll> x(n);
  rep(i, n) cin >> x[i];

  vector<int> sum(n + 1);
  UnionFind uf(n);

  rep(i, n) {
    int l = lower_bound(x.begin(), x.end(), x[i] + a) - x.begin();
    int r = upper_bound(x.begin(), x.end(), x[i] + b) - x.begin() - 1;
    if (r < l) continue;
    uf.unite(i, l);
    sum[l]++;
    sum[r]--;
  }

  rep(i, n) sum[i + 1] += sum[i];
  rep(i, n) if (sum[i] > 0) uf.unite(i, i + 1);
  rep(i, n) cout << uf.size(i) << endl;
}
0