HDU 1166 敌兵布阵

来源：互联网发布：js所有方法编辑：程序博客网时间：2024/05/21 06:53

刚学习线段树，先写一个最裸的单点更新。

听大大们说线段树是一种非常经典而又灵活的数据结构。一步步学起吧。

这里先写个静态的版本。模板是模仿某位大大的。。

#include <cstdio>#define lson rt<<1,L,M#define rson rt<<1|1,M + 1,Rusing namespace std;const int maxn = 50001;int sum[maxn << 2],ql,qr,val[maxn],v,tar;inline void pushup(int rt) {　　//更新当前节点    sum[rt] = sum[rt<<1] + sum[rt<<1|1];}void build(int rt,int L,int R) {    //建树    if(L == R) sum[rt] = val[L];    else {        int M = (L + R) >> 1;        build(lson); build(rson);        pushup(rt);    }}void update(int rt,int L,int R) {　　//更新点    if(L == R) sum[rt] += v;    else {        int M = (L + R) >> 1;        if(tar <= M) update(lson);        if(tar > M) update(rson);        pushup(rt);    }}int query(int rt,int L,int R) {　　//处理查询    if(ql <= L && qr >= R) return sum[rt];    int M = (L + R) >> 1,ans = 0;    if(ql <= M) ans += query(lson);    if(qr > M) ans += query(rson);    return ans;}int main() {    int c,N;    char cmd[10];    scanf("%d",&c);    for(int k = 1;k <= c;k++) {        printf("Case %d:\n",k);        scanf("%d",&N);        for(int i = 1;i <= N;i++) scanf("%d",val + i);        build(1,1,N);        while(scanf("%s",cmd),cmd[0] != 'E') {            if(cmd[0] == 'Q') {                scanf("%d%d",&ql,&qr);                printf("%d\n",query(1,1,N));            } else {                scanf("%d%d",&tar,&v);                if(cmd[0] == 'S') v = -v;                update(1,1,N);            }        }    }    return 0;}

0 0