hdu2795线段树点修改

原创于 2013-12-20 22:08:29 发布 · 455 阅读

0 ·

CC 4.0 BY-SA版权

数据结构专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种使用C++模板元编程技术来实现查找数组中最大值的方法，通过递归和模板特化，展示了元编程的强大之处。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#define MIN(a,b) a>b?b:a
const int maxn=200005;
int h,w,n;
struct node
{
    int l,r,max;
};
node tnode[maxn*4];
void Build(int o,int x,int y)
{
    tnode[o].l=x;
    tnode[o].r=y;
    tnode[o].max=w;
    if(x<y)
    {
        int mid=(x+y)/2;
        Build(o*2,x,mid);
        Build(o*2+1,mid+1,y);
    }
}
int Query(int o,int v)
{
    if(tnode[o].max<v)
        return -1;
    if(tnode[o].l==tnode[o].r)
        return tnode[o].l;
    if(tnode[o*2].max>=v)
        return Query(o*2,v);
    else
        return Query(o*2+1,v);
}
void Update(int o,int x,int v)
{
    if(tnode[o].l==tnode[o].r)
        tnode[o].max-=v;
    else
    {
        int mid=(tnode[o].l+tnode[o].r)/2;
        if(x<=mid)
            Update(o*2,x,v);
        else
            Update(o*2+1,x,v);
        tnode[o].max=tnode[o*2].max>tnode[o*2+1].max?tnode[o*2].max:tnode[o*2+1].max;
    }
}
int main()
{
    int x;
    while(~scanf("%d %d %d",&h,&w,&n))
    {

        Build(1,1,MIN(n,h));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&x);
            int ans=Query(1,x);
            printf("%d\n",ans);
            if(ans!=-1)
            {
                Update(1,ans,x);
            }
        }
    }
    return 0;
}