:heavy_check_mark: test/DataStructure/interval-set/yukicoder-2292.cpp

Depends on

Code

// competitive-verifier: PROBLEM https://yukicoder.me/problems/no/2292

#include "DataStructure/interval_set.hpp"
#include <bits/stdc++.h>

using namespace std;

int main()
{
    int N, Q;
    cin >> N >> Q;

    IntervalSet<int> interval_set;

    // 閉区間なので二倍して帳尻を合わせる
    for (int i = 0; i < Q; i++)
    {
        int t;
        cin >> t;
        if (t == 1)
        {
            int L, R;
            cin >> L >> R;
            interval_set.add_interval(2 * L, 2 * R + 1);
        }
        if (t == 2)
        {
            int L, R;
            cin >> L >> R;
            interval_set.erase_interval(2 * L + 1, 2 * R);
        }
        if (t == 3)
        {
            int u, v;
            cin >> u >> v;
            // cout << "!!!: " << t << " " << u << " " << v << endl;
            cout << ((u == v) || interval_set.same(2 * u, 2 * v)) << endl;
        }
        if (t == 4)
        {
            int u;
            cin >> u;
            int len = interval_set.length(2 * u) / 2 + 1;
            // cout << "!!!: " << t << " " << u << " " << len << endl;
            cout << len << endl;
        }

        // interval_set.print();
    }

    return 0;
}
#line 1 "test/DataStructure/interval-set/yukicoder-2292.cpp"
// competitive-verifier: PROBLEM https://yukicoder.me/problems/no/2292

#line 1 "DataStructure/interval_set.hpp"
#include <bits/stdc++.h>
using namespace std;

// contains
// find_intervalのpublic化
// 次の区間までの

template <typename T, typename Compare = std::less<T>>
    requires std::strict_weak_order<Compare, T, T>
class IntervalSet
{
private:
    // すべての区間長の合計
    T sum_length = 0;

    // 互いに交差しない半開区間 [first, second) の集合
    set<pair<T, T>> intervals;

    // 点 x を含む区間のイテレータ(なければ end)
    typename set<pair<T, T>>::const_iterator
    find_interval(const T &x) const
    {
        // first >= x となる最初の区間
        auto itr = intervals.lower_bound({x, x});

        // ひとつ左だけ、x を含みうる
        if (itr != intervals.begin())
        {
            auto prev_itr = prev(itr);
            if (prev_itr->first <= x && x < prev_itr->second)
                return prev_itr;
        }

        // ぴったり x から始まる区間
        if (itr != intervals.end() && itr->first <= x && x < itr->second)
            return itr;

        return intervals.end();
    }

public:
    IntervalSet() {}

    // [L, R) を追加し、必要なら隣接区間とマージする
    pair<T, T> add_interval(const T &L, const T &R)
    {
        return add_interval({L, R});
    }

    // [L, R) を追加し、必要なら隣接区間とマージする
    pair<T, T> add_interval(const pair<T, T> &LR)
    {
        assert(LR.first < LR.second);

        auto [l, r] = LR;

        // first >= l となる最初の区間
        auto itr = intervals.lower_bound(LR);

        // 左側に 1 つだけ重なりうる区間を見る
        if (itr != intervals.begin())
        {
            auto prev_itr = prev(itr);
            if (prev_itr->second >= l)
            {
                l = min(l, prev_itr->first);
                r = max(r, prev_itr->second);

                sum_length -= abs(prev_itr->second - prev_itr->first);
                itr = intervals.erase(prev_itr);
            }
        }

        // 先頭が r 以下の区間をすべてマージ
        while (itr != intervals.end() && itr->first <= r)
        {
            l = min(l, itr->first);
            r = max(r, itr->second);

            sum_length -= abs(itr->second - itr->first);
            itr = intervals.erase(itr);
        }

        intervals.insert({l, r});
        sum_length += abs(r - l);

        return {l, r};
    }

    // 区間集合から [L, R) を削除(集合としての差)
    void erase_interval(const T &L, const T &R)
    {
        erase_interval({L, R});
    }

    // 区間集合から [L, R) を削除(集合としての差)
    void erase_interval(const std::pair<T, T> &LR)
    {
        assert(LR.first < LR.second);

        const auto [l, r] = LR;
        if (intervals.empty())
            return;

        // first >= l となる最初の区間
        auto itr = intervals.lower_bound(LR);

        // 左側に 1 つだけ、[l, r) と交差しうる区間を見る
        if (itr != intervals.begin())
        {
            auto prev_itr = prev(itr);
            const auto [a, b] = *prev_itr;

            // [a, b) と [l, r) が交差
            if (b > l && a < r)
            {
                itr = intervals.erase(prev_itr);
                sum_length -= abs(b - a);

                if (a < l)
                {
                    sum_length += abs(l - a);
                    intervals.insert({a, l});
                }
                if (r < b)
                {
                    sum_length += abs(b - r);
                    intervals.insert({r, b});
                }
            }
        }

        // ここから先は first >= l が保証されている
        while (itr != intervals.end())
        {
            const auto [a, b] = *itr;

            if (a >= r)
                break; // これ以降は交差しない

            if (b <= r)
            {
                // 完全に [l, r) に含まれる区間
                itr = intervals.erase(itr);
                sum_length -= abs(b - a);
            }
            else
            {
                // a < r < b →右側 [r, b) だけ残す
                intervals.erase(itr);
                intervals.insert({r, b});
                sum_length -= abs(b - a);
                sum_length += abs(b - r);
                break;
            }
        }
    }

    // u と v が同じ区間に属するか
    bool same(T u, T v) const
    {
        if (intervals.empty())
            return false;

        if (v < u)
            swap(u, v);

        auto itr = find_interval(u);
        if (itr == intervals.end())
            return false;

        return v < itr->second;
    }

    // u が属する区間の長さ(なければ 0)
    T length(const T &u) const
    {
        auto itr = find_interval(u);
        if (itr == intervals.end())
            return T(0);
        return itr->second - itr->first;
    }

    // すべての区間長の合計
    T get_sum_lentgth() const
    {
        return sum_length;
    }

    void print() const
    {
        for (auto [l, r] : intervals)
        {
            cout << "[" << l << ", " << r << ") ";
        }
        cout << '\n';
    }
};
#line 5 "test/DataStructure/interval-set/yukicoder-2292.cpp"

using namespace std;

int main()
{
    int N, Q;
    cin >> N >> Q;

    IntervalSet<int> interval_set;

    // 閉区間なので二倍して帳尻を合わせる
    for (int i = 0; i < Q; i++)
    {
        int t;
        cin >> t;
        if (t == 1)
        {
            int L, R;
            cin >> L >> R;
            interval_set.add_interval(2 * L, 2 * R + 1);
        }
        if (t == 2)
        {
            int L, R;
            cin >> L >> R;
            interval_set.erase_interval(2 * L + 1, 2 * R);
        }
        if (t == 3)
        {
            int u, v;
            cin >> u >> v;
            // cout << "!!!: " << t << " " << u << " " << v << endl;
            cout << ((u == v) || interval_set.same(2 * u, 2 * v)) << endl;
        }
        if (t == 4)
        {
            int u;
            cin >> u;
            int len = interval_set.length(2 * u) / 2 + 1;
            // cout << "!!!: " << t << " " << u << " " << len << endl;
            cout << len << endl;
        }

        // interval_set.print();
    }

    return 0;
}

Test cases

Env Name Status Elapsed Memory
g++ 00_sample01.txt :heavy_check_mark: AC 4 ms 3 MB
g++ 00_sample02.txt :heavy_check_mark: AC 3 ms 3 MB
g++ 00_sample03.txt :heavy_check_mark: AC 3 ms 3 MB
g++ 00_sample04.txt :heavy_check_mark: AC 3 ms 3 MB
g++ 01_medium01.txt :heavy_check_mark: AC 156 ms 3 MB
g++ 01_medium02.txt :heavy_check_mark: AC 202 ms 3 MB
g++ 01_medium03.txt :heavy_check_mark: AC 196 ms 3 MB
g++ 01_medium04.txt :heavy_check_mark: AC 205 ms 3 MB
g++ 02_large01.txt :heavy_check_mark: AC 229 ms 3 MB
g++ 02_large02.txt :heavy_check_mark: AC 324 ms 3 MB
g++ 02_large03.txt :heavy_check_mark: AC 201 ms 3 MB
g++ 02_large04.txt :heavy_check_mark: AC 212 ms 3 MB
g++ 02_large05.txt :heavy_check_mark: AC 213 ms 3 MB
g++ 02_large06.txt :heavy_check_mark: AC 287 ms 3 MB
g++ 02_large07.txt :heavy_check_mark: AC 216 ms 3 MB
g++ 02_large08.txt :heavy_check_mark: AC 197 ms 3 MB
g++ 02_large09.txt :heavy_check_mark: AC 319 ms 3 MB
g++ 02_large10.txt :heavy_check_mark: AC 285 ms 3 MB
g++ 02_large11.txt :heavy_check_mark: AC 171 ms 13 MB
g++ 02_large12.txt :heavy_check_mark: AC 231 ms 13 MB
g++ 02_large13.txt :heavy_check_mark: AC 244 ms 13 MB
g++ 02_large14.txt :heavy_check_mark: AC 297 ms 9 MB
g++ 02_large15.txt :heavy_check_mark: AC 302 ms 9 MB
g++ 02_large16.txt :heavy_check_mark: AC 327 ms 9 MB
g++ 02_large17.txt :heavy_check_mark: AC 276 ms 9 MB
g++ 02_large18.txt :heavy_check_mark: AC 255 ms 9 MB
g++ 02_large19.txt :heavy_check_mark: AC 318 ms 9 MB
g++ 02_large20.txt :heavy_check_mark: AC 253 ms 9 MB
g++ 02_large21.txt :heavy_check_mark: AC 288 ms 9 MB
g++ 02_large22.txt :heavy_check_mark: AC 252 ms 9 MB
g++ 02_large23.txt :heavy_check_mark: AC 251 ms 9 MB
g++ 02_large24.txt :heavy_check_mark: AC 256 ms 9 MB
g++ 02_large25.txt :heavy_check_mark: AC 251 ms 9 MB
g++ 02_large26.txt :heavy_check_mark: AC 257 ms 9 MB
g++ 02_large27.txt :heavy_check_mark: AC 284 ms 9 MB
g++ 02_large28.txt :heavy_check_mark: AC 249 ms 9 MB
g++ 02_large29.txt :heavy_check_mark: AC 254 ms 9 MB
g++ 02_large30.txt :heavy_check_mark: AC 288 ms 9 MB
g++ 02_large31.txt :heavy_check_mark: AC 251 ms 9 MB
g++ 02_large32.txt :heavy_check_mark: AC 259 ms 9 MB
g++ 02_large33.txt :heavy_check_mark: AC 310 ms 9 MB
g++ 03_small01.txt :heavy_check_mark: AC 122 ms 3 MB
g++ 03_small02.txt :heavy_check_mark: AC 196 ms 3 MB
g++ 03_small03.txt :heavy_check_mark: AC 147 ms 3 MB
g++ 03_small04.txt :heavy_check_mark: AC 137 ms 3 MB
g++ 03_small05.txt :heavy_check_mark: AC 141 ms 3 MB
g++ 03_small06.txt :heavy_check_mark: AC 143 ms 3 MB
g++ 03_small07.txt :heavy_check_mark: AC 154 ms 3 MB
Back to top page