HDU4417

题目大意

给定一个数列,每次要求你查询区间[L,R]内不超过K的数的数量

题解

和静态的区间第K大差不多,这题是<=K,先建立好n颗主席树,然后用第R颗主席树区间[1,K]内数的数量减去第L-1颗主席树区间[1,K]内数的数量就得到了查询的结果

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define MAXN 100005
#define lson l,m,ls[s]
#define rson m+1,r,rs[s]
int ls[20*MAXN],rs[20*MAXN];
int cnt[20*MAXN],T[MAXN];
int a[2*MAXN],num[MAXN],tot;
int L[MAXN],R[MAXN],K[MAXN];
void build(int l,int r,int &s)
{
    s=++tot;
    cnt[s]=0;
    if(l==r) return;
    int m=(l+r)>>1;
    build(lson);
    build(rson);
}
void update(int last,int p,int l,int r,int &s)
{
    s=++tot;
    ls[s]=ls[last],rs[s]=rs[last];
    cnt[s]=cnt[last]+1;
    if(l==r) return;
    int m=(l+r)>>1;
    if(p<=m) update(ls[last],p,lson);
    else
        update(rs[last],p,rson);
}
int query(int ss,int tt,int ql,int qr,int l,int r)
{
    if(ql<=l&&r<=qr)
        return cnt[tt]-cnt[ss];
    int m=(l+r)>>1,ans=0;
    if(ql<=m) ans+=query(ls[ss],ls[tt],ql,qr,l,m);
    if(qr>m) ans+=query(rs[ss],rs[tt],ql,qr,m+1,r);
    return ans;
}
int main()
{
    int kase,p=0;
    scanf("%d",&kase);
    while(kase--)
    {
        int n,m,len=0;
        memset(cnt,0,sizeof(cnt));
        scanf("%d%d",&n,&m);
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&num[i]);
            a[++len]=num[i];
        }
        for(int i=1; i<=m; i++)
        {
            scanf("%d%d%d",&L[i],&R[i],&K[i]);
            a[++len]=K[i];
        }
        sort(a+1,a+len+1);
        len=unique(a+1,a+len+1)-a-1;
        build(1,len,T[0]);
        for(int i=1; i<=n; i++) num[i]=lower_bound(a+1,a+len+1,num[i])-a;
        for(int i=1; i<=n; i++) update(T[i-1],num[i],1,len,T[i]);
        printf("Case %d:
",++p);
        for(int i=1; i<=m; i++)
        {
            int k=lower_bound(a+1,a+len+1,K[i])-a;
            printf("%d
",query(T[L[i]],T[R[i]+1],1,k,1,len));
        }
    }
    return 0;
}
原文地址:https://www.cnblogs.com/zjbztianya/p/3350060.html