hdu 4417 线段树

最新推荐文章于 2019-07-13 22:14:00 发布

buptbh

最新推荐文章于 2019-07-13 22:14:00 发布

阅读量664

点赞数

分类专栏：数据结构文章标签：离线线段树

本文链接：https://blog.csdn.net/buptbh/article/details/50750217

版权

数据结构专栏收录该内容

2 篇文章 0 订阅

订阅专栏

本文介绍了一种使用线段树解决特定区间查询问题的方法。通过预先对输入数据进行排序，并按照数值大小顺序构建线段树，可以高效地处理一系列询问，即计算指定区间内小于给定阈值的元素数量。此算法的时间复杂度为O((n+m)log n)，适用于处理大量区间查询。

摘要由CSDN通过智能技术生成

题意是给你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;
}