hdu 1698 Just a Hook（线段树的区间更新及求和）

Just a Hook

题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=1698

题目大意：给你一组数据，要你更新某一区间的值或者求某一区间值的和，很轻松想到线段树的区间更新和求和。

题目思路：可以先弄懂线段树的概念，然后根据样例去理解代码，一步一步走，我当时就是这样理解的，下面会有两个代码，一个是超时的，但是这个代码更能让我们去理解线段树的概念；第二个是AC代码，也很容易理解的。

超时代码：（因为把每个结点都更新了，所以会超时）

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define N 100010
struct Node{
    int L;
    int R;
    int V;
}tree[4*N];
void BuildTree(int root,int l,int r){
    tree[root].L=l;
    tree[root].R=r;
    if(l==r){
        tree[root].V=1;
        return ;
    }
    int mid=(l+r)/2;
    BuildTree(2*root,l,mid);
    BuildTree(2*root+1,mid+1,r);
    tree[root].V=tree[2*root].V+tree[2*root+1].V;
}
int update(int root,int s,int e,int a){
    if(tree[root].L==tree[root].R){
        return tree[root].V=a;
    }
    int mid=(tree[root].L+tree[root].R)/2;
    if(e<=mid) return tree[root].V=update(2*root,s,e,a)+tree[2*root+1].V;
    else if(s>mid) return tree[root].V=update(2*root+1,s,e,a)+tree[2*root].V;
    else    return tree[root].V=update(2*root,s,mid,a)+update(2*root+1,mid+1,e,a);
}
int main(){
    int T,ca=1;
    scanf("%d",&T);
    while(T--){
        int n;
        scanf("%d",&n);
        BuildTree(1,1,n);
        int m;
        scanf("%d",&m);
        while(m--){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            update(1,a,b,c);
        }
        printf("Case %d: The total value of the hook is %d.\n",ca++,tree[1].V);
    }
    return 0;
}

AC代码（详解）：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define N 100010
struct Node{
    int L;
    int R;
    int V;
}tree[4*N];
void BuildTree(int root,int l,int r){
    tree[root].L=l;
    tree[root].R=r;
    tree[root].V=1;///表示颜色，该颜色默认价值为1
    if(l==r){
        return ;
    }
    int mid=(l+r)/2;
    BuildTree(2*root,l,mid);
    BuildTree(2*root+1,mid+1,r);
}
void update(int root,int s,int e,int a){
    if(tree[root].V==a){///区间内的颜色与要更新的一样，不用更新
        return ;
    }
    if(tree[root].L==s&&tree[root].R==e){///区间相同，直接更新
        tree[root].V=a;
        return ;
    }
    if(tree[root].V!=-1){///如果区间的颜色相同，但是与即将跟新的颜色不一样，则此区间就要变成杂色
        tree[2*root].V=tree[2*root+1].V=tree[root].V;
        ///先将区间的颜色更新到子结点
        ///因为每次更新只更新到父节点，如果第二组数据要用到子节点时，就应该先将父节点的颜色给子节点，再标记
        tree[root].V=-1;///区间此时不只有一种颜色了，标记为-1
    }
    int mid=(tree[root].L+tree[root].R)/2;
    if(e<=mid) update(2*root,s,e,a);
    else if(s>mid) update(2*root+1,s,e,a);
    else {
        update(2*root,s,mid,a);
        update(2*root+1,mid+1,e,a);
    }
}
int Query(int root,int s,int e){
    if(tree[root].V!=-1){///区间都为同一颜色
        return (e-s+1)*tree[root].V;
    }
    else{
        int mid=(tree[root].L+tree[root].R)/2;
        if(e<=mid) return Query(2*root,s,e);
        else if(s>mid) return Query(2*root+1,s,e);
        else {
            return Query(2*root,s,mid)+Query(2*root+1,mid+1,e);
        }
    }
}
int main(){
    int T,ca=1;
    scanf("%d",&T);
    while(T--){
        int n;
        scanf("%d",&n);
        BuildTree(1,1,n);
        int m;
        scanf("%d",&m);
        while(m--){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            update(1,a,b,c);
        }
        printf("Case %d: The total value of the hook is %d.\n",ca++,Query(1,1,n));
    }
    return 0;
}