結果

問題 No.3265 地元に帰れば天才扱い!
ユーザー 横山緑
提出日時 2025-09-06 14:52:56
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,211 ms / 2,500 ms
コード長 6,421 bytes
コンパイル時間 3,959 ms
コンパイル使用メモリ 282,076 KB
実行使用メモリ 16,512 KB
最終ジャッジ日時 2025-09-06 14:53:31
合計ジャッジ時間 32,770 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #

#line 2 "library/template.hpp"
// #pragma GCC target("avx2")
// #pragma GCC optimize("O3")
// #pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
#ifdef LOCAL
#include <debug.hpp>
#else
template <class T>
concept Streamable = requires(ostream os, T &x) { os << x; };
template <class mint>
concept is_modint = requires(mint &x) {
    { x.val() } -> std::convertible_to<int>;
};
#define debug(...)
#endif
template <Streamable T> void print_one(const T &value) { cout << value; }
template <is_modint T> void print_one(const T &value) { cout << value.val(); }
void print() { cout << '\n'; }
template <class T, class... Ts> void print(const T &a, const Ts &...b) {
    print_one(a);
    ((cout << ' ', print_one(b)), ...);
    cout << '\n';
}
template <ranges::range Iterable>
    requires(!Streamable<Iterable>)
void print(const Iterable &v) {
    for(auto it = v.begin(); it != v.end(); ++it) {
        if(it != v.begin())
            cout << " ";
        print_one(*it);
    }
    cout << '\n';
}
using vi = vector<int>;
using vii = vector<vector<int>>;
using pii = pair<int, int>;
using ll = long long;
using vl = vector<ll>;
using vll = vector<vl>;
using pll = pair<ll, ll>;
#define all(v) begin(v), end(v)
template <class T> void UNIQUE(T &v) {
    ranges::sort(v);
    v.erase(unique(all(v)), end(v));
}
template <typename T> inline bool chmax(T &a, T b) {
    return ((a < b) ? (a = b, true) : (false));
}
template <typename T> inline bool chmin(T &a, T b) {
    return ((a > b) ? (a = b, true) : (false));
}
// https://trap.jp/post/1224/
template <class... T> constexpr auto min(T... a) {
    return min(initializer_list<common_type_t<T...>>{a...});
}
template <class... T> constexpr auto max(T... a) {
    return max(initializer_list<common_type_t<T...>>{a...});
}
template <class... T> void input(T &...a) { (cin >> ... >> a); }
template <class T> void input(vector<T> &a) {
    for(T &x : a)
        cin >> x;
}
#define INT(...)                                                               \
    int __VA_ARGS__;                                                           \
    input(__VA_ARGS__)
#define LL(...)                                                                \
    long long __VA_ARGS__;                                                     \
    input(__VA_ARGS__)
#define STR(...)                                                               \
    string __VA_ARGS__;                                                        \
    input(__VA_ARGS__)
#define REP1_0(n, c) REP1_1(n, c)
#define REP1_1(n, c)                                                           \
    for(ll REP_COUNTER_##c = 0; REP_COUNTER_##c < (ll)(n); REP_COUNTER_##c++)
#define REP1(n) REP1_0(n, __COUNTER__)
#define REP2(i, a) for(ll i = 0; i < (ll)(a); i++)
#define REP3(i, a, b) for(ll i = (ll)(a); i < (ll)(b); i++)
#define REP4(i, a, b, c) for(ll i = (ll)(a); i < (ll)(b); i += (ll)(c))
#define overload4(a, b, c, d, e, ...) e
#define rep(...) overload4(__VA_ARGS__, REP4, REP3, REP2, REP1)(__VA_ARGS__)
ll inf = 3e18;
vl dx = {1, -1, 0, 0};
vl dy = {0, 0, 1, -1};
template <class T> constexpr T floor(T x, T y) noexcept {
    return x / y - ((x ^ y) < 0 and x % y);
}
template <class T> constexpr T ceil(T x, T y) noexcept {
    return x / y + ((x ^ y) >= 0 and x % y);
}
// yの符号に関わらず非負で定義 \bmod:texコマンド
template <class T> constexpr T bmod(T x, T y) noexcept {
    T m = x % y;
    return (m < 0) ? m + (y > 0 ? y : -y) : m;
}
template <std::signed_integral T> constexpr int bit_width(T x) noexcept {
    return std::bit_width((uint64_t)x);
}
template <std::signed_integral T> constexpr int popcount(T x) noexcept {
    return std::popcount((uint64_t)x);
}
constexpr bool kth_bit(auto n, auto k) { return (n >> k) & 1; }
#line 3 "library/data-structure/segtree.hpp"

template <class S, auto op, auto e> struct segtree {
    int n;
    vector<S> v;
    explicit segtree(int n) : n(n), v(2 * n, e()) {}
    explicit segtree(const vector<S> &v_) : n(v_.size()), v(2 * n, e()) {
        for(int i = 0; i < n; ++i)
            v[n + i] = v_[i];
        for(int i = n - 1; i >= 0; --i)
            v[i] = op(v[i << 1], v[i << 1 | 1]);
    }
    void set(int x, S p) {
        assert(0 <= x && x < n);
        x += n;
        v[x] = p;
        while(x > 1) {
            x >>= 1;
            v[x] = op(v[x << 1], v[x << 1 | 1]);
        }
    }
    S prod(int l, int r) const {
        assert(0 <= l && l <= r && r <= n);
        S pl(e()), pr(e());
        l += n, r += n;
        while(l < r) {
            if(l & 1) {
                pl = op(pl, v[l]);
            }
            if(r & 1) {
                pr = op(v[r - 1], pr);
            }
            l = (l + 1) >> 1;
            r >>= 1;
        }
        return op(pl, pr);
    }
    S get(int x) const { return v[n + x]; }
};
#line 3 "test.cpp"
ll op(ll a, ll b) { return a + b; }
ll e() { return 0; }
int main() {
    INT(n, m);
    segtree<ll, op, e> seg(m), cnt(m + 1);
    vl a(m), l(m), r(m), gen(m, 0);
    vi iwai(n);
    iota(all(iwai), 0);
    ll ans{};
    rep(i, n) {
        LL(ai, li, ri);
        a[i] = ai, l[i] = --li, r[i] = ri;
        seg.set(i, a[i]);
        cnt.set(li, cnt.get(li) + 1);
        cnt.set(ri, cnt.get(ri) - 1);
    }
    rep(i, n) {
        gen[i] = (r[i] - l[i]) * a[i] - seg.prod(l[i], r[i]);
        ans += gen[i];
    }
    debug(ans);
    // rep(i, m) { debug(i, cnt.prod(0, i + 1)); }
    INT(q);
    rep(q) {
        LL(ix, y, u, v);
        --ix, --y, --u;
        ll x = iwai[ix];
        iwai[ix] = y;

        ans -= a[x] * (r[x] - l[x]);
        ans += seg.prod(l[x], r[x]);
        // debug(ans);
        cnt.set(l[x], cnt.get(l[x]) - 1);
        cnt.set(r[x], cnt.get(r[x]) + 1);
        ans += a[x] * cnt.prod(0, x + 1);
        // debug(a[x] * (r[x] - l[x]), seg.prod(l[x], r[x]),
        //   a[x] * cnt.prod(0, x + 1));
        // debug(x, a[x], l[x], r[x], ans);
        seg.set(x, seg.get(x) - a[x]);

        // gen[x] = 0;
        a[y] = a[x];
        l[y] = u, r[y] = v;
        seg.set(y, seg.get(y) + a[y]);
        ans -= a[y] * cnt.prod(0, y + 1);
        // debug(a[y] * cnt.prod(0, y + 1));
        cnt.set(l[y], cnt.get(l[y]) + 1);
        cnt.set(r[y], cnt.get(r[y]) - 1);
        ans -= seg.prod(l[y], r[y]);
        ans += a[y] * (r[y] - l[y]);
        // rep(i, m) { debug(i, cnt.prod(0, i + 1)); }
        print(ans);
    }
}
0