hdu1698Just a Hook(线段树+懒惰标记)

这是一道成段更新的典型例题。成段更新需要用到懒惰标记。简单来说就是每次更新的时候先不更新到底,用懒惰标记使更新延迟到下一次要更新or询问时。(P.S. 大数据输出和读取不要用cin,cout,今天手残,没事做,tle,tle,。。。)

#include 
#include 
#include 
#include 
#include 
using namespace std;
#define ms(X) memset(X,0,sizeof(X));
typedef long long ll;
int tree[120005<<2],cnt[120005<<2],N,Q;
void pushdown(int tn,int m)
{
    if(cnt[tn]&&m>1){
        tree[tn<<1]=(m-(m>>1))*cnt[tn];
        tree[tn<<1|1]=(m>>1)*cnt[tn];
        cnt[tn<<1]=cnt[tn<<1|1]=cnt[tn];
        cnt[tn]=0;
    }
}
void build(int tn,int l,int r)
{
    tree[tn]=r-l+1;
    if(l==r)    return;
    int mid=(l+r)>>1;
    build(tn<<1,l,mid);
    build(tn<<1|1,mid+1,r);
}
void chang(int tn,int l,int r,int a,int b,int z)
{
    if(a<=l&&b>=r)  {cnt[tn]=z;tree[tn]=(r-l+1)*z;return;}
    pushdown(tn,r-l+1);
    int mid=(l+r)>>1;
    /*if(a<=mid) chang(tn<<1,l,mid,a,b,z);
    if(b>mid) chang(tn<<1|1,mid+1,r,a,b,z);*/
    if(b<=mid) chang(tn<<1,l,mid,a,b,z);
    else if(a>mid) chang(tn<<1|1,mid+1,r,a,b,z);
    else {
        chang(tn<<1,l,mid,a,mid,z);
        chang(tn<<1|1,mid+1,r,mid+1,b,z);
    }
    tree[tn]=tree[tn<<1]+tree[tn<<1|1];
}
int main(int argc, char const *argv[])
{
    int t;
    cin>>t;
    for(int ti=1;ti<=t;ti++)
    {
        ms(cnt);
        scanf("%d%d",&N,&Q);
        build(1,1,N);
        int a,b,z;
        while(Q--){
            scanf("%d%d%d",&a,&b,&z);
        chang(1,1,N,a,b,z);}
        cout<<"Case "<": The total value of the hook is "<1]<<'.'<return 0;
}

你可能感兴趣的:(acm之路,线段树)