1. 程式人生 > >hdu 4417 主席樹

hdu 4417 主席樹

解題思路:將陣列去重離散化,建成一顆線段樹,那麼我們只要去找H的排名就可以用第r顆樹減去第l顆樹的(0-H)的區間數就可以了

#include<bits/stdc++.h>
#define lson l,mid
#define rson mid+1,r
using namespace std;
typedef long long ll;
int n,m;
const int mx =1e5+10;
int num[mx],root[mx],so[mx];
int ls[20*mx],rs[20*mx],sum[20*mx],size;
void update(int x,int &y,int l,int r,int M,int v)
{
    y = ++size;
    ls[y] = ls[x],rs[y] = rs[x],sum[y] = sum[x] + v;
    int mid = (l+r)>>1;
    if(l==r) return ;
    if(M<=mid) update(ls[x],ls[y],lson,M,v);
    else update(rs[x],rs[y],rson,M,v);
}
int query(int x,int l,int r,int M)
{
    if(r<=M) return sum[x];
    int mid = (l+r)>>1;
    int ans = query(ls[x],lson,M);
    if(M>mid) ans += query(rs[x],rson,M);
    return ans;
}
int main()
{
    int t,cas = 1;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++){
            scanf("%d",num+i);
            so[i] = num[i];
        }
        size = 0;
        memset(sum,0,sizeof(sum));
        sort(so+1,so+1+n);
        int k = 1,L,R,H;
        for(int i=2;i<=n;i++)
        if(so[i]!=so[i-1]) so[++k] = so[i];
        for(int i=1;i<=n;i++){
            int p = lower_bound(so+1,so+1+k,num[i])-so;
            update(root[i-1],root[i],1,k,p,1);
        }
        printf("Case %d:\n",cas++);
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&L,&R,&H);
            int p = upper_bound(so+1,so+1+k,H)-so;
            p--;
            if(!p) puts("0");
            else printf("%d\n",query(root[R+1],1,k,p)-query(root[L],1,k,p));
        }
    }
    return 0;
}