hdu 1698 区间更新

HDU 1698




AC code:

//lrl's submission
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;

#define debug 0
#define LL long long
#define ls root << 1, l, mid
#define rs root << 1 | 1, mid + 1, r
#define inf 0x3f3f3f3f
#define M(a, b) memset(a, b, sizeof(a))

const int maxn = 100000 + 5;
int n, m;
struct node
{
    LL v;                                    
    LL la;
} sum[maxn << 2];

void pushDown(int root, int l, int r)                                //下移lazy
{
    if(sum[root].la)
    {
        int mid = (l + r) >> 1;
        sum[root << 1].la = sum[root].la;
        sum[root << 1 | 1].la = sum[root].la;
        sum[root << 1].v = (mid - l + 1) * sum[root].la;            //区间加和, 左孩子,右孩子分开处理
        sum[root << 1 | 1].v = (r - mid) * sum[root].la;
        sum[root].la = 0;
    }
}
void update(int root, int l, int r, int ul, int ur, int add)        //维护
{
    if(ul > r || ur < l)
        return;
    if(ul <= l && ur >= r)
    {
        sum[root].v = (r - l + 1) * add;
        sum[root].la = add;
        return;
    }
    pushDown(root, l, r);                                //lazy标记下移
    int mid = (l + r) >> 1;
    update(ls, ul, ur, add);
    update(rs, ul, ur, add);
    sum[root].v = sum[root << 1].v + sum[root << 1 | 1].v;
}

void build(int root, int l, int r)                                //建树
{
    sum[root].la = 0;
    if(l == r)
    {
       sum[root].v = 1;
        return;
    }
    int mid = (l + r) >> 1;
    build(ls);
    build(rs);
    sum[root].v = sum[root << 1].v + sum[root << 1 | 1].v;
}
LL query(int root, int l, int r, int ql, int qr)                 //查询
{
    if(ql > r || qr < l)
        return 0;                                            //无重合, 返回0
    if(ql <= l && qr >= r)
    {
        return sum[root].v;                                  //查询区间内直接返回节点v
    }
    pushDown(root, l, r);
    int mid = (l + r) >> 1;
    return query(ls, ql, qr) + query(rs, ql, qr);
}

int main()
{
#if debug
    freopen("in.txt", "r", stdin);
#endif // debug
    int a, b, c, t, q = 1;
    scanf("%d", &t);
    while(t--)
    {
        M(sum, 0);
        scanf("%d%d", &n, &m);
        build(1, 1, n);
        while(m--)
        {
            scanf("%d%d%d", &a, &b, &c);
            update(1, 1, n, a, b, c);
        }
        printf("Case %d: The total value of the hook is %I64d.\n", q++, sum[1].v);
        //printf("%I64d\n", sum[1].v);
    }
    return 0;
}


你可能感兴趣的:(tree),HDU)