码迷,mamicode.com
首页 > 其他好文 > 详细

【BZOJ3956】Count 主席树+单调栈

时间:2017-07-07 22:40:34      阅读:205      评论:0      收藏:0      [点我收藏+]

标签:output   getchar   put   alt   namespace   getc   esc   read   img   

【BZOJ3956】Count

Description

技术分享

Input

技术分享

Output

技术分享

Sample Input

3 2 0
2 1 2
1 1
1 3

Sample Output

0
3

HINT

M,N<=3*10^5,Ai<=10^9

题解:影魔那道题的简化版,依旧用单调栈记录每个数左(右)边第一个比它大的数为ld[i],rd[i],那么合法的点对只有(i,i+1)和(ld[i],rd[i])两种。为了防止重复,我们记录每个数左边第一个>=它的数le[i],当且仅当le[i]==ld[i]的时候我们才统计这样的点对。然后就变成了在一个二维平面内有一些点,每次问你一个矩形中点的个数,用主席树就行了。

但是自从观摩了单调栈+ST表的做法后感觉自己的做法烂成渣~

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int maxn=300010;
int n,m,typ,ans,tot,top,cnt;
struct sag
{
	int ls,rs,sum;
}s[maxn*30];
int ld[maxn],rd[maxn],le[maxn],v[maxn],st[maxn],to[maxn],next[maxn],head[maxn],rt[maxn];
void add(int a,int b)
{
	to[++cnt]=b,next[cnt]=head[a],head[a]=cnt;
}
int readin()
{
	int ret=0;	char gc=getchar();
	while(gc<‘0‘||gc>‘9‘)	gc=getchar();
	while(gc>=‘0‘&&gc<=‘9‘)	ret=ret*10+gc-‘0‘,gc=getchar();
	return ret;
}
void insert(int x,int &y,int l,int r,int pos)
{
	if(!y||y==x)	y=++tot,s[y].sum=s[x].sum;
	s[y].sum++;
	if(l==r)	return ;
	int mid=l+r>>1;
	if(pos<=mid)	s[y].rs=(!s[y].rs)?s[x].rs:s[y].rs,insert(s[x].ls,s[y].ls,l,mid,pos);
	else	s[y].ls=(!s[y].ls)?s[x].ls:s[y].ls,insert(s[x].rs,s[y].rs,mid+1,r,pos);
}
int query(int x,int y,int l,int r,int a,int b)
{
	if(a<=l&&r<=b)	return s[y].sum-s[x].sum;
	int mid=l+r>>1;
	if(b<=mid)	return query(s[x].ls,s[y].ls,l,mid,a,b);
	if(a>mid)	return query(s[x].rs,s[y].rs,mid+1,r,a,b);
	return query(s[x].ls,s[y].ls,l,mid,a,b)+query(s[x].rs,s[y].rs,mid+1,r,a,b);
}
int main()
{
	int i,j,a,b;
	n=readin(),m=readin(),typ=readin();
	for(i=1;i<=n;i++)	v[i]=readin();
	for(top=0,i=1;i<=n;i++)
	{
		while(top&&v[st[top]]<v[i])	rd[st[top--]]=i;
		if(top)	le[i]=st[top];
		st[++top]=i;
	}
	while(top)	rd[st[top--]]=n+1;
	for(top=0,i=1;i<=n;i++)
	{
		while(top&&v[st[top]]<=v[i])	top--;
		if(top)	ld[i]=st[top];
		st[++top]=i;
		if(ld[i]&&ld[i]==le[i])	add(rd[i],ld[i]);
	}
	for(i=1;i<=n;i++)
	{
		if(i>1)	insert(rt[i-1],rt[i],1,n,i-1);
		for(j=head[i];j;j=next[j])	insert(rt[i-1],rt[i],1,n,to[j]);
	}
	for(i=1;i<=m;i++)
	{
		a=(readin()+ans-1)%n+1,b=(readin()+ans-1)%n+1;
		if(a>b)	swap(a,b);
		printf("%d\n",ans=query(rt[a-1],rt[b],1,n,a,b)),ans*=typ;
	}
	return 0;
}

【BZOJ3956】Count 主席树+单调栈

标签:output   getchar   put   alt   namespace   getc   esc   read   img   

原文地址:http://www.cnblogs.com/CQzhangyu/p/7134142.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!