hdu6602 2019杭电多校第二场 Longest Subarray

这题考场上都没去自习思考。。。其实就是个cf div2 D题难度的线段树,比那个三角形简单多了。。。

我们枚举每一个区间r,看他左边最远的是的l-r合法的l,可以想到,r所在位置的数字a[r]从最左边到a[r]上k个前出现的位置lastk[r],1---lastk[r]是对于a[r]这个数字是满足条件的,因为从r到那段区间的位置a[r]的出现次数是>=k的。

我们建线段树,枚举r=1->n,val[l]表示当前r下 l 这个位置有多少个数字还没有>=k。那么想当于是吧 [ lastk[r]+1 , i ]这段区间+1,来表示这段区间不接受a[r]这个数字。那么最左边的 val[l]=0的数字,[l,r]这个区间就是以r为有边界 合法的最长的序列。

每次挪动r,[last[r]+1 , i] 加一   [lastk[last[r]]+1,lastk[i]]减一,然后线段树维护最小值,找最左边的0就行了。

 

#include
#define maxl 100010
using namespace std;

int n,c,k,ans;
int a[maxl],lastk[maxl],last[maxl];
vector  b[maxl];
struct node
{
	int l,r,mi,tag;
}tree[maxl*4];

inline void build(int k,int l,int r)
{
	tree[k].l=l;tree[k].r=r;tree[k].tag=0;
	if(l==r)
	{
		tree[k].mi=0;
		tree[k].tag=0;
		return;
	}
	int mid=(l+r)>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
	tree[k].mi=min(tree[k<<1].mi,tree[k<<1|1].mi);
}

inline void prework()
{
	for(int i=1;i<=n;i++)
		lastk[i]=0,last[i]=0;
	for(int i=1;i<=c;i++)
		b[i].clear();
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]),b[a[i]].push_back(i);
	for(int i=1;i<=c;i++)
	{
		int l=b[i].size();
		for(int j=1;j=k-1)
				lastk[b[i][j]]=b[i][j-k+1];
			last[b[i][j]]=b[i][j-1];
		}
	}
	build(1,1,n);
}

inline void gank(int k)
{
	int x=tree[k].tag;
	if(tree[k].tag!=0)
	{
		if(tree[k].l!=tree[k].r)
		{
			tree[k<<1].mi+=x;
			tree[k<<1|1].mi+=x;
			tree[k<<1].tag+=x;
			tree[k<<1|1].tag+=x;
		}
		tree[k].tag=0;
	}
}

inline void add(int k,int l,int r,int x)
{
	gank(k);
	if(tree[k].l==l && tree[k].r==r)
	{
		tree[k].mi+=x;tree[k].tag+=x;
		return;
	}
	int mid=(tree[k].l+tree[k].r)>>1;
	if(r<=mid)
		add(k<<1,l,r,x);
	else if(l>mid)
		add(k<<1|1,l,r,x);
	else
		add(k<<1,l,mid,x),add(k<<1|1,mid+1,r,x);
	tree[k].mi=min(tree[k<<1].mi,tree[k<<1|1].mi);
}

inline int query(int k)
{
	gank(k);
	if(tree[k].l==tree[k].r)
		return tree[k].l;
	int mid=(tree[k].l+tree[k].r)>>1;
	int ret=0;
	if(tree[k<<1].mi==0)
		ret=query(k<<1);
	else
	if(tree[k<<1|1].mi==0)
		ret=query(k<<1|1);
	tree[k].mi=min(tree[k<<1].mi,tree[k<<1|1].mi);
	return ret;
}

inline void mainwork()
{
	if(k==1)
	{
		ans=n;
		return;
	}
	ans=0;int la,l,r;
	for(int i=1;i<=n;i++)
	{
		if(lastk[i]>0)
		{
			la=last[i];
			add(1,lastk[la]+1,lastk[i],-1);
		}
		add(1,last[i]+1,i,1);
		if(tree[1].mi==0)
		{
			int d=query(1);
			if(d>0 && d<=i)
				ans=max(ans,i-d+1);
		}
	}
}

inline void print()
{
	printf("%d\n",ans);
}

int main()
{
//	freopen("1012.in","r",stdin);
	while(~scanf("%d%d%d",&n,&c,&k))
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}

 

你可能感兴趣的:(线段树)