牛客小白月赛9-E 换个角度思考

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/C_13579/article/details/84327172

地址：https://ac.nowcoder.com/acm/contest/275/E

思路：主席树 || 树状数组

主席树

Code:

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;

const int MAX_N=100005;
struct node{
	int x;
	int id;
	bool operator<(const node &p)const{
		return (x==p.x)?(id<p.id):(x<p.x);
	}
};
struct Tree{
	int l,r;
	int sum,Max;
};
int n,Q;
node a[MAX_N];
int d[MAX_N];
Tree tree[MAX_N*20];	//主席树空间大小O(nlog(n)) 最多 O(n*40) 
int cnt,root[MAX_N];	//当前线段树,线段树根节点位置

void Init();
void Update(int x,int &rt,int l,int r);
int Query(int L,int R,int k,int l,int r);
int main()
{
	scanf("%d%d",&n,&Q);
	for(int i=1;i<=n;a[i].id=i,++i)
		scanf("%d",&a[i].x);
	sort(a+1,a+n+1);
	for(int i=1;i<=n;++i)
		d[a[i].id]=i;
	Init();
	for(int i=1;i<=n;++i)
	{
		root[i]=root[i-1];
		Update(d[i],root[i],1,n);
	}
	int l,r,k;
	while(Q--){
		scanf("%d%d%d",&l,&r,&k);
		int t=lower_bound(a+1,a+n+1,node{k,(int)1e6})-a-1;
		if(t)	printf("%d\n",Query(root[l-1],root[r],t,1,n));
		else	printf("0\n");
	}
	
	return 0;
}

void Init()
{
	cnt=1;	root[0]=0;
	tree[0].l=tree[0].r=tree[0].sum=tree[0].Max=0;
}

void Update(int x,int &rt,int l,int r)
{
	tree[cnt++]=tree[rt];	rt=cnt-1;
	++tree[rt].sum;
	tree[rt].Max=max(tree[rt].Max,x);
	if(l==r)	return;
	int mid=(l+r)>>1;
	if(x<=mid)	Update(x,tree[rt].l,l,mid);
	else	Update(x,tree[rt].r,mid+1,r);
}

int Query(int L,int R,int k,int l,int r)
{
	if(l==r)	return tree[R].sum-tree[L].sum;
	int mid=(l+r)>>1,ans=0;
	if(k<=mid)	ans=Query(tree[L].l,tree[R].l,k,l,mid);
	else	ans=Query(tree[L].r,tree[R].r,k,mid+1,r)+tree[tree[R].l].sum-tree[tree[L].l].sum;
	return ans;
}

树状数组：将所有询问按k由小到大排序，在遍历一遍，每次将<=k的元素更新即可

Code:

#include <cstdio>
#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;

const int MAX_N=1e5+5;
struct node{
	int x;
	int id;
	bool operator<(const node &p)const{
		return (x==p.x)?(id<p.id):(x<p.x);
	}
};
struct que{
	int l,r,x,id;
	bool operator<(const que &p)const{
		return x<p.x;
	}
};
int n,m;
node a[MAX_N];
que b[MAX_N];
int res[MAX_N],C[MAX_N];

int Lowbit(int x);
void Update(int id,int x);
int Query(int id);
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;a[i].id=i,++i)
		scanf("%d",&a[i].x);
	sort(a+1,a+n+1);
    for(int i=1;i<=m;b[i].id=i,++i)
        scanf("%d%d%d",&b[i].l,&b[i].r,&b[i].x);
    sort(b+1,b+m+1);
    int li=1;
    for(int i=1;i<=m;++i)
    {
    	while(li<=n&&a[li].x<=b[i].x){
    		Update(a[li].id,1);
    		++li;
		}
		res[b[i].id]=Query(b[i].r)-Query(b[i].l-1);
	}
	for(int i=1;i<=m;++i)
		printf("%d\n",res[i]);
    
    return 0;
}

int Lowbit(int x)
{
	return x&(-x);
}

void Update(int id,int x)
{
	while(id<=n){
		C[id]+=x;
		id+=Lowbit(id);
	}
}

int Query(int id)
{
	int res=0;
	while(id>0){
		res+=C[id];
		id-=Lowbit(id);
	}
	return res;
}