Super Mario HDU - 4417 (主席树询问区间比k小的个数）

Mario is world-famous plumber. His “burly” figure and amazing jumping ability reminded in our memory. Now the poor princess is in trouble again and Mario needs to save his lover. We regard the road to the boss’s castle as a line (the length is n), on every integer point i there is a brick on height hi. Now the question is how many bricks in [L, R] Mario can hit if the maximal height he can jump is H.

InputThe first line follows an integer T, the number of test data.

For each test data:

The first line contains two integers n, m (1 <= n <=10^5, 1
<= m <= 10^5), n is the length of the road, m is the number of
queries.

Next line contains n integers, the height of each brick, the range is [0, 1000000000].

Next m lines, each line contains three integers L, R,H.( 0 <= L <= R < n 0 <= H <= 1000000000.)OutputFor each case, output "Case X: " (X is the case number
starting from 1) followed by m lines, each line contains an integer. The
ith integer is the number of bricks Mario can hit for the ith query.

Sample Input

1

10 10

0 5 2 7 5 4 3 8 7 7

2 8 6

3 5 0

1 3 1

1 9 4

0 1 0

3 5 5

5 5 1

4 6 3

1 5 7

5 7 3

Sample Output

Case 1:

4

0

0

3

1

2

0

1

5

1
题解：和查询区间第k小差不多，只不过修改了下询问函数，主要坑点就是特判下特殊情况。

#include<iostream>

#include<cstring>

#include<string>

#include<stdio.h>

#include<algorithm>

using namespace std;

const int maxn=;

int lson[maxn*],rson[maxn*],sum[maxn*],root[maxn],a[maxn];

int cnt,lsh[maxn];

void build(int &rt,int l,int r)

{

    rt=++cnt;

    if(l==r)return ;

    int mid=(l+r)/;

    build(lson[rt],l,mid);

    build(rson[rt],mid+,r);

}

int update(int last,int l,int r,int pos)

{

    int now=++cnt;

    lson[now]=lson[last],rson[now]=rson[last],sum[now]=sum[last]+;

    if(l==r)return now;

    int mid=(l+r)/;

    if(pos<=mid)lson[now]=update(lson[now],l,mid,pos);

    else rson[now]=update(rson[now],mid+,r,pos);

    return now;

}

int querysum(int L,int R,int l,int r,int pos)

{

    int mid=(l+r)/;

    int ans=;

    if(l==r){

        return sum[R]-sum[L];

    }

    if(pos<=mid){

        ans+=querysum(lson[L],lson[R],l,mid,pos);

    }

    else{

        ans+=sum[lson[R]]-sum[lson[L]];//左区间的全部符合题意，直接加上

        ans+=querysum(rson[L],rson[R],mid+,r,pos);

    }

    return ans;

}

int main()

{

    ios::sync_with_stdio();

    int T,top=;

    cin>>T;

    while(T--){

        memset(lson,,sizeof(lson));

        memset(rson,,sizeof(rson));

        memset(sum,,sizeof(rson));

        memset(root,,sizeof(root));

        cnt=;

        int n,m;

        cin>>n>>m;

        for(int i=;i<=n;i++){

            cin>>a[i];

            lsh[i]=a[i];

        }

        sort(lsh+,lsh++n);

        int len=unique(lsh+,lsh++n)-lsh-;

        build(root[],,len);

        for(int i=;i<=n;i++){

            int pos=lower_bound(lsh+,lsh++len,a[i])-lsh;

            root[i]=update(root[i-],,len,pos);

        }

        cout<<"Case "<<top++<<":"<<endl;

        for(int i=;i<=m;i++){

            int L,R,k;

            cin>>L>>R>>k;

            L++,R++;

            int pos=upper_bound(lsh+,lsh++len,k)-lsh-;

            if(pos==){//如果pos=0,不能进去querysum函数，否则会得到错误的结果

                cout<<<<endl;

                continue;

            }

            cout<<querysum(root[L-],root[R],,len,pos)<<endl;

        }

    }

    return ;

}

巴特西

Super Mario HDU - 4417 (主席树询问区间比k小的个数）

最新文章

热门文章