Just a Hook HDU - 1698 线段树赋值

题目链接:https://vjudge.net/problem/HDU-1698
题意:区间赋值,计算最终结果。
思路:平常用的板子是区间加一个数,这个是区间赋值,不太一样,不能在最终查询的时候在下传,要在区间更新时就不断下传。
比如
1
10
2
1 5 2
5 9 3
a[2]的lazy在输入5 9 3之前没有往下传,输入5 9 3之后计算a[5]时会出错,此时a[11]=3,但a[10]=1.
错误代码
第一遍tle 选择了在每次区间修改的时候下传且下传到底 并且每个case没有清空lazy

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=5e5+5;
struct node
{
    int l,r,sum,lazy;
    node()
    {
        l=r=sum=lazy=0;
    }
} a[maxn];
inline void update(int k)
{
    a[k].sum=a[k*2].sum+a[k*2+1].sum;
}
void build(int k,int l,int r)
{
    a[k].l=l,a[k].r=r;
    if(l==r)
    {
        a[k].sum=1;
        return ;
    }
    int mid=(l+r)/2;
    build(k*2,l,mid);
    build(k*2+1,mid+1,r);
    update(k);
}
void pushdown(int k)
{
    if(a[k].l==a[k].r)
    {
        a[k].lazy=0;
        return ;
    }
    a[k*2].sum=(a[k*2].r-a[k*2].l+1)*a[k].lazy;
    a[k*2+1].sum=(a[k*2+1].r-a[k*2+1].l+1)*a[k].lazy;
    a[k*2].lazy=a[k].lazy;
    a[k*2+1].lazy=a[k].lazy;
    a[k].lazy=0;
    pushdown(k*2);
    pushdown(k*2+1);
}
void changesegment(int k,int l,int r,int x)
{
    if(a[k].l==l&&a[k].r==r)
    {
        a[k].sum=(r-l+1)*x;
        a[k].lazy=x;
        pushdown(k);
        return ;
    }
    int mid=(a[k].l+a[k].r)/2;
    if(r<=mid)
    {
        changesegment(k*2,l,r,x);
    }
    else if(l>mid)
        changesegment(k*2+1,l,r,x);
    else
    {
        changesegment(k*2,l,mid,x);
        changesegment(k*2+1,mid+1,r,x);
    }
    update(k);
}
int query(int k,int l,int r)
{
    if(a[k].lazy)
        pushdown(k);
    if(a[k].l==l&&a[k].r==r)
        return a[k].sum;
    int mid=(a[k].r+a[k].l)/2;
    if(r<=mid)
        return query(k*2,l,r);
    if(l>mid)
        return query(k*2+1,l,r);
    return query(k*2,l,mid)+query(k*2+1,mid+1,r);
}
int main()
{
    int T;
    int kase=0;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        build(1,1,n);
        int q;
        scanf("%d",&q);
        while(q--)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            changesegment(1,a,b,c);
        }
        printf("Case %d: The total value of the hook is %d.\n",++kase,query(1,1,n));
    }
    return 0;
}

正确代码 标记不用下传到底 在区间更新时就要下传标记

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=5e5+5;
struct node
{
    int l,r,sum,lazy;
    node()
    {
        l=r=sum=lazy=0;
    }
} a[maxn];
inline void update(int k)
{
    a[k].sum=a[k*2].sum+a[k*2+1].sum;
}
void build(int k,int l,int r)
{
    a[k].l=l,a[k].r=r;
    a[k].lazy=0;
    if(l==r)
    {
        a[k].sum=1;
        return ;
    }
    int mid=(l+r)/2;
    build(k*2,l,mid);
    build(k*2+1,mid+1,r);
    update(k);
}
void pushdown(int k)
{
    if(a[k].l==a[k].r||!a[k].lazy)
    {
        a[k].lazy=0;
        return ;
    }
    a[k*2].sum=(a[k*2].r-a[k*2].l+1)*a[k].lazy;
    a[k*2+1].sum=(a[k*2+1].r-a[k*2+1].l+1)*a[k].lazy;
    a[k*2].lazy=a[k].lazy;
    a[k*2+1].lazy=a[k].lazy;
    a[k].lazy=0;
}
void changesegment(int k,int l,int r,int x)
{
    if(a[k].l==l&&a[k].r==r)
    {
        a[k].sum=(r-l+1)*x;
        a[k].lazy=x;

        return ;
    }
    pushdown(k);
    int mid=(a[k].l+a[k].r)/2;
    if(r<=mid)
    {
        changesegment(k*2,l,r,x);
    }
    else if(l>mid)
        changesegment(k*2+1,l,r,x);
    else
    {
        changesegment(k*2,l,mid,x);
        changesegment(k*2+1,mid+1,r,x);
    }
    update(k);
}
int query(int k,int l,int r)
{
    if(a[k].lazy)
        pushdown(k);
    if(a[k].l==l&&a[k].r==r)
        return a[k].sum;
    int mid=(a[k].r+a[k].l)/2;
    if(r<=mid)
        return query(k*2,l,r);
    if(l>mid)
        return query(k*2+1,l,r);
    return query(k*2,l,mid)+query(k*2+1,mid+1,r);
}
int main()
{
    int T;
    int kase=0;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        build(1,1,n);
        int q;
        scanf("%d",&q);
        while(q--)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            changesegment(1,a,b,c);
        }
        printf("Case %d: The total value of the hook is %d.\n",++kase,query(1,1,n));
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值