HDU 1698 & UESTC 1228 Just a hook

算是线段树中的一道水题了，必须用到懒操作，否则会超时。或者也可以刚开始不计算和，只更新节点，最后算整个线段的颜色和。

1.懒操作法

/* 908ms  3448KB  in HDU OJ*/
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <cstdlib>
using namespace std;
#define N 100011

struct node
{
    int sum;
    int mark;
}tree[*N];

int n,q;

void build(int l,int r,int rt)
{
    if(l == r)
    {
        tree[rt].sum = ;
        tree[rt].mark = ;
        return;
    }
    int mid = (l+r)/;
    build(l,mid,*rt);
    build(mid+,r,*rt+);
    tree[rt].sum = tree[*rt].sum + tree[*rt+].sum;
}

void update(int len,int rt)
{
    if(!tree[rt].mark)
        return;
    tree[*rt].mark = tree[*rt+].mark = tree[rt].mark;
    tree[*rt].sum = tree[*rt].mark*(len - len/);
    tree[*rt+].sum = tree[*rt+].mark*(len/);
    tree[rt].mark = ;
}

void change(int l,int r,int aa,int bb,int flag,int rt)
{
    if(aa<=l&&bb>=r)   //不用更新到底部，只要更新到区间
    {
        tree[rt].sum = flag*(r-l+);
        tree[rt].mark = flag;
        return;
    }
    update(r-l+,rt);
    int mid = (l+r)/;
    if(aa<=mid)
        change(l,mid,aa,bb,flag,*rt);
    if(bb>mid)
        change(mid+,r,aa,bb,flag,*rt+);
    tree[rt].sum = tree[*rt].sum + tree[*rt+].sum;
}

int main()
{
    int t,i;
    int cs = ;
    int aa,bb,val;
    scanf("%d",&t);
    while(t--)
    {
        memset(tree,,sizeof(tree));
        scanf("%d%d",&n,&q);
        build(,n,);
        for(i=;i<q;i++)
        {
            scanf("%d%d%d",&aa,&bb,&val);
            change(,n,aa,bb,val,);
        }
        printf("Case %d: The total value of the hook is %d.\n",cs++,tree[].sum);
    }
    return ;
}

2.最后求和法

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
using namespace std;
#define N 100011

struct node
{
    int le,ri;
    int sum;
}tree[*N];

int n,q;

void build(int l,int r,int rt)
{

    tree[rt].sum = ;
    tree[rt].le = l;
    tree[rt].ri = r;
    if(l == r)
    {
        return;
    }
    int mid = (l+r)/;
    build(l,mid,*rt);
    build(mid+,r,*rt+);
}

void change(int l,int r,int aa,int bb,int flag,int rt)
{
    if(aa<=l&&bb>=r)   //不用更新到底部，只要更新到区间
    {
        tree[rt].sum = flag;
        return;
    }
    if(tree[rt].sum != )
    {
        tree[*rt].sum = tree[*rt+].sum = tree[rt].sum;
        tree[rt].sum = ;
    }
    int mid = (l+r)/;
    if(aa<=mid)
        change(l,mid,aa,bb,flag,*rt);
    if(bb>mid)
        change(mid+,r,aa,bb,flag,*rt+);
}

int query(int rt)
{
    if(tree[rt].sum != )
        return (tree[rt].ri - tree[rt].le + )*tree[rt].sum;
    return query(*rt)+query(*rt+);
}

int main()
{
    int t,i;
    int cs = ;
    int aa,bb,val;
    scanf("%d",&t);
    while(t--)
    {
        memset(tree,,sizeof(tree));
        scanf("%d%d",&n,&q);
        build(,n,);
        for(i=;i<q;i++)
        {
            scanf("%d%d%d",&aa,&bb,&val);
            change(,n,aa,bb,val,);
        }
        printf("Case %d: The total value of the hook is %d.\n",cs++,query());
    }
    return ;
}