版权声明:本文为博主原创文章,未经博主允许不得转载。 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;
}