[ABC470G] ΣШX 题解

[ABC470G] ΣШX 题解

题意

给定一个长度为 $N$ 的序列 $A$（$0\le A_i\le N$），求所有区间的 $\mathrm{mex}$（未出现的最小非负整数）。

思路

虽然本题非常典，但由于我没做过所以记录一下。

考虑对于一个右端点 $r$，有多少个 $l$ 满足 $\mathrm{mex}(l,r)=k$。设 $po{s}_k$ 表示当前 $k$ 最后一次出现的位置（没有则为 $0$），则有 $l\ge {\max }_{\begin{array}{c}0\le i\le k-1\end{array}}po{s}_k$，即 $pos$ 中 $k-1$ 项的前缀最小值。那么对于每一个 $k$，满足 $\mathrm{mex}(l,r)\ge k+1$ 的 $l$ 的数量即为 $pos$ 中 $k$ 的前缀最小值。

由此可得，当前所有 $k$ 的前缀最小值的和即为当前右端点的贡献。

模拟一下：

$A=(2,1,0,3)$。

初始：

$r=1$ 时：

$an{s}_1=0+0+0+0=0$。

$r=2$ 时：

$an{s}_2=0+0+0+0=0$。

$r=3$ 时：

$an{s}_3=3+2+1+0=6$。

$r=4$ 时：

$an{s}_4=3+2+1+1=7$。

最终答案即为 $ans=0+0+6+7=13$。

如果暴力维护 $premin$ 显然是 $O(n^2)$ 的，需要优化。

考虑线段树维护区间前缀最大值和，难点在于如何合并。

想要将左区间和右区间合并，首先左区间的贡献是不变的，考虑右区间的贡献。

图中红色部分是有用的贡献，蓝色是无用的贡献，则右区间的贡献相当于在前面加一个左区间最小值时的答案。

设 $findc(p,v)$ 为当前在线段树上的 $p$ 节点，表示的区间为 $[l_p,r_p]$，前面有一个数 $v$ 时的贡献。设 $lc(p),rc(p)$ 是 $p$ 的左右节点，$m{n}_p,va{l}_p$ 分别表示节点所表示的区间的最小值以及答案。

当 $m{n}_{lc(p)}<v$ 时说明 $v$ 可能会对 $va{l}_{lc(p)}$ 产生影响，但不会对 $va{l}_{rc(p)}$ 产生影响（注：此处的 $va{l}_{rc(p)}$ 是合并后的，已经算了 $lc(p)$ 的影响，在代码中需要写成 $va{l}_p-va{l}_{lc(p)}$），即 $findc(p,v)\leftarrow findc(lc(p),c)+va{l}_{rc(p)}$。

否则 $v$ 会将 $va{l}_{lc(p)}$ 覆盖为 $(r_{lc(p)}-l_{lc(p)}+1)\times v$，会对右区间产生影响，所以 $findc(p,v)\leftarrow (r_{lc(p)}-l_{lc(p)}+1)\times v+findc(rc(p),c)$。

最后特判一下叶子节点即可。

$$findc(p,v)=\{\begin{array}{ll}\min (va{l}_p,v)& pisaleafnode\\ findc(lc(p),c)+va{l}_{rc(p)}& m{n}_{lc(p)}<v\\ (r_{lc(p)}-l_{lc(p)}+1)\times v+findc(rc(p),c)& other\end{array}$$

$findc$ 的时间复杂度为 $O(\log n)$，即一次 $pushup$ 的时间复杂度为 $O(\log n)$，修改一个值的时间复杂度为 $O({\log }^{2}n)$，一共 $N$ 个元素，总时间复杂度为 $O(n{\log }^{2}n)$。

代码

#include<bits/stdc++.h>
#define lc(p) ((p)<<1)
#define rc(p) ((p)<<1|1) 
using namespace std;
typedef long long ll;
int n,a[300010];
struct N{
	ll mn,c;
}tr[1200010];
ll findc(int p,int l,int r,ll v){
	if(l==r)return min(tr[p].mn,v); 
	int mid=(l+r)>>1;
	if(tr[lc(p)].mn<v)return findc(lc(p),l,mid,v)+tr[p].c-tr[lc(p)].c;
	else return (mid-l+1)*v+findc(rc(p),mid+1,r,v);
}
void pushup(int p,int l,int r){
	int mid=(l+r)>>1;
	tr[p].c=tr[lc(p)].c+findc(rc(p),mid+1,r,tr[lc(p)].mn);
	tr[p].mn=min(tr[lc(p)].mn,tr[rc(p)].mn);
}
void change(int p,int l,int r,int x,int v){
	if(l==r){
		tr[p].mn=tr[p].c=v;
		return ;
	}
	int mid=(l+r)>>1;
	if(x<=mid)change(lc(p),l,mid,x,v);
	else change(rc(p),mid+1,r,x,v);
	pushup(p,l,r);
}
int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin>>n;
	ll ans=0;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		change(1,0,n,a[i],i);
		ans+=tr[1].c;
	}
	cout<<ans;
	return 0; 
}