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

简介:

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

分析:第一次接触到线段树,上网看了一些资料,发现这道题可以用树状数组来代替线段树解决,似乎用起来更方便,于是乎就找了树状数组的资料来看,这里有一个是我从网上摘抄的树状数组的资料 树状数组 不是很齐全,但是初学者已经够用了。里边有基本的模板代码。

这道题是中文题目,题意很简单,就是典型的树状数组,线段树题目。

树状数组解法

复制代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define MAXN 50005
using namespace std;

int a[MAXN],c[MAXN];

int N;

// 树状数组模板
int Low_bit(int x)
{
    return x & (-x);
}
void modify(int n,int value)
{
    while(n<=N)
    {
        c[n]+=value;
        n += Low_bit(n);
    }
}
int sum(int n)
{
    int sum=0;
    while(n>0)
    {
        sum+=c[n];
        n-=Low_bit(n);
    }
    return sum;
}

int query(int i,int j)
{
    return sum(j)-sum(i-1);
}

int main()
{
    int t,x,y;
    scanf("%d",&t);
    char order[10];
    for(int i=1;i<=t;i++)
    {
       printf("Case %d:\n",i);
       scanf("%d",&N);
       memset(a,0,sizeof(a));
       memset(c,0,sizeof(c));
       for(int j=1;j<=N;j++)
       {
           scanf("%d",&a[j]);
           modify(j,a[j]);
       }
       while(scanf("%s",order)&&strcmp(order,"End")!=0)
       {
            scanf("%d%d",&x,&y);
            if(strcmp(order,"Query")==0)
            printf("%d\n",query(x,y));
            else if(strcmp(order,"Add")==0)
            modify(x,y);
            else if(strcmp(order,"Sub")==0)
            modify(x,-y);
       }
    }
    return 0;
}

线段树解法:
复制代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define MAXN 50005
using namespace std;

typedef struct Node
{
    int ld,rd;//左右边界
    struct Node *lc,*rc;//左右子树
    int value;
}Node,Tree;


int num[MAXN];
int N;

Tree *create(int a,int b)
{
    int mid;
    Node *node = (Node*)malloc(sizeof(Node));
    node->ld=a;
    node->rd=b;
    if(b-a>=1)
    {
        mid=(a+b)/2;
        node->lc=create(a,mid);
        node->rc=create(mid+1,b);
        node->value=node->lc->value+node->rc->value;
    }
    else node->value=num[node->ld];//也可以写成 node->value=a[node->rd];
    return node;
}

int query(Node *node,int x,int y)
{
    if(node->ld==x && node->rd==y)return node->value;
    int mid = (node->ld+node->rd)/2;
    if(mid>=y)return query(node->lc,x,y);
    else if(mid<x) return query(node->rc,x,y);
    else return query(node->lc,x,mid)+query(node->rc,mid+1,y);
}

void modify(Node *node,int x,int v)
{
    if(node->ld==node->rd)
    {
        node->value+=v;
        return;
    }
    if(node->lc->rd>=x)modify(node->lc,x,v);
    else if(node->rc->ld<=x)modify(node->rc,x,v);
    node->value = node->lc->value+node->rc->value;

}

int main()
{
    int t,x,y,value;
    Node *root;
    scanf("%d",&t);
    char order[10];
    for(int i=1;i<=t;i++)
    {
       printf("Case %d:\n",i);
       scanf("%d",&N);
       for(int j=1;j<=N;j++)
       scanf("%d",&num[j]);
       root = create(1,N);

       while(scanf("%s",order)&&strcmp(order,"End")!=0)
       {
            scanf("%d%d",&x,&y);
            if(strcmp(order,"Query")==0)
            printf("%d\n",query(root,x,y));
            else if(strcmp(order,"Add")==0)
            modify(root,x,y);
            else if(strcmp(order,"Sub")==0)
            modify(root,x,-y);
       }
    }
    return 0;
}
复制代码
 
 

 

 
复制代码
本文转自NewPanderKing51CTO博客,原文链接: http://www.cnblogs.com/newpanderking/archive/2012/11/17/2775235.html  ,如需转载请自行联系原作者
相关文章
|
2月前
lanqiao OJ 182 小朋友崇拜圈
lanqiao OJ 182 小朋友崇拜圈
32 2
PTA猴子选大王(约瑟夫环问题)
PTA猴子选大王(约瑟夫环问题)
132 1
|
算法 测试技术 Python
第十二届蓝桥杯《杨辉三角》-二分法
第十二届蓝桥杯《杨辉三角》-二分法
101 0
HDU-致命武器(线段树)
HDU-致命武器(线段树)
92 0
|
人工智能 Java
HDU-敌兵布阵(线段树 || 树状数组)
HDU-敌兵布阵(线段树 || 树状数组)
96 0
UPC-探险(线段树+二分)
UPC-探险(线段树+二分)
94 0
|
机器学习/深度学习 算法 搜索推荐
洛谷每日三题之第六天
洛谷每日三题之第六天
|
测试技术
HDU-4508,湫湫系列故事——减肥记I(完全背包)
HDU-4508,湫湫系列故事——减肥记I(完全背包)