結果

問題 No.1170 Never Want to Walk
ユーザー mikianaaamikianaaa
提出日時 2020-09-29 13:51:02
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 370 ms / 2,000 ms
コード長 2,290 bytes
コンパイル時間 1,614 ms
コンパイル使用メモリ 172,772 KB
実行使用メモリ 8,704 KB
最終ジャッジ日時 2024-07-03 20:17:11
合計ジャッジ時間 9,039 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

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(); }

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

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

    vector<ll> sum(N + 100);
    rep(i, N) {
        int l = left(x[i] + A);
        int r = right(x[i] + B) - 1;
        if (l > r)
            continue;
        if (l >= 0 && l < N)
            tree.merge(i, l);
        sum[l]++;
        sum[r]--;
    }

    rep(i, N) { sum[i + 1] += sum[i]; }

    rep(i, N - 1) {
        if (sum[i] > 0)
            tree.merge(i, i + 1);
    }

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