結果

問題 No.1170 Never Want to Walk
ユーザー outlineoutline
提出日時 2021-01-12 01:02:00
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,329 bytes
コンパイル時間 1,815 ms
コンパイル使用メモリ 154,232 KB
実行使用メモリ 62,068 KB
最終ジャッジ日時 2023-08-13 13:47:31
合計ジャッジ時間 6,520 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,376 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 1 ms
4,380 KB
testcase_04 AC 1 ms
4,376 KB
testcase_05 AC 1 ms
4,376 KB
testcase_06 AC 1 ms
4,376 KB
testcase_07 AC 1 ms
4,380 KB
testcase_08 AC 1 ms
4,380 KB
testcase_09 AC 2 ms
4,376 KB
testcase_10 AC 1 ms
4,380 KB
testcase_11 AC 1 ms
4,380 KB
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 AC 2 ms
4,380 KB
testcase_19 WA -
testcase_20 WA -
testcase_21 AC 3 ms
4,380 KB
testcase_22 AC 3 ms
4,376 KB
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 AC 3 ms
4,376 KB
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 <iostream>
#include <vector>
#include <algorithm>
#include <cmath>
#include <queue>
#include <string>
#include <map>
#include <set>
#include <stack>
#include <tuple>
#include <deque>
#include <array>
#include <numeric>
#include <bitset>
#include <iomanip>
#include <cassert>
#include <chrono>
#include <random>
#include <limits>
#include <iterator>
#include <functional>
#include <sstream>
#include <fstream>
#include <complex>
#include <cstring>
#include <unordered_map>
#include <unordered_set>
using namespace std;

using ll = long long;
using P = pair<int, int>;
constexpr int INF = 1001001001;
// constexpr int mod = 1000000007;
constexpr int mod = 998244353;

template<class T>
inline bool chmax(T& x, T y){
    if(x < y){
        x = y;
        return true;
    }
    return false;
}
template<class T>
inline bool chmin(T& x, T y){
    if(x > y){
        x = y;
        return true;
    }
    return false;
}

struct UnionFind{
    int sz;
    vector<int> par;
    vector<int> rank;

    UnionFind(int n) : sz(n) {
        par.resize(sz);
        rank.assign(sz, 0);
        for(int i = 0; i < sz; ++i){
            par[i] = i;
        }
    }

    int find(int x){
        if(par[x] == x) 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(rank[x] < rank[y])   par[x] = y;
        else{
            par[y] = x;
            if(rank[x] == rank[y])  ++rank[x];
        }
        return true;
    }
};

int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, a, b;
    cin >> n >> a >> b;
    vector<int> x(n);
    for(int i = 0; i < n; ++i)  cin >> x[i];

    int V = 1;
    while(V * 2 < n)    V <<= 1;
    V <<= 1;
    vector<vector<int>> g(V * 2);
    for(int k = 1; k < V; ++k){
        int chl = k << 1, chr = k << 1 | 1;
        g[k].emplace_back(chl);
        g[chl].emplace_back(k);
        g[k].emplace_back(chr);
        g[chr].emplace_back(k);
    }
    
    for(int i = 0; i < n - 1; ++i){
        int l = lower_bound(begin(x), end(x), x[i] + a) - begin(x);
        int r = upper_bound(begin(x), end(x), x[i] + b) - begin(x);
        if(r - l <= 0)  continue;
        vector<int> node(1, V + i);
        for(l += V, r += V; l < r; l >>= 1, r >>= 1){
            if(l & 1)   node.emplace_back(l++);
            if(r & 1)   node.emplace_back(--r);
        }
        g.emplace_back(node);
    }

    int N = g.size();
    UnionFind uf(N);
    vector<bool> used(N);
    for(int v = V * 2; v < N; ++v){
        for(int u : g[v]){
            used[u] = true;
            uf.unite(v, u);
        }
    }
    
    auto dfs = [&](auto&& self, int k, int l, int r, bool f = false) -> void {
        if(k >= V)  return;
        f |= used[k];
        if(f){
            uf.unite(k, k << 1);
            uf.unite(k, k << 1 | 1);
        }
        self(self, k << 1, l, (l + r) >> 2, f);
        self(self, k << 1 | 1, (l + r) >> 2, f);
    };
    dfs(dfs, 1, 0, V * 2);

    map<int, int> mp;
    for(int i = 0; i < n; ++i){
        int v = V + i;
        mp[uf.find(v)] += 1;
    }

    for(int i = 0; i < n; ++i){
        int v = V + i;
        cout << mp[uf.find(v)] << '\n';
    }

    return 0;
}
0