P3332 [ZJOI2013]K大数查询

https://www.luogu.com.cn/record/98238416

Luogu/P3332/P3332.cpp

@@ -0,0 +1,199 @@
+#include <iostream>
+#include <functional>
+#include <tuple>
+#include <vector>
+
+using std::cin;
+using std::cout;
+const char endl = '\n';
+
+template <typename T>
+class SegmentTree {
+  private:
+    struct node {
+        int l, r;
+        T sum, tag;
+        node *lchild, *rchild;
+
+        node(const int &_l = 0, const int &_r = 0)
+            : l(_l), r(_r), sum(0), tag(0), lchild(nullptr), rchild(nullptr) {}
+
+        ~node() {
+            if (lchild) delete lchild;
+            if (rchild) delete rchild;
+        }
+
+        void pushup() {
+            sum = 0;
+
+            if (lchild != nullptr) sum += lchild->sum;
+            if (rchild != nullptr) sum += rchild->sum;
+        }
+
+        void pushdown() {
+            int mid = l + r >> 1;
+
+            if (lchild == nullptr) lchild = new node(l, mid);
+            if (rchild == nullptr) rchild = new node(mid + 1, r);
+
+            lchild->sum += static_cast<T>(lchild->r - lchild->l + 1) * tag;
+            lchild->tag += tag;
+
+            rchild->sum += static_cast<T>(rchild->r - rchild->l + 1) * tag;
+            rchild->tag += tag;
+
+            tag = 0;
+        }
+    };
+
+    const int n;
+    node *root;
+
+    void modify(node *&cur, int l, int r, int ql, int qr, T val) {
+        if (cur == nullptr) cur = new node(l, r);
+
+        if (ql <= l && r <= qr) {
+            cur->sum += static_cast<T>(r - l + 1) * val;
+            cur->tag += val;
+
+            return;
+        }
+
+        int mid = l + r >> 1;
+
+        cur->pushdown();
+
+        if (ql <= mid) modify(cur->lchild, l, mid, ql, qr, val);
+        if (qr > mid) modify(cur->rchild, mid + 1, r, ql, qr, val);
+
+        cur->pushup();
+    }
+
+    T query(node *cur, int l, int r, int ql, int qr) {
+        if (cur == nullptr) return T();
+        if (ql <= l && r <= qr) return cur->sum;
+
+        int mid = l + r >> 1;
+        T res = 0;
+
+        cur->pushdown();
+
+        if (ql <= mid) res += query(cur->lchild, l, mid, ql, qr);
+        if (qr > mid) res += query(cur->rchild, mid + 1, r, ql, qr);
+
+        return res;
+    }
+
+  public:
+    SegmentTree(const int &_n)
+        : n(_n), root(nullptr) {}
+
+    ~SegmentTree() {
+        if (root) delete root;
+    }
+
+    void modify(int ql, int qr, T val) {
+        modify(root, 1, n, ql, qr, val);
+    }
+
+    T query(int ql, int qr) {
+        return query(root, 1, n, ql, qr);
+    }
+};
+
+int main() {
+    std::ios::sync_with_stdio(false);
+    cin.tie(nullptr);
+
+    int n, m, cnt = 0;
+    std::vector<std::tuple<int, int, int, long long, int>> qs;
+
+    cin >> n >> m;
+
+    for (int i = 1; i <= m; i++) {
+        int op, l, r;
+        long long k;
+
+        cin >> op >> l >> r >> k;
+
+        if (op == 1) {
+            qs.emplace_back(op, l, r, k, -1);
+        } else {  // op == 2
+            qs.emplace_back(op, l, r, k, cnt++);
+        }
+    }
+
+    std::vector<int> ans(cnt);
+    SegmentTree<long long> tr(n);
+
+    std::function<void(const std::vector<std::tuple<int, int, int, long long, int>> &, int, int)> solve = [&](const std::vector<std::tuple<int, int, int, long long, int>> &qs, int l, int r) -> void {
+        if (qs.empty()) return;
+
+        if (l == r) {
+            for (auto o : qs) {
+                int op, id;
+
+                std::tie(op, std::ignore, std::ignore, std::ignore, id) = o;
+
+                if (op == 2) ans[id] = l;
+            }
+
+            return;
+        }
+
+        int mid = l + r >> 1;
+        std::vector<std::tuple<int, int, int, long long, int>> ql, qr;
+        std::vector<long long> cur(qs.size());
+
+        for (int i = 0; i < qs.size(); i++) {
+            int op, l, r;
+            long long k;
+
+            std::tie(op, l, r, k, std::ignore) = qs[i];
+
+            if (op == 1) {
+                if (k > mid) tr.modify(l, r, 1);
+            } else {  // op == 2
+                cur[i] = tr.query(l, r);
+            }
+        }
+
+        for (int i = 0; i < qs.size(); i++) {
+            int op, l, r;
+            long long k;
+
+            std::tie(op, l, r, k, std::ignore) = qs[i];
+
+            if (op == 1 && k > mid) {
+                tr.modify(l, r, -1);
+            }
+        }
+
+        for (int i = 0; i < qs.size(); i++) {
+            int op, l, r, id;
+            long long k;
+
+            std::tie(op, l, r, k, id) = qs[i];
+
+            if (op == 1) {
+                if (k <= mid) ql.emplace_back(qs[i]);
+                else qr.emplace_back(qs[i]);
+            } else {  // op == 2
+                if (cur[i] < k) {
+                    ql.emplace_back(op, l, r, k - cur[i], id);
+                } else {
+                    qr.emplace_back(qs[i]);
+                }
+            }
+        }
+
+        solve(ql, l, mid);
+        solve(qr, mid + 1, r);
+    };
+
+    solve(qs, 0, n);
+
+    for (int x : ans) cout << x << endl;
+
+    return 0;
+}