結果

問題 No.1170 Never Want to Walk
ユーザー spihill
提出日時 2021-01-09 18:33:52
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 914 bytes
コンパイル時間 2,243 ms
コンパイル使用メモリ 195,776 KB
最終ジャッジ日時 2025-01-17 15:49:11
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1 RE * 1
other AC * 20 WA * 2 RE * 15
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
struct UF {
    vector<int> par;
    UF(int n) : par(n, -1) {}
    int root(int v) {
        if (par[v] < 0) return v;
        return par[v] = root(par[v]);
    }
    int size(int v) {
        return -par[root(v)];
    }
    void unite(int u, int v) {
        u = root(u);
        v = root(v);
        if (u == v) return;
        par[u] += par[v];
        par[v] = u;
    }
};

int main() {
    int N, A, B; cin >> N >> A >> B;
    vector<int> v(N); for (auto& x : v) cin >> x;
    UF uf(N);
    for (int i = 0, l = 0, r = 0; i < N; i++) {
        while (l < N && v[l] < v[i] - B) l++;
        while (l < N && v[l] <= v[i] - A) uf.unite(i, l++);
        l = min(l, l-1);
        while (r < N && v[r] < v[i] + A) r++;
        while (r < N && v[r] <= v[i] + B) uf.unite(i, r++);
        r = min(r, r-1);
    }
    for (int i = 0; i < N; i++) cout << uf.size(i) << endl;
}
0