主席树入门题-2018湘潭邀请赛部分题解*

(有任何问题欢迎留言或私聊


 本题完全是主席树裸题,我这里简单提一下我对主席树的理解。我的题解在后面,你可以直接去看题解。
 感觉我写的应该挺好理解的,只要你学了线段树,可能你不会写,但应该能有个印象。

主席树:

 主席树又称可持久化线段树,数据离散化后,编号为i的节点维护的是前缀1-i这段区间的线段树。
 如果用普通线段树的方法写,你要建n个线段树,时间和空间都太浪费了,于是某神人开发这意数据结构。
 你会发现i号节点维护的是前缀i这段区间的线段树,而i和它左右的两个线段树只差了一个元素而已;而且每一个线段树的结构是一模一样的,不同的只是每个线段树包含的值域不同。
 所以你要利用前一个线段树信息;相较于前一颗树只是多了一个值,也就只有包含这个值的log个节点不同,所以对于每一个线段树,我们只需要新开log个节点,其它的节点就用第i-1颗线段树的节点。

处理:

 对于主席树,我们要开一个root数组来储存每一个线段树的根,最开始是一个空树,所以root[0]=0;而主席树的节点储存的是左右儿子的编号信息。
 因为每个线段树的构造完全相同,只是所包含的值域不同;所以对于区间[L,R]这段线段树你可以由root[R]-root[L-1]得到。
 注意,相邻两颗线段树只有当前待处理的元素不同,其余位置完全一样。
 因此,如果待处理的元素进入线段树的左子树的话,右子树是完全一样的,可以共用,即直接让当前线段树节点的右子树指向相邻的上一棵线段树的右子树;若进入右子树,情况可以类比。
 这个过程时间复杂度为O(logn),空间复杂度为 O(logn)。


另外推荐一下:
qsc学姐的主席树算法直播课

C. Just h-index

The h-index of an author is the largest h where he has at least h papers with citations not less than h. Bobo has published n papers with citations a1,a2,…,an respectively. One day, he raises q questions. The i-th question is described by two integers li and ri, asking the h-index of Bobo if has only published papers with citations ali,ali+1,…,ari.
Input The input consists of several test cases and is terminated by end-of-file. The first line of each test case contains two integers n and q. The second line contains n integers a1,a2,…,an. The i-th of last q lines contains two integers li and ri.
Output For each question, print an integer which denotes the answer.
Constraint • 1≤ n,q ≤105 • 1≤ ai ≤ n • 1≤ li ≤ ri ≤ n • The sum of n does not exceed 250,000. • The sum of q does not exceed 250,000.
Sample Input
5 3
1 5 3 2 1
1 3
2 4
1 5
5 1
1 2 3 4 5
1 5
Sample Output
2 2 2
3

题意及思路:

 给你一段长度为n的序列,q次询问;每次询问[L,R]区间内的h值是多少?

h值的意思:
栗子:4 2 6 3 1;表示这区间内,1,2,3,4,6各出现了一次。h值表示大于等于h的数出现了不少于h次且h要最大。

你会发现这个题和主席树的入门题静态区间第k小十分相像;
本题要求的是最大的 第k大的数 出现了不少于k次。
其实就是求区间第k大问题,你可以直接求,也可以转化为求区间第k小,两种方法我都ac了。
对于最大化一个值的问题二分一下就行了。


用主席数求出第mid大的数,如果mid<=这个数,则L=mid+1;反之R=mid-1;
二分的写法大同小异,用自己习惯的写法就好。


数据量是1e5,主席树的复杂度是O(nlogn);q次查询,主席树查询是O(log),二分的复杂度是O(log),所以本题复杂度是O(nlog^2)二分加主席树完全可过。

AC代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
const int N = 200005;
const int INF = 0x3f3f3f3f;
int n,q,tot;
struct lp
{
	int l,r,sum;
	lp(){l=r=sum=0;}
}cw[N*20];
int ar[N],br[N],now[N];
int root[N];
void update(int l,int r,int last,int &cur,int x){
	cw[++tot]=cw[last];
	cw[tot].sum++;
	cur=tot;
	if(l==r)return;
	int mid = (l+r)>>1;
	if(x<=mid){
		update(l,mid,cw[last].l,cw[cur].l,x);
	}else {
		update(mid+1,r,cw[last].r,cw[cur].r,x);
	}
}
int query(int l,int r,int last,int cur,int k){
	if(l==r)return l;
	int l1=cw[last].l,l2=cw[cur].l,r1=cw[last].r,r2=cw[cur].r;
	int summ=cw[l2].sum-cw[l1].sum;
	int mid=(l+r)>>1;
	if(k<=summ){
		return query(l,mid,cw[last].l,cw[cur].l,k);
	}else{
		return query(mid+1,r,cw[last].r,cw[cur].r,k-summ);
	}
}
int main(){
	while(~scanf("%d%d",&n,&q)){
		memset(root,0,sizeof(root));
		cw[0].l=cw[0].r=cw[0].sum=0;
		for(int i=1;i<=n;++i){
			scanf("%d",&br[i]);
			ar[i]=br[i];
		}
		sort(br+1,br+1+n);
		int k=1;
		for(int i=2;i<=n;++i){
			if(br[i]!=br[i-1])br[++k]=br[i];
		}
		tot=0;
		for(int i=1;i<=n;++i){
			now[i]=lower_bound(br+1,br+1+k,ar[i])-br;
		}
		for(int i=1;i<=n;++i){
			update(1,k,root[i-1],root[i],now[i]);
		}
		while(q--){
			int u,v;
			scanf("%d%d",&u,&v);
			int L=1,R=v-u+1,mid,ans=1;
			while(L<=R){//二分的写法大同小异,用自己习惯的写法
				mid=(L+R)>>1;
				int tmp=query(1,k,root[u-1],root[v],v-u+1-mid+1);
				if(mid>br[tmp]){
					ans=mid-1;
					R=mid-1;
				}else{
					ans=mid;
					L=mid+1;
				}
			}
			printf("%d\n",ans);
		}
	}
	return 0;
}



主席树板子
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100000 + 5;

int a[N], b[N], rt[N * 20], ls[N * 20], rs[N * 20], sum[N * 20];

int n, k, tot, sz, ql, qr, x, q, T;

void Build(int& o, int l, int r){
    o = ++ tot;
    sum[o] = 0;
    if(l == r) return;
    int m = (l + r) >> 1;
    Build(ls[o], l, m);
    Build(rs[o], m + 1, r);
}

void update(int& o, int l, int r, int last, int p){
    o = ++ tot;
    ls[o] = ls[last];
    rs[o] = rs[last];
    sum[o] = sum[last] + 1;
    if(l == r) return;
    int m = (l + r) >> 1;
    if(p <= m)  update(ls[o], l, m, ls[last], p);
    else update(rs[o], m + 1, r, rs[last], p);
}

int query(int ss, int tt, int l, int r, int k){
    if(l == r) return l;
    int m = (l + r) >> 1;
    int cnt = sum[ls[tt]] - sum[ls[ss]];
    if(k <= cnt) return query(ls[ss], ls[tt], l, m, k);
    else return query(rs[ss], rs[tt], m + 1, r, k - cnt);
}

void work(){
    scanf("%d%d%d", &ql, &qr, &x);
    int ans = query(rt[ql - 1], rt[qr], 1, sz, x);
    printf("%d\n", b[ans]);
}

int main(){
    scanf("%d", &T);
    while(T--){
        scanf("%d%d", &n, &q);
        for(int i = 1; i <= n; i ++) scanf("%d", a + i), b[i] = a[i];
        sort(b + 1, b + n + 1);
        sz = unique(b + 1, b + n + 1) - (b + 1);
        tot = 0;
        Build(rt[0],1, sz);
        //for(int i = 0; i <= 4 * n; i ++)printf("%d,rt =  %d,ls =  %d, rs = %d, sum = %d\n", i, rt[i], ls[i], rs[i], sum[i]);
        for(int i = 1; i <= n; i ++)a[i] = lower_bound(b + 1, b + sz + 1, a[i]) - b;
        for(int i = 1; i <= n; i ++)update(rt[i], 1, sz, rt[i - 1], a[i]);
        for(int i = 0; i <= 5 * n; i ++)printf("%d,rt =  %d,ls =  %d, rs = %d, sum = %d\n", i, rt[i], ls[i], rs[i], sum[i]);
        while(q --)work();
    }
    return 0;
}

划分树归并树模板:点这里
分块模板:点这里

神代码


主席树类似于线段树,由于点更新每次只会更新logn个节点,我们可以申请内存(用数组模拟) ,当然一开始可以把整棵原始树要建立起来,更新的时候要用root数组记录当前版本树根的标号。这样点询问的时间和空间复杂度都是logn。

主席树支持还原历史版本(因为每个版本的树都已经存下来了) 数在区间第一次出现 区间有多少不同数 树套树的操作

关键在于如何建树,就是树的一维存什么,二维存什么,各自有什么含义。

http://poj.org/problem?id=2104 主席树询问区间第K大值 基本操作

http://blog.csdn.net/ied98/article/details/42799885 用启发式合并套主席树

http://blog.csdn.net/ied98/article/details/42977283 询问区间的众数

http://blog.csdn.net/ied98/article/details/46861035

http://blog.csdn.net/ied98/article/details/46861105 最基本的树套树

主席树是支持区间打标记的,具体用法和线段树的pushup pushdown类似 时间久了 博主并没有找到题所以在这里提一下。。

http://blog.csdn.net/eod_realize/article/details/60768476 询问区间不同数的个数

http://blog.csdn.net/eod_realize/article/details/60976267 区间不同数的个数相关

对于区间不同数的个数 有两种解法 不过推荐用离线倒着插点的方法,另一种解法很有局限性

http://blog.csdn.net/eod_realize/article/details/72848363 主席树脑洞题


极角排序等基础计算几何模板

struct lp
{
	double x,y,rad;
	int id;
	lp(){}
	lp(double a,double b){x=a,y=b;}
	lp operator - (const lp &b) const{
        return lp(x - b.x, y - b.y);
    }
}cw[N],st,p[N];
inline double dist(lp &a,lp &b){
	return (a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y);
}
inline double crossDot(lp a,lp b){
	return a.x*b.y-a.y*b.x;
}
inline double area2(lp a,lp b,lp c){
	return crossDot(b-a,c-a);
}
bool cmp2(lp a,lp b){
	lp c;
	c.x=c.y=0;
	double tmp=area2(c,a,b);
	if(tmp==0)return a.x<b.x;
	return tmp>0;
}
int Quadrant1(lp a)//象限排序,注意包含四个坐标轴
{
    if(a.x>0&&a.y>=0)  return 3;
    if(a.x<=0&&a.y>0)  return 4;
    if(a.x<0&&a.y==0)  return 4;
    if(a.x<0&&a.y<=0)  return 1;
    if(a.x>=0&&a.y<0)  return 2;
}
bool cmp5(lp &a,lp &b){
	int qa=Quadrant1(a),qb=Quadrant1(b);
	if(qa==qb){
		return cmp2(a,b);
	}
	return qa<qb;
}
double myatan(double y,double x)
{
	double t=atan2(y,x);
	return t>=0?t:2*PI+t;
}
bool cmp(lp &a,lp &b){
	return a.rad<b.rad;
}
bool Left(lp &a,lp &b){
	return (a.x*b.y-a.y*b.x)>=0;
}
struct lp
{
    double x,y,rad;
    int id;
    lp(){}
    lp(double a,double b){x=a,y=b;}
    lp operator - (const lp &b) const{
        return lp(x - b.x, y - b.y);
    }
}cw[N];
inline double crossDot(lp a,lp b){
    return a.x*b.y-a.y*b.x;
}
inline double area2(lp a,lp b,lp c){
    return crossDot(b-a,c-a);
}

bool cmp1(lp a,lp b){
    double p=atan2(a.y,a.x),q=atan2(b.y,b.x);
    if(p!=q){
        return p<q;
    }
    return a.x<b.x;
}
bool cmp2(lp a,lp b){
    lp c;
    c.x=c.y=0;
    double tmp=area2(c,a,b);
    if(tmp==0)return a.x<b.x;
    return tmp>0;
}
int Quadrant(lp a)//象限排序,注意包含四个坐标轴
{
    if(a.x>0&&a.y>=0)  return 1;
    if(a.x<=0&&a.y>0)  return 2;
    if(a.x<0&&a.y<=0)  return 3;
    if(a.x>=0&&a.y<0)  return 4;
    return -1;
}
bool cmp3(lp a,lp b){
    if(Quadrant(a)==Quadrant(b)){
        return cmp1(a,b);
    }
    return Quadrant(a)<Quadrant(b);
}

bool cmp4(const lp &a, const lp &b)//先按象限排序,再按极角排序,再按远近排序 
{
    if (a.y == 0 && b.y == 0 && a.x*b.x <= 0)return a.x>b.x;
    if (a.y == 0 && a.x >= 0 && b.y != 0)return true;
    if (b.y == 0 && b.x >= 0 && a.y != 0)return false;
    if (b.y*a.y <= 0)return a.y>b.y;
    lp one;
    one.y = one.x = 0;
    return area2(one,a,b) > 0 || (area2(one,a,b) == 0 && a.x < b.x);    
}
int Quadrant1(lp a)//象限排序,注意包含四个坐标轴
{
    if(a.x>0&&a.y>=0)  return 3;
    if(a.x<=0&&a.y>0)  return 4;
    if(a.x<0&&a.y==0)  return 4;
    if(a.x<0&&a.y<=0)  return 1;
    if(a.x>=0&&a.y<0)  return 2;
}
bool cmp5(lp &a,lp &b){
    int qa=Quadrant1(a),qb=Quadrant1(b);

    if(qa==qb){
        return cmp2(a,b);
    }
    return qa<qb;
}
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值