码迷,mamicode.com
首页 > 其他好文 > 详细

hdu1166敌兵布阵&&hdu1754I Hate It(线段树入门)

时间:2014-08-10 12:47:40      阅读:202      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   color   os   io   ar   div   

单点更新是最最基础的线段树,只更新叶子节点,然后把信息用pushup这个函数更新上来。

http://acm.hdu.edu.cn/showproblem.php?pid=1166

update单点更新,query区域求和。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#define N 200001
using namespace std;
struct node
{
    int l,r;
    __int64 w;
} q[4*N];
void pushup(int rt)
{
    q[rt].w=q[rt*2].w+q[rt*2+1].w;
}
void build(int l,int r,int rt)
{
    q[rt].l=l;
    q[rt].r=r;
    q[rt].w=0;
    if(l==r)
    {
        scanf("%I64d",&q[rt].w);
        return ;
    }
    build(l,(l+r)/2,rt*2);
    build((l+r)/2+1,r,rt*2+1);
    pushup(rt);
}
void update(int sum,int key,int l,int r,int rt)
{
    if(sum<l||sum>r)
        return ;
    if(l==r&&sum==l)
    {
        q[rt].w+=(__int64)key;
        return ;
    }
    update(sum,key,l,(l+r)/2,rt*2);
    update(sum,key,(l+r)/2+1,r,rt*2+1);
    pushup(rt);
}
__int64 query(int ll,int rr,int l,int r,int rt)
{
    if(ll>r||rr<l) return 0;
    if(ll<=l&&rr>=r)
    {
        return q[rt].w;
    }
    return query(ll,rr,l,(l+r)/2,rt*2)+query(ll,rr,(l+r)/2+1,r,rt*2+1);
}
int main()
{
    int n,sum1,sum2,T,K=0;
    char a[10];
    scanf("%d",&T);
    while(T--)
    {
        K++;
        scanf("%d",&n);
        build(1,n,1);
        printf("Case %d:\n",K);
        while(scanf("%s",a)!=EOF)
        {
            if(strcmp(a,"End")==0) break;
            else if(strcmp(a,"Query")==0)
            {
                scanf("%d%d",&sum1,&sum2);
                __int64 t=query(sum1,sum2,1,n,1);
                printf("%I64d\n",t);
            }
            else if(strcmp(a,"Sub")==0)
            {
                scanf("%d%d",&sum1,&sum2);
                update(sum1,-sum2,1,n,1);

            }
            else if(strcmp(a,"Add")==0)
            {
                scanf("%d%d",&sum1,&sum2);
                update(sum1,sum2,1,n,1);
            }
        }

    }
    return 0;
}

 

 

http://acm.hdu.edu.cn/showproblem.php?pid=1754

update单点替换,query区间最值

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#define N 200001
using namespace std;
struct node
{
    int l,r;
    __int64 w;
}q[4*N];
void pushup(int rt)
{
    q[rt].w=max(q[rt*2].w,q[rt*2+1].w);
}
void build(int l,int r,int rt)
{
    q[rt].l=l;
    q[rt].r=r;
    q[rt].w=0;
    if(l==r)
    {
        scanf("%I64d",&q[rt].w);
        return ;
    }
    build(l,(l+r)/2,rt*2);
    build((l+r)/2+1,r,rt*2+1);
    pushup(rt);
}
void update(int sum,int key,int l,int r,int rt)
{
    if(sum<l||sum>r)
        return ;
    if(l==r&&sum==l)
    {
        q[rt].w=(__int64)key;
        return ;
    }
    update(sum,key,l,(l+r)/2,rt*2);
    update(sum,key,(l+r)/2+1,r,rt*2+1);
    pushup(rt);
}
__int64 query(int ll,int rr,int l,int r,int rt)
{
    if(ll>r||rr<l) return 0;
    if(ll<=l&&rr>=r)
    {
        return q[rt].w;
    }
    return max(query(ll,rr,l,(l+r)/2,rt*2),query(ll,rr,(l+r)/2+1,r,rt*2+1));
}
int main()
{
    int n,m,sum1,sum2;
    char a[3];
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        build(1,n,1);
        while(m--)
        {
            scanf("%s%d%d",a,&sum1,&sum2);
            if(a[0]==U)
            {
                update(sum1,sum2,1,n,1);
            }
            else if(a[0]==Q)
            {
                __int64 t=query(sum1,sum2,1,n,1);
                printf("%I64d\n",t);
            }
        }
    }
    return 0;
}

 

hdu1166敌兵布阵&&hdu1754I Hate It(线段树入门),布布扣,bubuko.com

hdu1166敌兵布阵&&hdu1754I Hate It(线段树入门)

标签:style   blog   http   color   os   io   ar   div   

原文地址:http://www.cnblogs.com/zhangmingcheng/p/3902525.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!