HDU 1166 敌兵布阵 (线段树 单点更新)

题目链接

线段树掌握的很差,打算从头从最简单的开始刷一波, 嗯。。就从这个题开始吧!

 #include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <cstdlib>
#include <algorithm>
const int maxn = +;
using namespace std;
int a[maxn], n;
struct line
{
int l, r, val;
}tr[maxn<<]; void build(int o, int l, int r)
{
tr[o].l = l; tr[o].r = r;
if(l == r)
{
tr[o].val = a[l];
return;
}
int mid = (l+r)>>;
build(*o, l, mid);
build(*o+, mid+, r);
tr[o].val = tr[*o].val+tr[*o+].val;
}
int query(int o, int l, int r)
{
if(tr[o].l==l && tr[o].r==r)
return tr[o].val;
int mid = (tr[o].l+tr[o].r)>>;
if(r <= mid) query(*o, l, r);
else if(l > mid) query(*o+, l, r);
else
return (query(*o, l, mid)+query(*o+, mid+, r));
}
void update(int o, int p, int add)
{
if(tr[o].l==tr[o].r&&tr[o].l==p)
{
tr[o].val += add;
return;
}
int mid = (tr[o].l+tr[o].r)>>;
if(p<=mid) update(*o, p, add);
else update(*o+, p, add);
tr[o].val = tr[*o].val+tr[*o+].val;
}
int main()
{
int t, i, ca = ;
int p, add, l, r;
char s[];
scanf("%d", &t);
while(t--)
{
scanf("%d", &n);
for(i = ; i <= n; i++)
scanf("%d", &a[i]);
printf("Case %d:\n", ca++); build(, , n);
while(~scanf("%s", s))
{
if(strcmp(s, "End")==) break;
if(s[]=='Q')
{
scanf("%d%d", &l, &r);
printf("%d\n", query(, l, r));
}
if(s[]=='A')
{
scanf("%d%d", &p, &add);
update(, p, add);
}
if(s[]=='S')
{
scanf("%d%d", &p, &add);
update(, p, -add);
}
}
}
return ;
}

注释的代码:

 #include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <cstdlib>
#include <algorithm>
const int maxn = +;
using namespace std;
int a[maxn], n;
struct line
{
int l, r, val; //val表示该区间的和
}tr[maxn<<]; void build(int o, int l, int r) //o代表当前节点编号
{
tr[o].l = l; tr[o].r = r;
if(l == r)
{
tr[o].val = a[l];
return;
}
int mid = (l+r)>>;
build(*o, l, mid);
build(*o+, mid+, r);
tr[o].val = tr[*o].val+tr[*o+].val; //建树把值从下往上加起来
}
int query(int o, int l, int r) //求l到r的和
{
if(tr[o].l==l && tr[o].r==r) //节点的区间吻合返回
return tr[o].val;
int mid = (tr[o].l+tr[o].r)>>;
if(r <= mid) query(*o, l, r);
else if(l > mid) query(*o+, l, r);
else
return (query(*o, l, mid)+query(*o+, mid+, r)); //横跨了区间
}
void update(int o, int p, int add) //对p节点增加add
{
if(tr[o].l==tr[o].r&&tr[o].l==p)
{
tr[o].val += add;
return;
}
int mid = (tr[o].l+tr[o].r)>>;
if(p<=mid) update(*o, p, add);
else update(*o+, p, add);
tr[o].val = tr[*o].val+tr[*o+].val; //找到值以后的更新
}
int main()
{
int t, i, ca = ;
int p, add, l, r;
char s[];
scanf("%d", &t);
while(t--)
{
scanf("%d", &n);
for(i = ; i <= n; i++)
scanf("%d", &a[i]);
printf("Case %d:\n", ca++); build(, , n);
while(~scanf("%s", s))
{
if(strcmp(s, "End")==) break;
if(s[]=='Q')
{
scanf("%d%d", &l, &r);
printf("%d\n", query(, l, r));
}
if(s[]=='A')
{
scanf("%d%d", &p, &add);
update(, p, add);
}
if(s[]=='S')
{
scanf("%d%d", &p, &add);
update(, p, -add);
}
}
}
return ;
}
上一篇:docker + nginx 部署vuejs3.0项目


下一篇:android:android:background="#00000000",透明效果