线段树10pts

P1253 扶苏的问题

Fwio_ @ 2023-06-20 11:08:55

蒟蒻刚学线段树来搓这题,调了30min都没调出来,有没有大佬帮帮QwQ

Code:

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 1000010;
int a[N] , n , m;
struct Node{
    int l , r;
    int sum , maxn , add;
}tr[N << 2];
void pushup(int u){
    tr[u].sum += tr[u << 1].sum + tr[u << 1 | 1].sum;
    tr[u].maxn = max(tr[u << 1].maxn , tr[u << 1 | 1].maxn);
}
void pushdown(int u){
    if(tr[u].add){
        tr[u << 1].add += tr[u].add , tr[u << 1].sum += (tr[u << 1].r - tr[u << 1].l + 1) * tr[u].add;
        tr[u << 1 | 1].add += tr[u].add , tr[u << 1 | 1].sum += (tr[u << 1 | 1].r - tr[u << 1 | 1].l + 1) * tr[u].add;
        tr[u].add = 0;
    }
}
void build(int u , int l , int r){
    if(l == r){
        tr[u].l = l , tr[u].r = r;
        tr[u].sum = a[l] , tr[u].maxn = a[l] , tr[u].add = 0;
        return ;
    }
    else{
        tr[u].l = l , tr[u].r = r;
        int mid = l + r >> 1;
        build(u << 1 , l , mid);
        build(u << 1 | 1 , mid + 1 , r);
        pushup(u);
    }
}
void update(int u , int l , int r , int k){
    if(tr[u].l >= l && tr[u].r <= r){
        tr[u].maxn = k;
        tr[u].sum = (tr[u].r - tr[u].l + 1) * k;
        tr[u].add += k;
        return ;
    }
    else{
        pushdown(u);
        int mid = tr[u].l + tr[u].r >> 1;
        if(r <= mid) update(u << 1 , l , r , k);
        else if(l > mid) update(u << 1 | 1 , l , r , k);
        else update(u << 1 , l , mid , k) , update(u << 1 | 1 , mid + 1 , r , k);
        pushup(u);
    }
}
void modify(int u , int l , int r , int x){
    if(tr[u].l >= l && tr[u].r <= r){
        tr[u].sum += (tr[u].r - tr[u].l + 1) * x;
        tr[u].add += x;
        tr[u].maxn += x;
        return ;
    }
    else{
        pushdown(u);
        int mid = tr[u].l + tr[u].r >> 1;
        if(r <= mid) modify(u << 1 , l , r , x);
        else if(l > mid) modify(u << 1 | 1 , l , r , x);
        else modify(u << 1 , l , mid , x) , modify(u << 1 | 1 , mid + 1 , r , x);
        pushup(u);
    }
}
int query(int u , int l , int r){
    if(tr[u].l >= l && tr[u].r <= r) return tr[u].maxn;
    pushdown(u);
    int mid = tr[u].l + tr[u].r >> 1;
    int maxn = 0;
    if(r <= mid) maxn = query(u << 1 , l , r);
    else if(l > mid) maxn = query(u << 1 | 1 , l , r);
    else maxn = max(max(query(u << 1 , l , mid) , maxn) , query(u << 1 | 1 , mid + 1 , r));
    return maxn;
}
int main(){
    scanf("%d%d" , &n , &m);
    for(int i = 1;i <= n;i++) scanf("%d" , &a[i]);
    build(1 , 1 , n);
    while(m--){
        int opt;
        scanf("%d" , &opt);
        if(opt == 1){
            int l , r , x;
            scanf("%d%d%d" , &l , &r , &x);
            update(1 , l , r , x);
        }
        else if(opt == 2){
            int l , r , x;
            scanf("%d%d%d" , &l , &r , &x);
            modify(1 , l , r , x);
        }
        else{
            int l , r;
            scanf("%d%d" , &l , &r);
            printf("%d\n" , query(1 , l , r));
        }
    }
    return 0;
}

by Link_Cut_Y @ 2023-06-20 11:14:36

update 函数内部有问题吧,区间覆盖为什么 add += k 啊?


by Link_Cut_Y @ 2023-06-20 11:15:40

@HVeo 不对啊,你为什么不维护覆盖标记啊


by Fwio_ @ 2023-06-20 11:37:04

@Link_Cut_Y 大佬,可不可以把赋值转化为区间加


by Link_Cut_Y @ 2023-06-20 11:37:32

@HVeo 就在你的结构体里再维护一个 cover 标记,然后 update 函数这样写:

void update(int u , int l , int r , int k){
    if(tr[u].l >= l && tr[u].r <= r){
        tr[u].maxn = k;
        tr[u].sum = (tr[u].r - tr[u].l + 1) * k;
        tr[u].add = 0;
        tr[u].cover = k;
        return ;
    }
    else{
        pushdown(u);
        int mid = tr[u].l + tr[u].r >> 1;
        if(r <= mid) update(u << 1 , l , r , k);
        else if(l > mid) update(u << 1 | 1 , l , r , k);
        else update(u << 1 , l , mid , k) , update(u << 1 | 1 , mid + 1 , r , k);
        pushup(u);
    }
}

by Link_Cut_Y @ 2023-06-20 11:37:57

@HVeo

  1. 我不是大佬

  2. 不可以。


by Fwio_ @ 2023-06-20 11:39:12

@Link_Cut_Y 如果加了一个cover,pushup和pushdown要不要改变


by Link_Cut_Y @ 2023-06-20 11:39:13

@HVeo

说明你对线段树懒标记还是不理解。建议重学。


by Link_Cut_Y @ 2023-06-20 11:39:38

@HVeo pushup 不用吧,pushdown 要改。


by Fwio_ @ 2023-06-20 11:40:35

@Link_Cut_Y en,好的


by Fwio_ @ 2023-06-20 11:41:30

@Link_Cut_Y 复习一遍再来搓这题,谢谢大佬,关注了


| 下一页