题目
分析
边录边处理
这道题不能把信息全录进树状数组再查询,我们需要先把排名和问题都先录进数组中,校长的问题信息按 a从小到大 排个序。然后按照学号,依次录入排名,当循环中录到第a个学生时,且校长的其中一个问题也问到前a个学生,此时就可以查询前树状数组的前b个和,这个前b个和就是答案。
AC
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int MAXN=1e5+1;
int ans[MAXN],c[MAXN],num[MAXN];
int n,m;
struct node{//存储问题
int a,b,pos;
bool operator<(const node f)const{
return a<f.a;
}
}question[MAXN];
void change(int x){
for(;x<=MAXN;x+=x&(-x))
c[x]++;
return;
}
int getsum(int x){
int re=0;
for(;x;x-=x&(-x))
re+=c[x];
return re;
}
int main()
{
memset(c,0,sizeof(c));
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){//录入学号排名
scanf("%d",&num[i]);
}
for(int i=1;i<=m;i++){//录入校长问题
scanf("%d%d",&question[i].a,&question[i].b);
question[i].pos=i;//记录这是第几个该问题
}
sort(question+1,question+m+1);//将问题信息按a从小到大排序
int cnt=1;
for(int i=1;i<=n;i++){
change(num[i]);
while((question[cnt].a==i)){//当找到问题中的学号与当前录入的学号匹配
ans[question[cnt].pos]=getsum(question[cnt].b);//查询并录入信息
cnt++;
}
}
for(int i=1;i<=m;i++){
printf("%d\n",ans[i]);
}
return 0;//give me five
}