結果

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

ソースコード

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 = max(l-1, 0);
        while (r < N && v[r] < v[i] + A) r++;
        while (r < N && v[r] <= v[i] + B) uf.unite(i, r++);
        r = max(r-1, 0);
    }
    for (int i = 0; i < N; i++) cout << uf.size(i) << endl;
}
0