結果

問題 No.1170 Never Want to Walk
ユーザー kenta255kenta255
提出日時 2020-08-14 23:03:57
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 465 ms / 2,000 ms
コード長 2,131 bytes
コンパイル時間 3,017 ms
コンパイル使用メモリ 200,864 KB
最終ジャッジ日時 2025-01-13 00:37:21
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define P pair<ll,ll>
#define FOR(I,A,B) for(ll I = ll(A); I < ll(B); ++I)
#define FORR(I,A,B) for(ll I = ll((B)-1); I >= ll(A); --I)
#define TO(x,t,f) ((x)?(t):(f))
#define SORT(x) (sort(x.begin(),x.end())) // 0 2 2 3 4 5 8 9
#define POSL(x,v) (lower_bound(x.begin(),x.end(),v)-x.begin()) //xi>=v  x is sorted
#define POSU(x,v) (upper_bound(x.begin(),x.end(),v)-x.begin()) //xi>v  x is sorted
#define NUM(x,v) (POSU(x,v)-POSL(x,v))  //x is sorted
#define REV(x) (reverse(x.begin(),x.end())) //reverse
ll gcd_(ll a,ll b){if(a%b==0)return b;return gcd_(b,a%b);}
ll lcm_(ll a,ll b){ll c=gcd_(a,b);return ((a/c)*(b/c)*c);}
#define NEXTP(x) next_permutation(x.begin(),x.end())
const ll INF=ll(1e16)+ll(7);
const ll MOD=1000000007LL;
#define out(a) cout<<fixed<<setprecision((a))
//tie(a,b,c) = make_tuple(10,9,87);
#define pop_(a) __builtin_popcount((a))
ll keta(ll a){ll r=0;while(a){a/=10;r++;}return r;}
ll ketawa(ll a){ll r=0;while(a){r+=a%10;a/=10;}return r;}

class UnionFind{
  vector<int> par,siz;
  //0-indexed
  void init(int n){
    component = n;
    par.resize(n);
    for(int i=0;i<n;++i)par[i]=i;
    siz.resize(n,1);
    msiz = 1;
  }
public:
  int msiz,component;
  UnionFind(int n_){init(n_);}
  int root(int x){
    if(par[x]==x)return x;
    return par[x] = root(par[x]);
  }
  void unite(int x,int y){
    x = root(x);
    y = root(y);
    if(x==y)return;
    siz[y] += siz[x];
    siz[x] = 0;
    par[x] = y;
    msiz = max(msiz,siz[y]);
    component--;
  }
  int size(int x){
    return siz[root(x)];
  }
  bool same(int x,int y){
    return root(x)==root(y);
  }
};

int main(){

  ll N,A,B;
  cin >> N >> A >> B;

  vector<ll> X(N+1,INF);

  FOR(i,0,N){
    cin >> X[i];
  }
  UnionFind uf(N);
  vector<ll> imos(N+1,0);

  FOR(i,0,N){
    ll l = POSL(X,X[i]+A);
    ll r = POSU(X,X[i]+B);
    if(l+1 <= r){
      imos[l]++;
      imos[r-1]--;
      uf.unite(i,l);
    }
  }

  FOR(i,0,N){
    imos[i+1] += imos[i];
  }
  FOR(i,0,N-1){
    if(imos[i]){
      uf.unite(i,i+1);
    }
  }

  FOR(i,0,N){
    cout << uf.size(i) << endl;
  }

}
0