HDU2795(灵活的线段树求区间最大值)

这个题一开始有点傻,用的求和的线段树,多了很多无用的查找步骤,然后超时,然后优化了好久才想起来用区间最大值可以避免很多无用查找层次,然后重写。
经过不断的改写,重写,初步掌握线段树的灵活飘逸的代码风格。
code:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;

#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

const int maxn=200100;
int sumtree[maxn<<2];

void pushUp(int rt)
{
    sumtree[rt]=max(sumtree[rt<<1],sumtree[rt<<1|1]);
}

void build(int w,int l,int r,int rt)
{
    if(l==r)
    {
        sumtree[rt]=w;
        return ;
    }
    int m=(l+r)/2;
    build(w,lson);
    build(w,rson);
    pushUp(rt);
}

void updateandQuery(int w,int l,int r,int rt)
{
    //printf("l=%d-------r=%d\n",l,r);
    if(l==r&&sumtree[rt]>=w)
    {
        sumtree[rt]-=w;
        printf("%d\n",l);
        return;
    }
    int m=(l+r)/2;
    if(w<=sumtree[rt<<1])
    {
        updateandQuery(w,lson);
    }
    else if(w<=sumtree[rt<<1|1])
    {
        updateandQuery(w,rson);
    }
    pushUp(rt);
    //printf("sum[%d]=%d",rt,sumtree[rt]);
}

int main()
{
    int h,w,n,x;
    while(scanf("%d%d%d",&h,&w,&n)!=EOF)
    {
        int na=min(n,h);
        memset(sumtree,0,sizeof(sumtree));
        build(w,1,na,1);
        for(int i=0;i<n;i++)
        {
            scanf("%d",&x);
            if(x>sumtree[1])
                printf("-1\n");
            else
                updateandQuery(x,1,na,1);
        }
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值