COJ1981-小M的魔术表演

小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 <script type="math/tex" id="MathJax-Element-20">x</script>的个数。
解题思路：主席树

#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;
}