hdu 1166(线段树或者树状数组)

原创 2015年07月09日 16:30:13

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

题意:

给出营地的总数和各个营地初始人数,通过对某个营地 add(增加人数)或sub(减少人数)操作 

求第a个——第b个营地范围内的人数总和


可以采用线段树也可以是树状数组

树状数组

#include <iostream>
#include "stdio.h"
#include "stdlib.h"
#define maxn 50005
using namespace std;
int c[maxn]={0};
int lowbit(int x){return x&(-x);}

void update(int target,int change,int n)
{
    while(target<=n)
    {
         c[target]+=change;
         target+=lowbit(target);
    }
}
int sum(int x)
{
    int ans=0;
    while(x>0)
    {
        ans+=c[x];
        x-=lowbit(x);
    }
    return ans;
}
int main()
{
    int t,deap=0;
    scanf("%d",&t);
    while(t--)
    {
        int n;
        memset(c,0,sizeof(c));
        char temp[10];
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            int tmp;
            scanf("%d",&tmp);
            update(i,tmp,n);
            }
        printf("Case %d:\n",++deap);
        while(scanf("%s",temp)!=EOF)
        {
            int a,b;
            if(temp[0]=='E')  break;
            else if(temp[0]=='Q') {
                scanf("%d%d",&a,&b);
                printf("%d\n",sum(b)-sum(a-1));
            }
            else{
                scanf("%d%d",&a,&b);
                if(temp[0]=='S') b=b*(-1);
                update(a,b,n);
            }
        }
    }
    return 0;
}

线段树:

#include <iostream>
#include "stdio.h"
#include "stdlib.h"
#define maxn 50005
using namespace std;

int sum[maxn<<2]={0};

void build(int k,int l,int r)
{
    if(l==r) {
        sum[k]=0;
        return;
    }
    int mid=(l+r)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
}
void update(int k,int left,int right,int pos,int val)
{
    if(left==right)  {
        sum[k]+=val;
        return;
        }
     int mid=(left+right)>>1;
     if(pos<=mid) update(k<<1,left,mid,pos,val);
     else update(k<<1|1,mid+1,right,pos,val);
     sum[k]=sum[k<<1|1]+sum[k<<1];
}
int query(int k,int left,int right,int ll,int rr)
{
     if(ll<=left&&rr>=right) return sum[k];
     int mid=(left+right)>>1;
     int sum1=0,sum2=0;
     if(ll<=mid)  sum1=query(k<<1,left,mid,ll,rr);
     if(rr>mid)  sum2=query(k<<1|1,mid+1,right,ll,rr);
     return sum1+sum2;
}
int main()
{
    int t,ans=0;
    scanf("%d",&t);
    while(t--)
    {
        int n;
        char  temp[10];
        scanf("%d",&n);
        build(1,1,n);
        for(int i=1;i<=n;i++)
        {
            int tmp;
            scanf("%d",&tmp);
            update(1,1,n,i,tmp);
        }
        printf("Case %d:\n",++ans);
        while(1)
        {
            scanf("%s",temp);
            if(temp[0]=='E')  break;
            else if(temp[0]=='Q') {
                    int a,b;
                    scanf("%d%d",&a,&b);
                    printf("%d\n",query(1,1,n,a,b));
            }
            else{
                    int a,b;
                    scanf("%d%d",&a,&b);
                    if(temp[0]=='S') b=(-1)*b;
                    update (1,1,n,a,b);
                }
            }
        }
        return 0;
}



版权声明:本文为博主原创文章,未经博主允许不得转载。

hdu 1166 树状数组 线段树

敌兵布阵 Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others) Total Sub...

HDU1166 敌兵布阵[线段树/树状数组]

X - 敌兵布阵 Time Limit:1000MS     Memory Limit:32768KB     64bit IO Format:%I64d & %I64u Submit Status ...

HDU 1166 敌兵布阵 线段树/树状数组入门题

很水的一道题。单点更新 线段树代码 先取最小的n使2^n>=所表示范围,则线段树节点个数应为2^(n+1); #include #include int n; struct Tree{ int...

hdu 1166 树状数组 & 线段树

//树状数组 #include "stdafx.h" #include using namespace std; const int MAXN = 50010; int num[MAXN]; in...

hdu 1166 敌兵布阵(线段树OR树状数组)

去年学树状数组的时候用树状数组写过这题,如今学线段树,用线代树写了一次,发现线段树的代码好像更长,程序运行时间也比树状数组久 现在树状数组也忘光光了,,, (┬_┬)。。。 PS: 某渣看到以前的...

hdu1166树状数组||线段树。。

题目链接:自己找去== 题目思路:嗯。。。点修改,区间查询。。

HDU 1166 敌兵布阵【线段树,树状数组入门题,单点更新,区间求和】

敌兵布阵 Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others) Total Submi...

hdu 1166 敌兵布阵(线段树,树状数组)

http://acm.hdu.edu.cn/showproblem.php?pid=1166 直接线段树模板 AC代码: #include #include #include using ...

HDU_1166_敌兵布阵(线段树OR树状数组)

题型:数据结构 题意:中文不多说。 分析: 两种方法:线段树和树状数组。 两者之间的比较:        线段树是一种二叉搜索树,与区间树相似,它将一个区间划分成一些单元区间...

hdu 1166 敌兵布阵(线段树,树状数组)

敌兵布阵 Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others) Total Submi...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:hdu 1166(线段树或者树状数组)
举报原因:
原因补充:

(最多只允许输入30个字)