浅谈主席树:https://www.cnblogs.com/AKMer/p/9956734.html
题目传送门:https://www.lydsy.com/JudgeOnline/problem.php?id=3524
假设数列不是一开始就给你的,而是一次一次修改操作让你插入数值\(x\),而询问是问你从第\(L\)次插入到第\(R\)次操作间,有没有哪个数值插入次数超过\((R-L+1)/2\),把题意这么一转化,就很好用主席树写了。
我们可以在值域上建立主席树,每个结点统计值域在\([l,r]\)内的数字有多少个。那么只需要用第\(r\)个版本的主席树的\(cnt\)减去第\(l-1\)个版本的主席树的\(cnt\),就是操作\([L,R]\)里插入了多少次值域在\([l,r]\)的数字。如果\([l,mid]\)的次数大于\((R-L+1)/2\)那么答案就在\([l,mid]\),同理可以去判断答案是否在\([mid+1,r]\)内,如果都不满足,那么就不存在这样的数值,直接返回\(0\)就行了。
时间复杂度:\(O((n+m)logn)\)
空间复杂度:\(O(nlogn)\)
代码如下:
#include <cstdio>
using namespace std;
const int maxn=5e5+5;
int n,m;
int rt[maxn];
int read() {
int x=0,f=1;char ch=getchar();
for(;ch<'0'||ch>'9';ch=getchar())if(ch=='-')f=-1;
for(;ch>='0'&&ch<='9';ch=getchar())x=x*10+ch-'0';
return x*f;
}
struct tree_node {
int cnt,ls,rs;
};
struct Chairman_tree {
int tot;
tree_node tree[maxn*20];
void updata(int p) {
tree[p].cnt=tree[tree[p].ls].cnt+tree[tree[p].rs].cnt;
}
void ins(int lst,int &now,int l,int r,int pos) {
now=++tot;tree[now]=tree[lst];//先把左右儿子信息全部继承下来
if(l==r) {tree[now].cnt++;return;}//值为l的数字比上一版本的主席树多了一个
int mid=(l+r)>>1;
if(pos<=mid)ins(tree[lst].ls,tree[now].ls,l,mid,pos);//新建左儿子
else ins(tree[lst].rs,tree[now].rs,mid+1,r,pos);//新建右儿子
updata(now);//更新当前节点信息
}
int query(int L,int R,int l,int r,int limit) {
if(l==r) {
if(tree[R].cnt-tree[L].cnt>limit)return l;//记得判断
else return 0;
}
int mid=(l+r)>>1;
int tmp1=tree[tree[R].ls].cnt-tree[tree[L].ls].cnt;
int tmp2=tree[tree[R].rs].cnt-tree[tree[L].rs].cnt;
if(tmp1>limit)return query(tree[L].ls,tree[R].ls,l,mid,limit);
if(tmp2>limit)return query(tree[L].rs,tree[R].rs,mid+1,r,limit);
return 0;//同题解所述
}
}T;
int main() {
n=read(),m=read();
for(int i=1;i<=n;i++) {
int x=read();
T.ins(rt[i-1],rt[i],1,n,x);//rt[i]表示第i个版本的主席树
}
for(int i=1;i<=m;i++) {
int l=read(),r=read(),limit=(r-l+1)>>1;
int ans=T.query(rt[l-1],rt[r],1,n,limit);
printf("%d\n",ans);
}
return 0;
}
原文:https://www.cnblogs.com/AKMer/p/9957065.html