hdu4417 Super Mario 线段树端点更新

37 篇文章 0 订阅
先按枪的高低进行排序,再对询问的高度值进行排序。若当前墙比下一次 询问的高度高 就统计答案,否则把墙加入树中。
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <vector>
using namespace std;
#define maxn 110000
struct node
{
    int l,r,h,id;
}ask[maxn];
struct node2
{
    int pos,h;
}block[maxn];
bool cmp(node2 aa,node2 bb)
{
        return aa.h<bb.h;
}
bool cmp2(node aa,node bb)
{
        return aa.h<bb.h;
}
int ans[maxn],tree[maxn<<2];
void pushup(int rt)
{
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
void update(int rt,int l,int r,int x)
{
    if(l==r)
    {
        tree[rt]=1;
        return;
    }
    int m=(l+r)>>1;
    if(x<=m)
    {
        update(rt<<1 , l , m , x);
    }
    else
    {
        update(rt<<1|1 , m+1 , r ,x);
    }
    pushup(rt);
}
int query(int rt,int l,int r,int x,int y)
{
    if(x==l&&r==y)
    {
        return tree[rt];
    }
    int m=(l+r)>>1;
    if(x>m) return query(rt<<1|1 ,m+1,r,x,y);
    else if(y<=m) return query(rt<<1,l,m,x,y);
    else return query(rt<<1 ,l,m,x,m)+query(rt<<1|1,m+1,r,m+1,y);
}

int main()
{
    int cas=0,n,m,t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&block[i].h);
            block[i].pos=i;
        }
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&ask[i].l,&ask[i].r,&ask[i].h);
            ask[i].l++;ask[i].r++;ask[i].id=i;
        }
        sort(block+1,block+1+n,cmp);
        sort(ask+1,ask+1+m,cmp2);
        printf("Case %d:\n",++cas);
        memset(tree,0,sizeof(tree));
        int j=1;
        for(int i = 1;i <= n;)
        {
            if(block[i].h > ask[j].h)
            {
                ans[ask[j].id] = query(1,1,n,ask[j].l,ask[j].r);
                j ++;
                if(j > m)
                    break;
            }
            else
            {
                update(1,1,n,block[i].pos);
                i ++;
            }
        }

        for(;j<=m;j++)
            ans[ask[j].id]=query(1,1,n,ask[j].l,ask[j].r);
        for(int i=1;i<=m;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值