結果

問題 No.1170 Never Want to Walk
ユーザー MASATO338MASATO338
提出日時 2020-08-14 22:59:24
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 368 ms / 2,000 ms
コード長 1,255 bytes
コンパイル時間 1,664 ms
コンパイル使用メモリ 173,608 KB
実行使用メモリ 8,448 KB
最終ジャッジ日時 2024-10-10 16:19:52
合計ジャッジ時間 7,870 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using ll = long long;
using namespace std;
const int INFint = 1e9+1;
const ll INFll = (ll)1e18+1;
ll MOD=1e9+7;

struct UnionFind{
  vector<int> par;
  vector<int> size_a;

  UnionFind(int N): par(N),size_a(N){
    for(int i=0;i<N;i++){
      par[i]=i;
      size_a[i] = 1;
    }
  }

  int root(int x){
    if(par[x]==x) return x;
    else return par[x]=root(par[x]);
  }

  void unite(int x,int y){
    int rx=root(x);
    int ry=root(y);
    if(rx!=ry){
      par[rx]=ry;
      size_a[ry] += size_a[rx];
      size_a[rx] = 0;
    }
  }

  bool sameroot(int x,int y){
    return root(x)==root(y);
  }

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

int main(){
  ll N,A,B;
  cin>>N>>A>>B;
  vector<ll> x(N);
  for(int i(0);i<N;i++){
    cin>>x[i];
  }
  UnionFind uf(N);
  int fmi(0);
  for(int i(0);i<N;i++){
    int h = x[i]+A,m = x[i]+B;
    int hi = lower_bound(x.begin(),x.end(),h) - x.begin();
    int mi = upper_bound(x.begin(),x.end(),m) - x.begin();
    // cout << "a";
    // cout << hi << " " << mi << endl;
    for(int j(max(hi,fmi-1));j<mi;j++){
      //cout << i << " " << j << endl;
      uf.unite(i,j);
    }
    fmi = mi;
  }
  for(int i(0);i<N;i++){
    cout << uf.size(i) << endl;
  }
  return 0;
}
0