您的位置:首页 > 理论基础 > 数据结构算法

HDU1166 敌兵布阵(线段树)

2016-10-15 16:12 323 查看
题意:

中文题

要点:

就是一个单点更新的线段树问题,之前一直用线状数组,现在学了一下线段树,这两个能处理的问题都差不多,线段树好理解一些,基本就是存储将一个区间不断二分,形成一个树,方便查询和更新。

#include<iostream>
#include<string>
#include<algorithm>
using namespace std;
const int N = 50000 + 5;
int sum[N << 2];

int pushup(int node)
{
return sum[node] = sum[2 * node] + sum[2 * node + 1];
}
void build(int node, int l, int r)
{
if (l == r)
{
scanf("%d",&sum[node]);
return;
}
int m = (l + r) / 2;
build(2 * node, l, m);
build(2 * node + 1, m + 1, r);
pushup(node);
}
void update(int p, int val, int l, int r,int node)
{
if (l == r)
{
sum[node] += val;
return;
}
int m = (l + r)/2;
if (p <= m)
update(p, val, l, m, 2 * node);
else
update(p, val, m + 1, r, 2 * node + 1);
pushup(node);//回溯更新
}
int query(int ll, int rr, int l, int r, int node)
{
if (ll <= l&&rr >= r)
return sum[node];
int m = (l + r) / 2;
int ans = 0;
if (ll <= m)
ans += query(ll, rr, l, m, 2 * node);
if (rr > m)
ans += query(ll, rr, m + 1, r, 2 * node + 1);
return ans;
}

int main()
{
int t;
char order[100];
scanf("%d", &t);
for(int kase=1;kase<=t;kase++)
{
printf("Case %d:\n", kase);
int n;
scanf("%d", &n);
build(1, 1, n);
while (scanf("%s", order) && order[0] != 'E')
{
int x, y;
scanf("%d%d", &x, &y);
if (order[0] == 'A')
{
update(x, y, 1, n, 1);
}
else if (order[0] == 'S')
{
update(x, -y, 1, n, 1);
}
else if (order[0] == 'Q')
{
int ans=query(x, y, 1, n, 1);
printf("%d\n", ans);
}
}
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息