区间计数(笛卡尔树+暴力)

【转载】作者：GuxueAkioi

学习之用，为防丢失，故此转载，万望见谅，若有侵权，联系删除。

题意：

规模：n、k<=300000，a[i]<=n。

分析：将式子移项就得到了 max(a[l]~a[r])-k<=区间长度，max和区间结合起来，容易让人想到笛卡尔树。把中间“区间内数字各不相同”的条件去掉的话，就是一个笛卡尔树的裸题了——把树建出来，然后对于每个点x，它子树的size就是以它为max的区间长度，然后把较小的儿子的子树中的节点拿来枚举。同时要求区间的另一个端点必须在点x和x子树的边界之间，并且长度要>=a[x]-k，全部加起来就完事了。既然想到这里，不难发现可以直接预处理出来每个点往左/右不经过重复的点最远可以到哪里，然后上述的区间再加一个不能超过这个边界的条件即可。时间上因为笛卡尔树是一个二叉树，较小的子树节点数一定<size/2，所以总的枚举的点数不会超过nlog2n个(如果每次左右子树size相同就只有log2n层，如果不同枚举的点数就会较少)。

Horse： 

#include<bits/stdc++.h>
using namespace std;
int n,m;
struct node{int x,son[2],siz,l,r;}a[300003];
int s[300003],top,root,lst[300003],lt[300003],rt[300003];
void buildtree()
{
	for(int i=1;i<=n;i++)
	{
		while(top&&a[s[top]].x<a[i].x) a[i].son[0]=s[top--];
		if(top) a[s[top]].son[1]=i;
		s[++top]=i;
	}
	return;
}
long long ans;
void dfs(int x,int l,int r)
{
	if(a[x].son[0]) dfs(a[x].son[0],l,x-1);
	if(a[x].son[1]) dfs(a[x].son[1],x+1,r);
	int len=max(a[x].x-m,1);
	if(x-l<r-x)
	{
		for(int i=l;i<=x;i++)
			if(rt[i]>=x)
				ans+=max(0,min(rt[i],r)-max(i+len-1,x)+1);
	}
	else
	{
		for(int i=x;i<=r;i++)
			if(lt[i]<=x)
				ans+=max(0,min(i-len+1,x)-max(lt[i],l)+1);
	}
	return;
}
int main()
{
	scanf("%d%d",&n,&m);
	int maxn=0;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i].x);
		if(a[i].x>maxn) maxn=a[i].x,root=i;
		lt[i]=max(lst[a[i].x]+1,lt[i-1]);
		lst[a[i].x]=i;
	}
	for(int i=1;i<=n;i++) lst[i]=n+1;
	rt[n]=n;
	for(int i=n-1;i;i--)
	{
		rt[i]=min(lst[a[i].x]-1,rt[i+1]);
		lst[a[i].x]=i;
	}
	buildtree();
	dfs(root,1,n);
	printf("%lld\n",ans);
	return 0;
} 作者：GuxueAkioi https://www.bilibili.com/read/cv13579847?from=search&spm_id_from=333.337.0.0 出处：bilibili