杭电2019多校第十场 HDU-6701 Make Rounddog Happy（单调栈+枚举）

链接：http://acm.hdu.edu.cn/showproblem.php?pid=6701

题意：求区间内数各不相同并且max(al,al+1,…,ar)−(r−l+1)≤k的区间的个数。

思路：当时读题忽略了区间内数各不相同这个条件，想着不就是单调栈加简单的组合数学吗？后来，写完样例不对才发现。。。。然后就想求出某一个数为终点的最长不重复区间和一某个数为起点的最长不重复区间，再和单调栈的求出的范围取个较大、较小值不就找出该数的作用区间了吗？后来想明白了，以他为终点和以他为起点的区间合起来不一定数不重复啊！然后就卡在怎么快速求包含一个数的最长不重复区间。然后，就没有然后了。。。。看了别人的思路，直接预处理出以某一个数为终点的最长不重复区间的L和一某个数为起点的最长不重复区间的端点R,然后直接枚举单调栈求出最大值左右区间较小的那一块求。如果就这么看的觉得时间太多，但其实均摊为nlong(n)，为啥咱也不知道。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 3e5+10;
const int maxm=3e5+10;
int l[N],r[N],L[N],R[N],sta[N],top,pos1,pos2;
int n,a[N],k,x;
int pre[N],suf[N];
ll ans,len;
void getL()
{
	for(int i=1;i<=n;i++) pre[i]=0;
	L[0]=0;
	for(int i=1;i<=n;i++)
	{
		L[i]=max(pre[a[i]]+1,L[i-1]);
		pre[a[i]]=i;
	}
}
void getR()
{
	for(int i=1;i<=n;i++) suf[i]=n+1;
	R[n+1]=n+1;
	for(int i=n;i>=1;i--)
	{
		R[i]=min(suf[a[i]]-1,R[i+1]);
		suf[a[i]]=i;	
	}  
}
void getl()
{	
	top=0;
	for(int i=1;i<=n;i++)
	{
		while(top&&a[sta[top]]<=a[i])
			top--;
		if(top) l[i]=sta[top]+1;
		else l[i]=1;
		sta[++top]=i;
	}		
}
void getr()
{	
	top=0;
	for(int i=n;i>=1;i--)
	{
		while(top&&a[sta[top]]<=a[i])
			top--;
		if(top) r[i]=sta[top]-1;
		else r[i]=n;
		sta[++top]=i;
	}		
}
int main(void) 
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&k);
		ans=0; 
		for(int i=1;i<=n;i++) 
			scanf("%d",&a[i]);
		getL();
		getR();
		getl();
		getr();
		for(int i=1;i<=n;i++)
		{
			x=a[i]-k;
			//取左右两边个数少的来枚举 
			if(i-l[i]<r[i]-i)
			{
				//枚举区间左端点 
				for(int j=l[i];j<=i;j++)
				{
					//第一个可选的右端点，区间内必须要有i 
					pos1=max(i,j+x-1);
					//最后一个可选的右端点 
					pos2=min(r[i],R[j]);
					if(pos2>=pos1) ans+=pos2-pos1+1;	
				}
			} 
			else
			{
				//枚举区间右端点 
				for(int j=i;j<=r[i];j++)
				{
					//最后一个可选的左端点  
					pos1=max(l[i],L[j]);
					//第一个可选的左端点，区间内必须要有i 
					pos2=min(i,j-x+1);
					if(pos2>=pos1) ans+=pos2-pos1+1;
				}
			}
		}
		printf("%lld\n",ans);
	}
	return 0;
}