绍兴一中模拟赛10.30——序列

蜗角虚名羊

于 2018-10-30 16:47:10 发布

阅读量223

点赞数 1

CC 4.0 BY-SA版权

分类专栏：一中差分线段树 main

本文链接：https://blog.csdn.net/xumingyang0/article/details/83544603

线段树同时被 3 个专栏收录

20 篇文章

订阅专栏

一中

13 篇文章

订阅专栏

main

13 篇文章

订阅专栏

本文介绍了一种解决区间中位数查询问题的算法，针对长度不超过10万的序列和最多5次查询，提供了两种解决方案：一种是O(nmlogn)复杂度的方法，通过标记小于等于目标值的元素进行区间和计算；另一种是O(nm)复杂度的灵活差分方法，提高了查询效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：给出一个长度为 $n (n < = 100000)$ 的序列，有 $m (m < = 5)$ 次询问，每次询问有多少个长度 $\in [l 2, r 2]$ 的区间，中位数大小 $\in [l 1, r 1]$

Solution

$O (n m l o g n)$ ：把 $< = x$ 的记为 $1$ ， $> x$ 的记为 $- 1$ ，计算区间和 $> = 0$ 的区间数
$O (n m)$ ：灵活差分

Code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100001;
int a[N],b[N],c[N<<1],n,m,i,l1,r1,l2,r2;
inline char gc(){
	static char buf[100000],*p1=buf,*p2=buf;
	return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int rd(){
	int x=0,fl=1;char ch=gc();
	for (;ch<48||ch>57;ch=gc())if(ch=='-')fl=-1;
	for (;48<=ch&&ch<=57;ch=gc())x=(x<<3)+(x<<1)+(ch^48);
	return x*fl;
}
ll ask(int x,int y){
	memset(c,0,(2*n+1)<<2);
	b[0]=n;
	int s=0;ll ans=0;
	for (int i=1;i<=n;i++){
		if (a[i]<=x) b[i]=b[i-1]+1,s+=c[b[i]];
		else s-=c[b[i-1]],b[i]=b[i-1]-1;
		if (i>=y) c[b[i-y]]++,s+=(b[i]>=b[i-y]);
		ans+=s;
	}
	return ans;
}
int main(){
	n=rd();
	for (i=1;i<=n;i++) a[i]=rd();
	for (m=rd();m--;){
		l1=rd(),r1=rd(),l2=rd(),r2=rd();
		printf("%lld\n",ask(r1,l2)-ask(l1-1,l2)-ask(r1,r2+1)+ask(l1-1,r2+1));
	}
}