Just a Hook (HDU 1698) 懒惰标记

Just a Hook (HDU 1698)

题链

每一次都将一个区间整体进行修改，需要用到懒惰标记，懒惰标记的核心在于在查询前才更新，比如将当前点rt标记为col[rt],那么此点的左孩子和右孩子标记必然和其一致(直接替换，如果是累积则另当别论)，同时这个区间也能很快求出了
线段树功能：区间更新+区间查询

#include <cstdio>
#include <utility>
#include <queue>
#include <cstring>
#define scan(x) scanf("%d",&x)
#define scan2(x,y) scanf("%d%d",&x,&y)
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
#define root 1,1,n
using namespace std;
const int Max=1e5+10;
int sum[Max<<2],col[Max<<2];
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]=1;
        col[rt]=0;
        return;
    }
    int mid=(l+r)>>1;
    Build(lson);
    Build(rson);
    Pushup(rt);
}
void Pushdown(int rt,int rage)
{
    if(col[rt])
    {
        col[rt<<1]=col[rt<<1|1]=col[rt];
        sum[rt<<1]=col[rt]*(rage-(rage>>1));
        sum[rt<<1|1]=col[rt]*(rage>>1);
        col[rt]=0;
    }
}
void Update(int L,int R,int x,int rt,int l,int r)
{
    if(L<=l&&r<=R)
    {
        sum[rt]=x*(r-l+1);
        col[rt]=x;
        return;
    }
    Pushdown(rt,(r-l)+1);
    int mid=(l+r)>>1;
    if(L<=mid) Update(L,R,x,lson);
    if(mid<R)  Update(L,R,x,rson);
    Pushup(rt);
}
int main()
{
    int T,ca=1;
    for(scan(T);T;T--)
    {
        memset(sum,0,sizeof(sum));
        memset(col,0,sizeof(col));
        int l,r,k,n,m;
        scan2(n,m);
        Build(root);
        while(m--)
        {
            scanf("%d%d%d",&l,&r,&k);
            Update(l,r,k,root);
        }
        printf("Case %d: The total value of the hook is %d.\n",ca++,sum[1]);
    }
    return 0;
}

重新写了一次，学习了一份新的模板

#include<queue>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define ll long long
#define inf 1000000000
#define mod 1000000007
using namespace std;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,m;
int a[100005];
int ls[400005],rs[400005];
int M[400005],tag[400005];
void build(int k,int l,int r)
{
    int mid=(l+r)>>1;
    ls[k]=l;rs[k]=r;tag[k]=0;M[k]=0;
    if(l==r){M[k]=1;return;}
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    M[k]=M[k<<1]+M[k<<1|1];
}
void pushdown(int k)
{
    if(!tag[k]||ls[k]==rs[k])return;
    tag[k<<1]=tag[k];
    tag[k<<1|1]=tag[k];
    int rage=(rs[k]-ls[k]+1);
    M[k<<1]=(rage-(rage>>1))*tag[k];
    M[k<<1|1]=(rage>>1)*tag[k];
    tag[k]=0;
}
void add(int k,int x,int y,int v)
{
    pushdown(k);
    int l=ls[k],r=rs[k],mid=(l+r)>>1;
    if(x==l&&y==r)
    {
        tag[k]=v;
        M[k]=v*(r-l+1);
        return;
    }
    if(x<=mid) add(k<<1,x,min(y,mid),v);
    if(y>mid) add(k<<1|1,max(x,mid+1),y,v);
    M[k]=M[k<<1]+M[k<<1|1];
}
int query(int k,int x,int y)
{
    pushdown(k);
    int l=ls[k],r=rs[k],mid=(l+r)>>1,ans=0;
    if(x==l&&y==r) return M[k];
    if(x<=mid)ans+=query(k<<1,x,min(y,mid));
    if(y>mid)ans+=query(k<<1|1,max(x,mid+1),y);
    return ans;
}
int main()
{
    int T,ca=1;
    for(T=read();T;T--)
    {
        int l,r,v,n,m;
        n=read();m=read();
        build(1,1,n);
        while(m--)
        {
            scanf("%d%d%d",&l,&r,&v);
            add(1,l,r,v);
        }
        printf("Case %d: The total value of the hook is %d.\n",ca++,query(1,1,n));
    }
    return 0;
}