結果

問題 No.1170 Never Want to Walk
コンテスト
ユーザー mikianaaa
提出日時 2020-09-28 21:57:56
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,253 bytes
コンパイル時間 1,843 ms
コンパイル使用メモリ 172,376 KB
実行使用メモリ 12,544 KB
最終ジャッジ日時 2024-07-02 16:55:05
合計ジャッジ時間 9,596 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30 TLE * 1 -- * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define all(x) (x).begin(), (x).end()
#define ll long long
#define ld long double
#define INF 1000000000000000000
typedef pair<ll, ll> pll;

class UnionFind {
  public:
    vector<ll> par; // 各元の親を表す配列
    vector<ll> siz; // 素集合のサイズを表す配列(1 で初期化)

    // Constructor
    UnionFind(ll sz_) : par(sz_), siz(sz_, 1LL) {
        for (ll i = 0; i < sz_; ++i)
            par[i] = i; // 初期では親は自分自身
    }
    void init(ll sz_) {
        par.resize(sz_);
        siz.assign(sz_, 1LL); // resize だとなぜか初期化されなかった
        for (ll i = 0; i < sz_; ++i)
            par[i] = i; // 初期では親は自分自身
    }

    // Member Function
    // Find
    ll root(ll x) { // 根の検索
        while (par[x] != x) {
            x = par[x] = par[par[x]]; // x の親の親を x の親とする
        }
        return x;
    }

    // Union(Unite, Merge)
    bool merge(ll x, ll y) {
        x = root(x);
        y = root(y);
        if (x == y)
            return false;
        // merge technique（データ構造をマージするテク．小を大にくっつける）
        if (siz[x] < siz[y])
            swap(x, y);
        siz[x] += siz[y];
        par[y] = x;
        return true;
    }

    bool issame(ll x, ll y) { // 連結判定
        return root(x) == root(y);
    }

    ll size(ll x) { // 素集合のサイズ
        return siz[root(x)];
    }
};

vector<int> x;

int right(int key) { return upper_bound(all(x), key) - x.begin() - 1; }

int left(int key) { return lower_bound(all(x), key) - x.begin() - 1; }

int main() {
    int N, A, B;
    cin >> N >> A >> B;
    x.resize(N);
    rep(i, N) { cin >> x[i]; }
    UnionFind tree(N);

    rep(i, N) {
        int ans = 0;
        for (int j = left(x[i] + A) + 1; j <= left(x[i] + B); j++) {
            if (j >= 0 && j < N)
                tree.merge(i, j);
        }

        for (int j = left(x[i] - B) + 1; j <= right(x[i] - A); j++) {
            if (j >= 0 && j < N)
                tree.merge(i, j);
        }
    }

    rep(i, N) { cout << tree.size(i) << endl; }
}
0