結果

問題 No.1170 Never Want to Walk
ユーザー deuteridayodeuteridayo
提出日時 2021-02-05 11:32:30
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,718 bytes
コンパイル時間 1,815 ms
コンパイル使用メモリ 174,228 KB
実行使用メモリ 11,064 KB
最終ジャッジ日時 2024-07-02 00:59:17
合計ジャッジ時間 5,800 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
権限があれば一括ダウンロードができます

ソースコード

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 find(int x) {
        if (par[x] < 0) return x;
        else return par[x] = find(par[x]);
    }
    
    bool same(int x, int y) {
        return find(x) == find(y);
    }
    
    bool unite(int x, int y) {
        x = find(x); y = find(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[find(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);
    UnionFind unif(n);
    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();
        
        if(l < r){
            unif.unite(i,l);
            tmp[l]++;
        tmp[r-1]--;
        }
        // cout << i<<" " << l << " " << r << endl;
    }
    for(int i=1;i<n;i++){
        tmp[i] += tmp[i-1];
    }
    // for(int i=0;i<n;i++)cout << tmp[i]<<" ";
    cout << endl;

    for(int i=1;i<n;i++){
        if(tmp[i] > 0 && tmp[i-1] > 0){
            unif.unite(i,i-1);
        }
    }

    for(int i=0;i<n;i++){
        cout << unif.size(i)<<endl;
    }

}
0