关于 risrqnis

这道题里最有用的( 114514 1919810

Range Insert Subset
Range Query [n?] In Set

破案了 我那五个点是因为维护不知道有什么用的东西炸了
删了就过了

题面

[JRKSJ R4] risrqnis

给你一个长度为 \(n\) 的序列 \(a\),有 \(q\) 次操作,初始有 \(m\) 个空集 \(S\),共有两种操作,如下:

1 l r k,将 \(l\sim r\) 加入集合 \(S_k\),即 \(S_k\gets S_k\cup\{x|l\le x\le r\}\)
2 l r k,查询对于所有 \(l\le i\le r\)\(a_i\) 中有多少个在集合 \(S_k\) 中,即查询 \(\displaystyle\sum_{i=l}^r[a_i\in S_k]\)

\(\text{Subtask}\) \(n,q\le\) \(m\le\) 分值 \(\text{Time Limit}\)
\(1\) \(10^6\) \(1\) \(30\) \(1500\text{ ms}\)
\(2\) \(5\times 10^3\) \(3\times 10^5\) \(15\) \(1000\text{ ms}\)
\(3\) \(10^5\) \(10^5\) \(15\) \(1500\text{ ms}\)
\(4\) \(3\times 10^5\) \(10^9\) \(40\) \(3000\text{ ms}\)

对于 \(100\%\) 的数据,\(1\le n,q\le 10^6\)\(1\le m\le 10^9\)\(1\le a_i\le 10^9\)

操作 \(1\)\(1\le l\le r\le 10^9\),操作 \(2\)\(1\le l\le r\le n\)。所有操作中 \(1\le k\le m\)

没有一个 \(\text{Subtask}\) 取到所有数据范围的最大值,各个 \(\text{Subtask}\) 都有自己的限制。请阅读数据范围表。

我把所有对做题有帮助的部分都粘过来了。最有帮助的在最开头。第二有帮助的是“本题输入文件较大”后面那段。
然后开始解题。

Solution

观察操作性质可以发现每个集合的操作都是独立的。因此可以将各集合的操作分别进行。\(m \le 10^9\) 就是唬你的

首先讨论当 \(m = 1\) 时的解法。由于这时只有一个集合,而每个 \(a_i\) 只需要被加入一次集合,因此可以暴力进行。
沿用这篇闲话里对元素标记的方式,我们使用并查集维护“加入第 \(i\) 个元素时应当加入哪个元素”,跳跃的复杂度均摊是 \(O(n \ \alpha(n))\) 的。加入元素直接使用树状数组在下标位置插入一个 \(1\),查询时前缀和差分即可。这部分的总时间复杂度是 \(O(n\log n)\)的。

然后 \(m \neq 1\)。这时如果直接沿用上面的方式是很不优的。考虑根号分治。我们分别讨论操作数 \(\le \sqrt q\)\(> \sqrt q\) 的两种集合。

1. \(> \sqrt q\)

此类集合的数量最多是 \(\sqrt q\) 个,一个元素总共加入次数是 \(O(\sqrt q)\),因此可以考虑沿用如上思路。但是我们发现加入一个元素的总复杂度是 \(O(\sqrt q \log n\ \alpha(n))\) 的,但查询的总复杂度是 \(O(\log n)\) 的。这样很不平衡。因此考虑有限制根号平衡,即使用加入复杂度 \(O(1)\) 查询复杂度 \(O(\sqrt n)\) 的值域分块代替树状数组,因此有这部分的复杂度 \(O(n \ \alpha(n)\sqrt q + n \sqrt n)\)
看题解说这个 \(\alpha\) 似乎可以通过把并查集换成 ODT 得到严格复杂度。好像确实 因为这样在这部分的总花销是 \(O(n \log n)\) 的,然后摊掉就是对的了。我反正写不出来,wa四个点。正好是我忘记维护加链表时对的那几个点。
我猜测这题数据是卡着分治的边出的,每个点对应一种分治方法。

2. \(\le \sqrt q\)

这时候暴力加入元素寄了。
但我们可以考虑特殊性质。容易发现每个集合对应的值域连续段数 \(\le \sqrt q\)
考虑使用一个颜色段均摊结构维护值域连续段,查询时就看 \(a\) 在这段查询区间内有多少个数落在当前集合的值域段内就行。这时为了维护查询,我们再考虑一个根号平衡。插入顶多进行 \(O(n)\) 次,但是查询次数是 \(O(n\sqrt q)\) 的。因此考虑使用一个加入复杂度 \(O(\sqrt n)\) 查询复杂度 \(O(1)\) 的前缀和值域分块代替两行前你脑子里出现的什么数据结构。
我们发现这么写保存区间需要 \(O(n \sqrt q)\) 的空间,算一下发现炸的很死。考虑一个区间离线,我们把最终区间用一个前向星啥的链式数据结构存起来,到最后只需要用前向星做一下扫描线就行。
这部分的复杂度是 \(O(n \sqrt n + n \sqrt q)\)。线性空间。

你把这几部分的做法拼吧拼吧就能得到200+行 然后调起来费死劲

然后我发现这题玄学调一下块长能很快
我猜测 \(\color{black}{\text{L}}\color{red}{\text{etItDown}}\) 学长是“稍微”进行了一波块长的优化
反正我是不打算卡

总结:是一道比较优秀的根号分治题目,综合运用了各式根号分治与平衡思想。

但其实想写这题的初心还是上琴糖。

code
#include <bits/stdc++.h>
using namespace std;
inline void get(int & x){ /* 快读 */ } 
#define rep(i,a,b) for (register int (i) = (a); (i) <= (b); ++(i))
#define pre(i,a,b) for (register int (i) = (a); (i) >= (b); --(i))

const int N = 1e6 + 10;
int n, q, m, sqn, sqq, sn, lsh[N], ans[N];
int lp[N], rp[N], bl[N], opr[N], a[N];
bool cmp(const int & a, const int & b) { 
    if (lsh[a] != lsh[b]) return lsh[a] < lsh[b];
    return a < b;
}

namespace Subtask1 {
    int typ, l, r, tmp, fa[N];
    int find(int u) { return u == fa[u] ? u : fa[u] = find(fa[u]); }

    int Index[N];
    void add(int p) { for (; p <= n; p += p & -p) ++ Index[p]; }
    int query(int p) {
        int ret = 0;
        for (; p; p ^= p & -p) ret += Index[p];
        return ret;
    }

    void main() {
        rep(i,1,n+1) fa[i] = i;
        while (q--) {
            get(typ, l, r, tmp);
            if (typ == 1) {
                l = lower_bound(lsh+1, lsh+1+n, l) - lsh;
                r = upper_bound(lsh+1, lsh+1+n, r) - lsh - 1;
                for (int i = find(l); i <= r; i = find(i + 1)) add(a[i]), fa[i] = fa[i + 1];
            } else cout << query(r) - query(l - 1) << '\n';
        }
    }
}

struct queries {
    int typ, l, r, k, id;
    bool operator < (const queries & b) const {
        if (k != b.k) return k < b.k;
        return id < b.id;
    }
} qry[N];

namespace Les {
    int head[N], mlc;
    struct ep {
        int next, l, r, cont;
    } e[N<<2];
    void adde(int x, int l, int r, int cont) {
        e[++mlc] = { head[x], l, r, cont };
        head[x] = mlc;
    }

    struct node {
        int l, r, st;
        node (int _l, int _r, int _s) : l(_l), r(_r), st(_s) {}
        bool operator < (const node & b) const { return l < b.l; }
    } ; set < node > st;
    template<typename iter> iter erase(iter itr, int t) {
        adde(itr->r, itr->st, t, 1);
        adde(itr->l-1, itr->st, t, -1);
        return st.erase(itr);
    }

    int cntblk[1000], cntpts[N];
    void add(int p) {
        int bel = bl[p];
        rep(i, p, rp[bel]) cntpts[i]++;
        rep(i, bel+1, sn) cntblk[i]++;
    }
    int query(int x) { return cntpts[x] + cntblk[bl[x]]; }

    void solve(int L, int R) {
        st.clear();
        rep(i,L,R) {
            if (qry[i].typ == 2) continue;
            int l = lower_bound(lsh+1, lsh+1+n, qry[i].l) - lsh;
            int r = upper_bound(lsh+1, lsh+1+n, qry[i].r) - lsh - 1;
            if (l > r) continue;
            if (!st.empty()) {
                auto itl = st.lower_bound({l, 0, 0}), itr = st.upper_bound({r, 0, 0});
                if (itl != st.begin() and prev(itl)->r >= l) -- itl;
                if (itr != st.begin() and (--itr)->r >= l) {
                    l = min(l, itl->l), r = max(r, itr->r);
                    while(itl != itr) itl = erase(itl, i);
                    erase(itr, i);
                } 
            } st.emplace(l, r, i);
        } for (auto x : st) adde(x.r, x.st, R, 1), adde(x.l - 1, x.st, R, -1); 
    }

    void getAns() {
        rep(u,1,n) {
            add(a[u]);
            for (int i = head[u]; i; i = e[i].next) {
                rep(j,e[i].l,e[i].r) if (qry[j].typ == 2) 
                    ans[qry[j].id] += e[i].cont * (query(qry[j].r) - query(qry[j].l - 1));
            }
        }
    }
}

namespace Gre {
    int fa[N];
    int find(int u) { return u == fa[u] ? u : fa[u] = find(fa[u]); }

    int cntblk[1000], cntpts[N];
    void add(int p) { cntpts[p]++; cntblk[bl[p]]++; }
    int query(int l, int r) {
        int bl_l = bl[l], bl_r = bl[r], ret = 0;
        if (bl_l == bl_r) rep(i, l, r) ret += cntpts[i];
        else {
            rep(i, l, rp[bl_l]) ret += cntpts[i];
            rep(i, lp[bl_r], r) ret += cntpts[i];
            rep(i, bl_l+1, bl_r-1) ret += cntblk[i];
        } return ret;
    }

    void solve(int L, int R) {
        rep(i,1,n+1) fa[i] = i;
        rep(i,1,sn) cntblk[i] = 0;
        rep(i,1,n) cntpts[i] = 0;
        rep(i,L,R) {
            if (qry[i].typ == 1) {
                int l = lower_bound(lsh+1, lsh+1+n, qry[i].l) - lsh;
                int r = upper_bound(lsh+1, lsh+1+n, qry[i].r) - lsh - 1;
                for (int i = find(l); i <= r; i = find(i + 1)) add(a[i]), fa[i] = fa[i + 1];
            } else ans[qry[i].id] = query(qry[i].l, qry[i].r);
        }
    }
}

namespace SubtaskElse {
    void main() {
        rep(i,1,sn) lp[i] = (i-1) * sqn + 1, rp[i] = min(n, i * sqn);
        rep(i,1,n) bl[i] = (i-1) / sqn + 1;
        rep(i,1,q) get(qry[i].typ, qry[i].l, qry[i].r, qry[i].k), opr[i] = qry[i].typ, qry[i].id = i;
        sort(qry+1, qry+1+q);
        for (int l = 1, r; l <= q; l = r + 1) {
            r = l;
            while (qry[l].k == qry[r+1].k) r++;
            if (r - l + 1 <= sqq) Les :: solve(l, r);
            else Gre :: solve(l, r);
        } Les :: getAns();
        rep(i,1,q) if(opr[i] == 2) cout << ans[i] << '\n';
    }
}

signed main() {
    ios::sync_with_stdio(false); cout.tie(0);
    get(n, q, m);
    sqn = pow(n, 0.6); sqq = pow(q, 0.6); sn = (n + sqn - 1) / sqn;
    rep(i,1,n) get(lsh[i]), a[i] = i;
    sort(a+1, a+1+n, cmp); sort(lsh+1, lsh+1+n);
    if (m == 1) Subtask1 :: main();
    else SubtaskElse :: main();
    return 0;
}
posted @ 2022-09-28 09:29  joke3579  阅读(192)  评论(2编辑  收藏  举报