結果

問題 No.1170 Never Want to Walk
ユーザー deuteridayo
提出日時 2021-02-04 23:49:10
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,651 bytes
コンパイル時間 1,739 ms
コンパイル使用メモリ 174,896 KB
実行使用メモリ 17,140 KB
最終ジャッジ日時 2024-07-01 04:45:17
合計ジャッジ時間 8,175 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 1 RE * 1
other AC * 6 WA * 21 RE * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
// #include<atcoder/all>
using namespace std;
// using namespace atcoder;
using lint = long long;
using graph = vector<vector<int>>;
#define endl '\n'
lint const mod = 1e9+7;
//long const mod = 998244353;
struct UnionFind {
    vector<int> par;
    
    UnionFind(int n) : par(n, -1) { }
    void init(int n) { par.assign(n, -1); }
    
    int root(int x) {
        if (par[x] < 0) return x;
        else return par[x] = root(par[x]);
    }
    
    bool same(int x, int y) {
        return root(x) == root(y);
    }
    
    bool unite(int x, int y) {
        x = root(x); y = root(y);
        if (x == y) return false;
        if (par[x] > par[y]) swap(x, y); // merge technique
        par[x] += par[y];
        par[y] = x;
        return true;
    }
    
    int size(int x) {
        return -par[root(x)];
    }
};

int main(){
    lint n,a,b;
    cin >> n >> a >> b;
    vector<lint>x(n);
    for(int i=0;i<n;i++)cin >> x[i];
    graph g(n);
    vector<int>tmp(n+1);
    for(int i=0;i<n;i++){
        auto lItr = lower_bound(x.begin(), x.end(), x[i]+a);
        auto rItr = upper_bound(x.begin(), x.end(), x[i] +b);
        int l = lItr - x.begin();
        int r = rItr - x.begin();
        tmp[l]++;
        tmp[r]--;
        g[i].push_back(l);
    }
    for(int i=1;i<n;i++){
        tmp[i] += tmp[i-1];
    }
    for(int i=1;i<n;i++){
        if(tmp[i] > 0 && tmp[i-1] > 0){
            g[i-1].push_back(i);
        }
    }
    UnionFind unif(n);
    for(int i=0;i<n;i++){
        for(int j: g[i]){
            unif.unite(i,j);
        }
    }
    for(int i=0;i<n;i++){
        cout << unif.size(i)<<endl;
    }

}
0