#846202 (C++14) No.1170 Never Want to Walk

提出ソース

結果

問題	No.1170 Never Want to Walk
ユーザー	AnchorBlues
提出日時	2023-03-04 18:14:57
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	WA
実行時間	-
コード長	11,359 bytes
コンパイル時間	2,306 ms
コンパイル使用メモリ	187,864 KB
実行使用メモリ	17,676 KB
最終ジャッジ日時	2023-10-18 04:37:28
合計ジャッジ時間	9,540 ms
ジャッジサーバーID （参考情報）	judge15 / judge13

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 4,348 KB
testcase_01	AC	2 ms 4,348 KB
testcase_02	WA	-
testcase_03	AC	1 ms 4,348 KB
testcase_04	WA	-
testcase_05	AC	2 ms 4,348 KB
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	AC	2 ms 4,348 KB
testcase_10	AC	2 ms 4,348 KB
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	-

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>

using namespace std;
using ll = long long;
template <typename T>
using min_priority_queue = priority_queue<T, vector<T>, greater<T>>;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
using Graph = vector<vector<int>>;

const ll INF = 1LL << 60;

template <class T>
void chmax(T& a, T b) {
    if (b > a) a = b;
}
template <class T>
void chmin(T& a, T b) {
    if (b < a) a = b;
}

template <typename T, typename S>
std::ostream& operator<<(std::ostream& os, const pair<T, S>& x) noexcept {
    return os << "(" << x.first << ", " << x.second << ")";
}

template <typename T>
void print_vector(vector<T> a) {
    cout << '[';
    for (int i = 0; i < a.size(); i++) {
        cout << a[i];
        if (i != a.size() - 1) {
            cout << ", ";
        }
    }
    cout << ']' << endl;
}

class BIT {
   public:
    // コンストラクタ
    BIT();
    BIT(int);

    // 区間加算
    void add(int, int, ll);

    // i番目の要素への加算
    void add(int, ll);

    // 区間和
    ll sum(int, int) const;

    // i番目の要素の値
    ll at(int) const;

   private:
    void add_sub(int, int, ll);
    ll sum_sub(int, int) const;
    vector<vector<ll>> _bit;
    int _size;
};

// コンストラクタ
BIT::BIT() {}

BIT::BIT(int N) {
    _size = N + 1;
    _bit = vector<vector<ll>>(2, vector<ll>(_size, 0));
}

void BIT::add_sub(int p, int i, ll x) {
    for (int idx = i; idx < _size; idx += (idx & -idx)) {
        _bit[p][idx] += x;
    }
}

void BIT::add(int l, int r, ll x) {
    l++;
    r++;
    add_sub(0, l, -x * (l - 1));
    add_sub(0, r, x * (r - 1));
    add_sub(1, l, x);
    add_sub(1, r, -x);
}

void BIT::add(int i, ll x) { this->add(i, i + 1, x); }

ll BIT::sum_sub(int p, int i) const {
    ll s = 0;
    for (int idx = i; idx > 0; idx -= (idx & -idx)) {
        s += _bit[p][idx];
    }
    return s;
}

ll BIT::sum(int l, int r) const {
    ll sl = sum_sub(0, l) + sum_sub(1, l) * l;
    ll sr = sum_sub(0, r) + sum_sub(1, r) * r;
    return sr - sl;
}

ll BIT::at(int i) const { return this->sum(i, i + 1); }

class UnionFind {
   public:
    // コンストラクタ
    UnionFind();
    UnionFind(int);

    // 根を求める
    int root(int x) const;

    // x と y が同じグループに属するかどうか (根が一致するかどうか)
    bool issame(int x, int y) const;

    // x を含むグループと y を含むグループとを併合する
    bool unite(int x, int y);

    // 要素の追加。まだ追加されていない要素だったらその要素だけから成る集合を作る
    void add_elem(int x);
    // x を含むグループのサイズ
    int size(int x) const;

    // 島の個数
    int n_unions() const;

    // 要素xがすでに追加されているかどうかを判定
    bool is_added(int x) const;

   private:
    // 経路圧縮する。根を返す
    int path_compression(int x);
    std::vector<int> parents;
    std::vector<int> sizes;
};

// コンストラクタ
UnionFind::UnionFind() {}

UnionFind::UnionFind(int N) {
    parents = std::vector<int>(N, -1);
    sizes = std::vector<int>(N, -1);
}

// 根を求める
int UnionFind::root(int x) const {
    if (parents[x] == -1) return x;  // x が根の場合は x を返す

    return root(parents[x]);
}

// x と y が同じグループに属するかどうか (根が一致するかどうか)
bool UnionFind::issame(int x, int y) const { return root(x) == root(y); }

// x を含むグループと y を含むグループとを併合する
bool UnionFind::unite(int x, int y) {
    // x, y をそれぞれ根まで移動する
    if (!is_added(x)) sizes[x] = 1;
    if (!is_added(y)) sizes[y] = 1;
    int root_x = path_compression(x);
    int root_y = path_compression(y);

    // すでに同じグループのときは何もしない
    if (root_x == root_y) return false;

    if (!is_added(root_x)) sizes[root_x] = 1;
    if (!is_added(root_y)) sizes[root_y] = 1;

    // union by size (y 側のサイズが小さくなるようにする)
    if (sizes[root_x] < sizes[root_y]) std::swap(root_x, root_y);

    // y を x の子とする
    parents[root_y] = root_x;
    sizes[root_x] += sizes[root_y];
    return true;
}

void UnionFind::add_elem(int x) {
    // すでに追加されていたら何もしない
    if (is_added(x)) return;
    sizes[x] = 1;
}

// x を含むグループのサイズ。x が追加されていない要素の場合は「1」を返す
int UnionFind::size(int x) const {
    if (is_added(x)) return sizes[root(x)];
    return 1;
}

// 島の個数
int UnionFind::n_unions() const {
    std::unordered_set<int> s;
    for (size_t i = 0; i < sizes.size(); i++)
        if (is_added(i)) s.insert(root(i));

    return s.size();
}

// 経路圧縮をしながら根を求める
int UnionFind::path_compression(int x) {
    if (parents[x] == -1) return x;  // x が根の場合は x を返す
    parents[x] = path_compression(parents[x]);
    return parents[x];
}

bool UnionFind::is_added(int x) const { return sizes[x] != -1; }

class LazySegmentTree {
   public:
    // コンストラクタ
    LazySegmentTree();
    LazySegmentTree(int, ll, function<ll(ll, ll)> op);

    // 値の更新
    void update(int, ll);

    // 区間クエリの処理
    ll query(int, int);

    // 区間の更新
    void update(int, int, ll);

    // 木の可視化
    void show_tree() const;

    // サイズ
    int size() const;

   private:
    void update_sub(int, int, ll, int, int, int);
    ll query_sub(int, int, int, int, int);
    void eval(int k);
    function<ll(ll, ll)> _op;
    int _size;
    ll _default_value;
    std::vector<ll> data;
    std::vector<ll> lazy;
};

// コンストラクタ
LazySegmentTree::LazySegmentTree() {}

LazySegmentTree::LazySegmentTree(int N, ll default_value,
                                 function<ll(ll, ll)> op) {
    // サイズ: 2の累乗かつN以上の整数の中で最小の値
    _op = op;
    _size = 1;
    while (true) {
        if (_size >= N) break;
        _size = _size * 2;
    }
    _default_value = default_value;
    data = std::vector<ll>(2 * _size - 1, _default_value);
    lazy = std::vector<ll>(2 * _size - 1, _default_value);
}

// 配列のk番目を更新
void LazySegmentTree::eval(int k) {
    if (lazy[k] == _default_value) return;  // 更新するものが無ければ終了
    if (k < _size - 1) {                    // 葉でなければ子に伝搬
        lazy[k * 2 + 1] = lazy[k];
        lazy[k * 2 + 2] = lazy[k];
    }
    // 自身を更新
    data[k] = lazy[k];
    lazy[k] = _default_value;
}

// i 番目の値をxに変更
void LazySegmentTree::update(int i, ll v) {
    // i: 葉ノードの番号
    i += _size - 1;
    data[i] = v;
    while (true) {
        if (i <= 0) break;
        i = (i - 1) / 2;
        data[i] = _op(data[i * 2 + 1], data[i * 2 + 2]);
    }
}

ll LazySegmentTree::query_sub(int a, int b, int k, int l, int r) {
    eval(k);
    if ((r <= a) || (b <= l)) return _default_value;
    if ((a <= l) && (r <= b)) return data[k];

    ll vl = this->query_sub(a, b, k * 2 + 1, l, (l + r) / 2);
    ll vr = this->query_sub(a, b, k * 2 + 2, (l + r) / 2, r);
    return _op(vl, vr);
};

// 区間クエリの処理
ll LazySegmentTree::query(int a, int b) { return query_sub(a, b, 0, 0, _size); }

// [a, b)の値をxで更新
void LazySegmentTree::update_sub(int a, int b, ll x, int k, int l, int r) {
    eval(k);
    if (a <= l && r <= b) {
        lazy[k] = x;
        eval(k);
    } else if (a < r && l < b) {
        update_sub(a, b, x, k * 2 + 1, l, (l + r) / 2);
        update_sub(a, b, x, k * 2 + 2, (l + r) / 2, r);
        data[k] = _op(data[k * 2 + 1], data[k * 2 + 2]);
    }
};

void LazySegmentTree::update(int a, int b, ll x) {
    update_sub(a, b, x, 0, 0, _size);
}

// 木を可視化する
void LazySegmentTree::show_tree() const {
    int idx = 1;
    int cnt = 0;
    for (int i = 0; i < 2 * _size - 1; i++) {
        std::cout << data[i];
        cnt++;
        if (cnt == idx) {
            std::cout << "\n";
            cnt = 0;
            idx *= 2;
        } else {
            std::cout << ' ';
        }
    }
}

int LazySegmentTree::size() const { return _size; }

ll sumll(ll a, ll b) { return a + b; }

vector<ll> sg2v(LazySegmentTree& sg, int N) {
    vector<ll> vec;
    for (int i = 0; i < N; i++) {
        vec.push_back(sg.query(i, i + 1));
    }
    return vec;
}

int main() {
    ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    ll N, A, B;
    cin >> N >> A >> B;
    vector<ll> X(N);
    for (int i = 0; i < N; i++) {
        cin >> X[i];
    }
    auto sg = LazySegmentTree(N, 0, sumll);
    for (int i = 0; i < N; i++) {
        sg.update(i, -1);
    }
    auto uf = UnionFind(N);
    vector<int> ld(N, 0);
    for (int i = 0; i < N; i++) {
        ll x = X[i];
        auto itb = upper_bound(X.begin(), X.end(), x + B);
        auto ita = lower_bound(X.begin(), X.end(), x + A);
        auto ib = distance(X.begin(), itb);
        auto ia = distance(X.begin(), ita);
        if (ia == ib) continue;
        int root;
        bool c1 = sg.query(i, i + 1) == -1;
        bool c2 = ia == N;
        bool c3 = sg.query(ia, ia + 1) == -1;
        if ((c1 && c2) || (c1 && c3)) {
            // 新たな島を作成
            uf.unite(i, ia);
            root = uf.root(i);
            ld[root] = i;
        } else if (c1) {
            // 既存の島に融合
            root = uf.root(ia);
        } else {
            // 既存の島に融合
            root = uf.root(i);
        }
        // std::cout << i << ' ' << root << "\n";
        // std::cout << c2 << ' ' << c3 << "\n";
        sg.update(i, ld[root]);
        if ((!c2) && c3) {
            // std::cout << ia << ' ' << ib << "\n";
            // std::cout << ld[root] << "\n";
            if (ib == ia + 1) {
                sg.update(ia, ld[root]);
            } else {
                sg.update(ia, ib, ld[root]);
            }
        }
        // print_vector(sg2v(sg, N));
    }
    vector<ll> C(N);
    for (int i = 0; i < N; i++) {
        C[i] = sg.query(i, i + 1);
    }
    // print_vector(C);
    vector<int> counts(N, 0);
    for (int i = 0; i < N; i++) {
        if (C[i] == -1) continue;
        counts[C[i]]++;
    }
    for (int i = 0; i < N; i++) {
        if (C[i] == -1) {
            std::cout << 1 << "\n";
        } else {
            std::cout << counts[C[i]] << "\n";
        }
    }
    // auto bit1 = BIT(N);
    // for (int i = 0; i < N; i++) {
    //     ll x = X[i];
    //     auto itb = lower_bound(X.begin(), X.end(), x - B);
    //     auto ita = upper_bound(X.begin(), X.end(), x - A);
    //     auto ib = distance(X.begin(), itb);
    //     auto ia = distance(X.begin(), ita);
    //     bit1.add(i, bit1.sum(ib, ia) + ia - ib);
    // }
    // auto bit2 = BIT(N);
    // for (int i = N - 1; i >= 0; i--) {
    //     ll x = X[i];
    //     auto itb = upper_bound(X.begin(), X.end(), x + B);
    //     auto ita = lower_bound(X.begin(), X.end(), x + A);
    //     auto ib = distance(X.begin(), itb);
    //     auto ia = distance(X.begin(), ita);
    //     bit2.add(i, bit2.sum(ia, ib) + ib - ia);
    // }
    // for (int i = 0; i < N; i++) {
    //     std::cout << bit1.sum(i, i + 1) + bit2.sum(i, i + 1) + 1 << "\n";
    // }
    return 0;
}

yukicoder

結果

テストケース

ソースコード