hdu 4417 线段树

题意是给你n个数,然后m组询问,每组为l,r,h,问你在区间[l,r]里有多少个比h小

思路:将n个数从小到大排列,将询问按h从小到大排列,然后按顺序将n个数插入线段树中。具体来说是对于每一个h(从小到大),不停的插入新元素直到当前元素比h大为止,那么可以保证目前线段树上点的值都不大于h,然后就可以区间求和了,复杂度为(nlogn+mlogm)

#include<cstdio>
#include<cstring>
#include<algorithm>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
const int N=1e5+5;
int sum[N<<2],ans[N],vis[N];
struct node
{
    int v,pos;
    bool operator <(const node& b)const
    {
        return v<b.v;
    }
}p[N];
struct qq { int l,r,h,id; bool operator <(const qq& b)const { return h<b.h; }
}q[N];
void PushUp(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void update(int pos,int v,int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=v;
        return;
    }
    int m=(l+r)>>1;
    if(pos<=m)update(pos,v,lson);
    else update(pos,v,rson);
    PushUp(rt);
}
int query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)return sum[rt];
    int m=(l+r)>>1;
    int ans=0;
    if(m>=L)ans+=query(L,R,lson);
    if(m<R)ans+=query(L,R,rson);
    return ans;
}
int main()
{
    int T,n,m,kase=0;
    //freopen("a.txt","r",stdin);
    scanf("%d",&T);
    while(T--)
    {
        printf("Case %d:\n",++kase);
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&p[i].v);p[i].pos=i;
        }
        sort(p+1,p+n+1);
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&q[i].l,&q[i].r,&q[i].h);
            q[i].l++;q[i].r++;
            q[i].id=i;
        }
        sort(q+1,q+m+1);
        memset(sum,0,sizeof(sum));
        int k=1;
        for(int i=1;i<=m;i++)
        {
            while(k<=n&&p[k].v<=q[i].h)
            {
                update(p[k].pos,1,1,n,1);
                k++;
            }
            ans[q[i].id]=query(q[i].l,q[i].r,1,n,1);
        }
        for(int i=1;i<=m;i++)printf("%d\n",ans[i]);
    }
    return 0;
}

你可能感兴趣的:(离线线段树)