hdu 4417 主席樹
阿新 • • 發佈:2019-01-26
解題思路:將陣列去重離散化,建成一顆線段樹,那麼我們只要去找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; }