結果

問題 No.1170 Never Want to Walk
ユーザー ScottShelby
提出日時 2020-10-30 23:44:38
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 372 ms / 2,000 ms
コード長 1,216 bytes
コンパイル時間 1,640 ms
コンパイル使用メモリ 171,988 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-22 03:21:07
合計ジャッジ時間 8,973 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<int> x(n);
  rep(i, n) cin >> x[i];
  UnionFind uf(n);
  vector<int> sums(n);

  rep(i, n) {
    auto s = lower_bound(x.begin(), x.end(), x[i] + a) - x.begin();
    auto e = upper_bound(x.begin(), x.end(), x[i] + b) - x.begin() - 1;
    if (s > e) continue;
    uf.unite(i, s);
    sums[s]++;
    sums[e]--;
  }

  rep(i, n - 1) sums[i + 1] += sums[i];
  rep(i, n - 1) if (sums[i] > 0) uf.unite(i, i + 1);
  rep(i, n) {
    int ans = uf.size(i);
    cout << ans << endl;
  }
}
0