COJ1981-小M的魔术表演

来源:互联网 发布:模拟退火算法题目应用 编辑:程序博客网 时间:2024/06/10 18:46

小M的魔术表演

Description
小M听说会变魔术的男生最能吸引女生注意啦~所以小M费了九牛二虎之力终于学会了一个魔术:
首先在桌面上放N张纸片,每张纸片上都写有一个数字。小M每次请女生给出一个数字x,然后划定任意一个区间[L,R],小M就能立马告诉对方这个区间内有多少个数字比x小。
小M当然是知道答案的啦,但是你呢?

Input
第一行为一个数字T(T<=10)表示数据组数
第二行为两个数字n、m(1<=n,m<=200000)表示序列长度和询问次数
第三行为n个数字表示原始序列A (0 < A[i] < 1000000000)
接下来m行,每行三个数字l r x 表示询问[l,r]之间小于x的有几个(1<=l<=r<=n,0<=x<=1000000000)
保证数据合法

Output
输出为m行,第i行表示第i个询问的答案

Sample Input
1
10 3
2 3 6 9 8 5 4 7 1 1
1 3 5
2 8 7
3 6 4
Sample Output
2
4
0
Hint
Source
2017年暑期集训校队选拔

Author
卢铭威

题目大意: 求区间[l,r]中小于x的个数。
解题思路:主席树

#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>#include<vector>using namespace std;const int MAXN=2e5+5;int n,m,cnt,root[MAXN],a[MAXN],x,y,k;struct node{    int l,r,sum;}T[MAXN*30];vector<int> v;int getid(int x){    return lower_bound(v.begin(),v.end(),x)-v.begin()+1;}//update(1,n,root[i],root[i-1],getid(a[i]));void update(int l,int r,int &x,int y,int pos){    T[++cnt]=T[y],T[cnt].sum++,x=cnt;    if(l==r) return;    int mid=(l+r)>>1;    if(mid>=pos) update(l,mid,T[x].l,T[y].l,pos);    else update(mid+1,r,T[x].r,T[y].r,pos);}//query(1,n,root[x-1],root[y],k)int query(int l,int r,int x,int y,int k){    if(l==r) return T[y].sum-T[x].sum;    int mid=(l+r)>>1;    int sum=T[T[y].l].sum-T[T[x].l].sum;    if(mid>=k) return query(l,mid,T[x].l,T[y].l,k);    else return sum+query(mid+1,r,T[x].r,T[y].r,k);}int main(){    int T;    scanf("%d",&T);    while(T--)    {        scanf("%d%d",&n,&m);        v.clear();        cnt=0;        memset(root,0,sizeof(root));        for(int i=1;i<=n;i++)        {            scanf("%d",&a[i]);            v.push_back(a[i]);        }        sort(v.begin(),v.end());        v.erase(unique(v.begin(),v.end()),v.end());        int len=v.size();//        for(int i=0;i<len;i++)//        {//            cout<<v[i]<<" ";//        }//        cout<<endl;        for(int i=1;i<=n;i++)        {            update(1,n,root[i],root[i-1],getid(a[i]));        }        while(m--)        {            scanf("%d%d%d",&x,&y,&k);            int pos=getid(k);            //cout<<"pos: "<<pos<<endl;            if(pos>len)            {                printf("%d\n",y-x+1);            }else            {                if(pos==1)                {                    printf("0\n");                }else                {                    pos--;                    printf("%d\n",query(1,n,root[x-1],root[y],pos));                }            }        }    }    return 0;}
原创粉丝点击