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

hdu1166 敌兵布阵 线段树 点更新

2016-08-18 23:29 441 查看
链接:点击打开链接

[align=left]Input[/align]
第一行一个整数T,表示有T组数据。

每组数据第一行一个正整数N(N<=50000),表示敌人有N个工兵营地,接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人(1<=ai<=50)。

接下来每行有一条命令,命令有4种形式:

(1) Add i j,i和j为正整数,表示第i个营地增加j个人(j不超过30)

(2)Sub i j ,i和j为正整数,表示第i个营地减少j个人(j不超过30);

(3)Query i j ,i和j为正整数,i<=j,表示询问第i到第j个营地的总人数;

(4)End 表示结束,这条命令在每组数据最后出现;

每组数据最多有40000条命令

 

[align=left]Output[/align]
对第i组数据,首先输出“Case i:”和回车,

对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。

 

[align=left]Sample Input[/align]

1
10
1 2 3 4 5 6 7 8 9 10
Query 1 3
Add 3 6
Query 2 7
Sub 10 2
Add 6 3
Query 3 10
End

 

[align=left]Sample Output[/align]

Case 1:
6
33
59

思路:非常典型的线段树点更新,为什么不能用数组表示每个兵营呢?因为虽然最多只有50000个,但最多进行40000条输出,会超时。

这个题是求区段总和,所以不需要求出区段最大值,而且更新时加减操作,模板需要稍稍改变一下;

吐槽:(并没有ac,因为杭电这个题数组卡的太严啦交了好几次都是越界,实在没脾气了......大概改成c可以过?

(哪天心情平静的时候再战!

终于知道哪里错了!!!实力懵逼!!!

代码:

#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<cmath>
#include<algorithm>
#include<string>
#include<string.h>
#include<set>
#include<queue>
#include<stack>
#include<vector>
#include<functional>
#include<map>
using namespace std;
const int mxn = 50000 + 10;

struct node {
int l, r, maxn, sum;//存储最大值与和
}tree[mxn<<2];

int t, n;
char str[10];
int x, y;
int a[mxn];

void build(int m, int l, int r) {
tree[m].l = l;
tree[m].r = r;
if (l == r) {
tree[m].sum = a[l];
return;
}
int mid = (l + r) >> 1;
build(m << 1, l, mid);
build((m << 1) + 1, mid + 1, r);
//tree[m].maxn = max(tree[m << 1].maxn, tree[(m << 1) + 1].maxn);
tree[m].sum = tree[m << 1].sum + tree[(m << 1) + 1].sum;
}

void update(int m, int a, int val) {
if (tree[m].l == a && tree[m].r == a) {
//tree[m].maxn += val;
tree[m].sum += val;
return;
}
int mid = (tree[m].l + tree[m].r) >> 1;
if (a <= mid)
update(m << 1, a, val);
else
update((m << 1) + 1, a, val);
//tree[m].maxn = max(tree[m << 1].maxn, tree[(m << 1) + 1].maxn);
tree[m].sum = tree[m << 1].sum + tree[(m << 1) + 1].sum;
}
int query_sum(int m, int l, int r) {
if (l == tree[m].l && r == tree[m].r)
return tree[m].sum;
// return tree[m].maxn;
int mid = (tree[m].l + tree[m].r) >> 1;
if (r <= mid)
return query_sum(m << 1, l, r);
if (l > mid)
return query_sum((m << 1) + 1, l, r);
return query_sum(m << 1, l, mid) + query_sum((m << 1) + 1, mid + 1, r);
//return max(query_max(m << 1, l, mid), query_max((m << 1) + 1, mid + 1, r));
}

int main()
{
int k = 0;
scanf("%d", &t);
while (t--) {	//居然,,,是这里,,,错了
scanf("%d", &n);
for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
printf("Case %d:\n", ++k);
build(1, 1, n);
while (scanf("%s", str)) {
if (str[0] == 'E')   break;
if (str[0] == 'A') {
scanf("%d %d", &x, &y);
update(1, x, y);
}
if (str[0] == 'S') {
scanf("%d %d", &x, &y);
update(1, x, -y);
}
if (str[0] == 'Q') {
scanf("%d %d", &x, &y);
printf("%d\n", query_sum(1, x, y));
}
}
}
//system("pause");
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  acm hdu 数据结构