【HDU4747】【JZOJ4680】自然数

本文介绍了一种求解区间MEX之和的算法。给定n个数,通过线段树维护每个区间[i,j]内最小未出现的自然数(MEX),并最终计算出所有区间MEX值的总和。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

###Description
给n个数,mex(i,j)mex(i,j)mex(i,j)表示区间[i,j][i,j][i,j]最小没有出现的自然数。

求:
∑i=1n∑j=inmex(i,j)\sum_{i=1}^n\sum_{j=i}^nmex(i,j)i=1nj=inmex(i,j)

###Solution

首先我们O(n)O(n)O(n)求出mex(1,i)mex(1,i)mex(1,i)

我们现在先考虑从mex(1,i)mex(1,i)mex(1,i)推到mex(2,i)mex(2,i)mex(2,i),其实就是删掉第一个元素。

很显然:mex(i,i),mex(i,i+1)⋯mex(i,n)mex(i,i),mex(i,i+1)\cdots mex(i,n)mex(i,i),mex(i,i+1)mex(i,n)是单调递增的。

我们有一串序列a1,a2,⋯ ,ax,⋯ ,ana_1,a_2,\cdots,a_x,\cdots,a_na1,a2,,ax,,an

这里我们假设a1=axa_1=a_xa1=ax,那么如果删掉了a1a_1a1,那么做出的影响是:所有111xxx之间的元素,如果mexmexmex值大于a1a_1a1,那么mexmexmex值要变为a1a_1a1。同时,由于这个mexmexmex是单调递增的,我们可以用线段树维护,在线段树上二分。

根据上面的例子,我们就可以从mex(j,i)(j<i)mex(j,i)(j<i)mex(j,i)(j<i)推到mex(j+1,i)mex(j+1,i)mex(j+1,i)

于是线段树区间修改查询即可。

###Code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
#define N 200001
#define ll long long
using namespace std;
int a[N];
int mex[N];
int nx[N];
int cx[N];
bool bz[N];
struct node{
	int mx;
	ll h;
	int lz;
}tr[N*4];
int rt;
int n;
void build(int v,int l,int r)
{
	if(l==r)
	{
		tr[v].h=tr[v].mx=mex[l];
		tr[v].lz=-1;
		if(l==n) rt=v;
		return;
	}
	int mid=(l+r)/2;
	build(v*2,l,mid);
	build(v*2+1,mid+1,r);
	tr[v].h=tr[v*2].h+tr[v*2+1].h;
	tr[v].mx=max(tr[v*2].mx,tr[v*2+1].mx);
	tr[v].lz=-1;
}
void put(int v,int l,int r)
{
	int mid=(l+r)/2;
	int p=tr[v].lz;
	if(p==-1) return;
	tr[v*2].h=p*1ll*(mid-l+1);
	tr[v*2+1].h=p*1ll*(r-mid);
	tr[v*2].mx=tr[v*2+1].mx=p;
	tr[v*2].lz=tr[v*2+1].lz=p;
	tr[v].lz=-1;
}
void change(int v,int l,int r,int x,int y,int t)
{
	if(l==x && r==y)
	{
		tr[v].lz=t;
		tr[v].h=t*1ll*(r-l+1);
		tr[v].mx=t;
		return;
	}
	put(v,l,r);
	int mid=(l+r)/2;
	if(y<=mid) change(v*2,l,mid,x,y,t);
	else if(x>mid) change(v*2+1,mid+1,r,x,y,t);
	else
	{
		change(v*2,l,mid,x,mid,t);
		change(v*2+1,mid+1,r,mid+1,y,t);
	}
	tr[v].h=tr[v*2].h+tr[v*2+1].h;
	tr[v].mx=max(tr[v*2].mx,tr[v*2+1].mx);
}
int findpos(int v,int l,int r,int t)
{
	if(l==r) return l;
	put(v,l,r);
	int mid=(l+r)/2;
	if(tr[v*2].mx<t) return findpos(v*2+1,mid+1,r,t);
	else return findpos(v*2,l,mid,t);
}
ll find(int v,int l,int r,int x,int y)
{
	if(l==x && r==y) return tr[v].h;
	put(v,l,r);
	int mid=(l+r)/2;
	if(y<=mid) return find(v*2,l,mid,x,y);
	else if(x>mid) return find(v*2+1,mid+1,r,x,y);
	else return find(v*2,l,mid,x,mid)+find(v*2+1,mid+1,r,mid+1,y);
}
int main()
{
	cin>>n;
	fo(i,1,n)
	{
		scanf("%d",&a[i]);
		if(a[i]>n) a[i]=n;
		if(cx[a[i]]) nx[cx[a[i]]]=i;
		cx[a[i]]=i;
	}
	int p=0;
	fo(i,1,n)
	{
		bz[a[i]]=true;
		while(bz[p]) p++;
		mex[i]=p;
	}
	build(1,1,n);
	ll ans=0;
	ans+=tr[1].h;
	fo(i,1,n-1)
	{
		if(!nx[i]) nx[i]=n+1;
		int pos=findpos(1,1,n,a[i]);
		if(pos<=nx[i]-1 && tr[rt].mx>a[i]) change(1,1,n,pos,nx[i]-1,a[i]);
		ans+=find(1,1,n,i+1,n);
	}
	cout<<ans;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值