NOIP模拟 序列(主席树)

【题目描述】

最近小 X 对序列的研究产生了一定的兴趣。

小 X 把一个区间的中位数看成将这个区间内的数排序之后位于中间的数,如果区间的长度为偶数的话则是较小的数。

一开始,小 X 想研究一个序列的所有区间中,中位数最大的区间是哪一个。

后来他想了想觉得太简单了,于是决定把这个问题加大一点难度:小 X 现在想知道有多少区间的中位数位于他给定的 [l1, r1] 之间。

后来他想了想便自己解决了这个问题,于是他再加入了一个限制:区间的长度必须位于 [l2, r2]之间。

小 X 发现加了这样一个限制之后,他自己也不会做了。于是他想让你帮他解决这个问题。

【输入格式】

第一行有两个正整数 n,m,表示序列的长度和询问的次数。

第二行有 n 个数,分别表示原来序列的数。

接下来有 m 行,每一行有四个正整数,分别表示每个询问的 l1, r1, l2, r2。

【输出格式】

输出 m 行,每一行表示每个询问对应的答案。

【样例输入】

4 1 2 2 5 

2 4 2 3 

2 4 1 3 

2 5 2 5 

3 5 2 5 

1 5 3 5

【样例输出】

5

8

8

0

6

【备注】

对于 10% 的数据,n ≤ 100。

对于 30% 的数据,n ≤ 1000。

另有 40% 的数据,l2 = 1, r2 = n。

对于 100% 的数据,n, ai ≤ 100000,m ≤ 5。

【题目分析】

(下面全部copy,哪天补了主席树再来补更)

如果固定 c2 的位置,相当于限制了 c1 的选择范围,也就是要支持区间内查询小等于某个数的个数,可以通过把线段树可持久化来实现。

当然你不会可持久化线段树也没关系,可以离线下来做,把查询一个区间拆成两个前缀和相减,然后我们只需要支持插入,权值区间查询即可。用线段树就可以了。

时间复杂度 O(nmlog n)。

【代码~(也是嫖的。。。。)】

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int Rlen=1<<18|1;
const int MAXN=1e5+10;
const int L=50;

int n,m,a[MAXN],rt[MAXN],lc[MAXN*L],rc[MAXN*L],siz[MAXN*L];
int tot;

void init()
{
	for(int i=1;i<=tot;++i)
	  lc[i]=rc[i]=siz[i]=0;
	for(int i=1;i<=n;++i)
	  rt[i]=0;
	tot=0;
}

void copy(int &x,int y)
{
	x=++tot;
	lc[x]=lc[y];
	rc[x]=rc[y];
	siz[x]=siz[y];
}

void build(int x,int &y,int l,int r,int p)
{
	copy(y,x);
	++siz[y];
	if(l==r)
	  return ;
	int mid=l+r>>1;
	if(p<=mid)
	  build(lc[x],lc[y],l,mid,p);
	else
	  build(rc[x],rc[y],mid+1,r,p);
}

LL query(int x,int y,int l,int r,int L,int R)
{
	if(L<=l&&r<=R)
	  return siz[y]-siz[x];
	int mid=l+r>>1;
	if(R<=mid)
	  return query(lc[x],lc[y],l,mid,L,R);
	else
	{
		if(L>mid)
		  return query(rc[x],rc[y],mid+1,r,L,R);
		else
		  return query(lc[x],lc[y],l,mid,L,R)+query(rc[x],rc[y],mid+1,r,L,R);
	}
}

LL solve(int v,int l,int r)
{
	init();
	build(rt[1],rt[1],0,2*n,0+n);
	int cnt=0;
	LL sum=0;
	for(int i=1;i<=n;++i)
	{
		if(a[i]<=v)
		  ++cnt;
		else
		  cnt--;
		build(rt[i],rt[i+1],0,2*n,cnt+n);
		sum+=query(rt[max(i-r,0)],rt[max(i-l+1,0)],0,2*n,0,cnt+n);
	}
	return sum;
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;++i)
	  scanf("%d",&a[i]);
	scanf("%d",&m);
	while(m--)
	{
		int l1,r1,l2,r2;
		scanf("%d%d%d%d",&l1,&r1,&l2,&r2);
		printf("%lld\n",solve(r1,l2,r2)-solve(l1-1,l2,r2));
	}
	return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值