颜色段均摊

区间覆盖,维护颜色信息。

直接维护每一个点的颜色,和维护颜色段有不同的效果。后者显然对信息处理效率更高。

考虑一个区间覆盖的影响。
image

最多加入一个区间,删掉若干个区间。

如果一开始有 \(O(n)\) 段颜色,有 \(O(q)\) 段区间覆盖,那么总时间复杂度不超过 \(O((n + q) \log)\)。这个 \(\log\) 是 set 里面找的时间。

joisc 2023 Day 3 C

【题意】

给定一个 \(n\) 个点的树,有一个 \(m\) 个数的序列 \(a\)\(q\) 次询问,每次求从任意一个点开始走一条树上路径经过 \(a_l \sim a_r\) 中所有点的话,路径上经过至少一次的点数最少是多少。

【分析】

容易发现这题是求区间上点的虚树大小。

方法 1:
考虑莫队。
考虑加入某一个数。
如果直接加入并维护虚树,需要找到虚树上其最近祖先,二分线段树可以做到 \(\log^2\),但是显然过不去。
考虑虚树性质,可以 dfn 排序然后计算相邻两两距离之和。
如果使用不删除莫队,加入 \(\log\),用 set 维护,但是注意到这个题只需要查询 kth 的 rank,可以使用权值树状数组代替 set,具体这么做:

  • 查询 rank:直接查询前缀和即可。
  • 查询 kth:考虑树状数组上二分。

树状数组一种二分方式是直接模拟线段树二分(不同之处在于一个能求区间 kth,一个能求全局 kth)
image
树状数组(记为 \(a\))长这样,我们考虑初始令 \(now = n, digit = \log_2 n - 1, k = k\)。这个 \(digit\) 的意思就是在哪一位。比如 \(digit = 2\) 就是在第三层(\(4\) 那一层)
如果 \(a_{now - 2^{digit}} \ge k\) 那么向左边走(\(now -= 2^{digit}\));否则向右边走(\(k -= a_{now - 2^{digit}}\))。不管往哪边走,\(digit--\)
注意 \(n\) 要补到 \(2^n\),这个空间上要注意一下,但是由于后面的都是 \(0\),是不需要做其他事情的。

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define f(i, a, b) for(int i = (a); i <= (b); i++)
#define cl(i, n) i.clear(),i.resize(n);
#define endl '\n'
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
const int inf = 1e9;
//#define cerr if(false)cerr
//#define freopen if(false)freopen
#define watch(x) cerr  << (#x) << ' '<<'i'<<'s'<<' ' << x << endl
void pofe(int number, int bitnum) {
    string s; f(i, 0, bitnum) {s += char(number & 1) + '0'; number >>= 1; } 
    reverse(s.begin(), s.end()); cerr << s << endl; 
    return;
}
template <typename TYP> void cmax(TYP &x, TYP y) {if(x < y) x = y;}
template <typename TYP> void cmin(TYP &x, TYP y) {if(x > y) x = y;}
//调不出来给我对拍!
//use std::array.
int n;
struct szsz {
    int a[100010];
    int lowbit(int x) {return x & -x;}
    void add(int x, int k) {
        while(x <= n) {
            a[x] += k;
            x += lowbit(x);
        }
    }
    int rnk(int x) {
        int res = 0;
        while(x > 0) {
            res += a[x];
            x -= lowbit(x);
        }
        return res;
    }
    int kth(int k) {
        int newn = 1 << ((int)log2(n - 1) + 1);
        int now = newn, dig = log2(newn) - 1; 
    	while(dig >= 0)  {
    		if(a[now - (1 << dig)] >= k) {
    			now -= (1 << dig);
    		}
    		else {
    			k -= a[now - (1 << dig)];
    		}
    		dig--;
    	}
    	return now;
    }
}tr;
signed main() {
    ios::sync_with_stdio(0);
    cin.tie(NULL);
    cout.tie(NULL);
    //freopen();
    //freopen();
    //time_t start = clock();
    //think twice,code once.
    //think once,debug forever.
    cin >> n; int q; cin >> q;
    f(i, 1, q) {
        int op; cin >> op;
        if(op == 1){int x, k; cin >> x >> k; tr.add(x, k);}
        else if(op == 2) {int x; cin >> x; cout << tr.rnk(x) << endl;}
        else {int x; cin >> x; cout << tr.kth(x) << endl;}
    }
    //time_t finish = clock();
    //cout << "time used:" << (finish-start) * 1.0 / CLOCKS_PER_SEC <<"s"<< endl;
    return 0;
}
/*
2023/x/xx
start thinking at h:mm


start coding at h:mm
finish debugging at h:mm
*/

上面的这份代码,在点有权值的情况下,rnk 出来的是最大 rnk,kth 就是 kth。

于是我们成功把 set 改成了常数十分小的权值树状数组。这样 \(O(n \sqrt n \log n)\) 就可以在 4s 内卡过啦!(确信)

方法 2:

我们考虑把不删除改成不加入。(回滚莫队不仅能做不删除,也可以做不加入,具体方法就考虑某一块先计算 \(head \sim n\) 的答案,然后移动两边的指针)

这样做有什么好处呢?我们删除的时候直接用链表删除即可。(之前小瞧了这个数据结构,但是确实很有用)这样不用维护一棵平衡树,是 \(O(1)\) 的。考虑每一块插入过程。可以对 \(O(n)\) 长度的数组直接排序。暴力插入还是带一个 \(\log\),但是我们可以从右往左进行。这样做,这部分复杂度直接变成 \(O(n \log n)\),那么整体复杂度就是纯根号的。

方法 3:

考虑虚树的另一种形式:虚树根到原树根的一条链加上原树根到点集中所有点的并。

首先链是好求的,就一个区间 lca,这里不说了。重点是并怎么算。注意到这里就转化为了到根的链上问题了。如果不知道接下来干什么,接着想。

考虑扫描线的第二种形式,扫 \(r\),维护所有 \(l\) 的答案。考虑二维数点那个做法,维护最后一次出现位置。可以启发到这题上。注意到一个 \(r\) 是给 \(r\)\(1\) 的链上这些点全部加入一次虚树,那么我们考虑维护时间戳,也就是每一个点是什么时候加入的虚树。

发现这个全部加入类似颜色段均摊处理的东西,考虑树剖之后是 \(\log\) 段连续段,这意味着每次扫描线右端点动了的时候只会插入 \(\log\) 段区间,所以这部分的时间复杂度是 \(O(\log^2 )\) 的。

考虑查询怎么做。额外维护一个权值树状数组拿来查询 rank。只需要实时维护颜色为某一个数的区间总大小即可。查询部分的时间复杂度是 \(O(\log)\) 的。

分几步:

  1. 树剖预处理。
  2. 扫描线,预先把所有区间打到 \(r\) 上。每次 \(r\) 动的时候,沿着 \(1 \sim c_r\) 的链走,更新颜色段,并在树状数组里更新。
  3. 处理查询,在树状数组里查询即可。树状数组支持 rank。

总时间复杂度 \(O(n + n \log^2 n + q \log n)\)

posted @ 2023-03-22 22:08  OIer某罗  阅读(220)  评论(0编辑  收藏  举报