HDU 1166 敌兵布阵（线段树 单点更新）

 点我看题目 

题意 ：HDU的中文题也不常见。。。。这道题我就不详述了。。。。。

思路 ：这个题用线段树用树状数组都可以，用线段树的时候要注意输入那个地方，输入一个字符串的时候不要紧接着输入两个数字，因为我就是这样贡献了好几个RE。。。。

不要用cin，cout，因为也是这样我又贡献了好几个TLE。。。。血一般的教训啊，这道题是水题，模板题。

这个是线段树的代码

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>

using namespace std;

const int maxn =  ;
int a[maxn] ;
int ans ;
struct node
{
    int l,r,value ;
} Node[*maxn] ;

void build(int v ,int l,int r)
{
    Node[v].l = l ;
    Node[v].r = r ;
    if(l == r)
    {
        Node[v].value = a[r] ;
        return ;
    }
    int mid = (l+r)>> ;
    build(v*,l,mid) ;
    build(v*+,mid+,r) ;
    Node[v].value = Node[v*].value+Node[v*+].value ;
}

int query(int v,int l,int r)
{
    if(Node[v].l == l && Node[v].r == r)
        return Node[v].value ;
    int mid = (Node[v].l+Node[v].r) >>  ;
    if(r <= mid)
        return query(v*,l,r) ;
    else
    {
        if(l > mid)
            return query(v*+,l,r) ;
        else
            return query(v*,l,mid)+query(v*+,mid+,r) ;
    }
}

void update(int v,int n,int m)
{
    Node[v].value += m ;
    if(Node[v].l == n &&  Node[v].r == n)
        return  ;
    else
    {
        int mid = (Node[v].l+Node[v].r)>> ;
        if(n <= mid)
        update(*v,n,m) ;
        else if(n > mid)
        update(*v+,n,m) ;
    }
}

void sub(int v,int n,int m)
{
    Node[v].value -= m ;
    if(Node[v].l == n &&  Node[v].r == n)
        return ;
    else
    {
        int mid = (Node[v].l+Node[v].r)>> ;
        if(n <= mid)
        sub(*v,n,m) ;
        else if(n > mid)
        sub(*v+,n,m) ;
    }
}

int main()
{
    int T ,x=;
    scanf("%d",&T) ;
    while(T--)
    {
        int n ;
        ans = ;
        scanf("%d",&n) ;
        for(int i =  ; i <= n ; i++)
        scanf("%d",&a[i]) ;
        build(,,n) ;
        printf("Case %d:\n",x++) ;
        char ch[] ;
        while(true)
        {
            scanf("%s",ch) ;
            int a,b ;
            if(ch[] == 'Q')
            {
                scanf("%d %d",&a,&b) ;
                printf("%d\n",query(,a,b)) ;
            }
            else if(ch[] == 'A'){
                scanf("%d %d",&a,&b) ;
            update(,a,b) ;
            }
            else if(ch[] == 'S'){
                scanf("%d %d",&a,&b) ;
            sub(,a,b) ;
            }
            if(ch[] == 'E')
            break ;
        }
    //    printf("\n") ;
    }
    return ;
}

这个是树状数组的

#include <stdio.h>
#include <iostream>
#include <string.h>

using namespace std ;

const int maxn =  ;
int ch[maxn] ;
int data ;
int n ;

int lowbit(int i)
{
    return i&(-i) ;
}
int sum(int i)
{
    int ans =  ;
    while(i > )
    {
        ans += ch[i] ;
        i -= lowbit(i) ;
    }
    return ans ;
}
void add(int x,int value)
{
    while(x <= n)
    {
        ch[x] += value ;
        x += lowbit(x) ;
    }
}
int main()
{
    int T ;
    scanf("%d",&T) ;
    for(int i =  ; i <= T ; i++)
    {
        scanf("%d",&n) ;
        memset(ch,,sizeof(ch)) ;
        for(int j =  ; j <= n ; j++)
        {
            scanf("%d",&data) ;
            add(j,data) ;
        }
        printf("Case %d:\n",i) ;
        char sh[] ;
        int a,b ;
        while(~scanf("%s",sh))
        {
            if(sh[] == 'E')
                break ;
            if(sh[] == 'A')
            {
                scanf("%d %d",&a,&b) ;
                add(a,b) ;
            }
            if(sh[] == 'Q')
            {
                scanf("%d %d",&a,&b) ;
                printf("%d\n",sum(b)-sum(a-)) ;
            }
            if(sh[] == 'S')
            {
                scanf("%d %d",&a,&b);
                add(a,-b) ;
            }
        }
    }
    return  ;
}

当时初学时代码太乱，跟崔老师学习了一点，现在重新刷了一遍

 #include <cstdio>
 #include <string>
 #include <cstring>
 #include <iostream>

 using namespace std ;

 int p[*],a ;
 //string sh ;
 char sh[] ;
 //void pushdown(int rt, int m)
 //{
 //
 //}
 void pushup(int rt)
 {
     p[rt] = p[rt << ] + p[rt <<  | ] ;
 }
 void build(int l,int r,int rt)
 {
     if(l == r)
     {
         cin >> a ;
         p[rt] = a ;
         return  ;
     }
     int mid = (l + r) >>  ;
     build(l,mid,rt << ) ;
     build(mid +  , r , rt <<  | ) ;
     pushup(rt) ;
 }
 void update(int l , int r,int sc,int rt,int flag,int x)
 {
     if(l == r)
     {
         if(flag > )
         p[rt] += sc ;
         else p[rt] -= sc ;
         return ;
     }
     int mid = (l+r) >>  ;
     if(mid >= x)
         update(l,mid,sc,rt << ,flag,x) ;
     if(mid < x)
         update(mid+,r,sc,rt <<  |  ,flag,x) ;
     pushup(rt) ;
 }
 int query(int L,int R,int l,int r,int rt)
 {
     int sum = ;
     if(l >= L && r <= R)
     {
         return p[rt] ;
     }
     int mid = (l+r) >>  ;
     if(mid >= L)
         sum += query(L,R,l,mid,rt << ) ;
     if(mid < R)
         sum += query(L,R,mid+,r,rt <<  | ) ;
     return sum ;
 }
 int main()
 {
     int T,N ,x,y;
 //    cin >> T ;
     scanf("%d",&T) ;
     for(int i =  ; i <= T ; i++)
     {
         scanf("%d",&N) ;
         build(,N,) ;
         printf("Case %d:\n",i) ;
         while(~scanf("%s",sh))
         {
             if(sh[] == 'E')
                 break ;
 //            cin >> x >> y ;
 scanf("%d %d",&x,&y) ;
             if(sh[] == 'Q')
             {
                 int ans = query(x,y,,N,) ;
 //                cout << ans << endl ;
 printf("%d\n",ans) ;
             }
             else if(sh[] == 'A')
             {
                 update(,N,y,,,x) ;
             }
             else if(sh[] == 'S')
             {
                 update(,N,y,,-,x) ;
             }
         }
     }
     return ;
 }

推荐文章

线段树总结