Fwio_ @ 2023-06-20 11:08:55
蒟蒻刚学线段树来搓这题,调了30min都没调出来,有没有大佬帮帮QwQ
#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
我不是大佬
不可以。
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 复习一遍再来搓这题,谢谢大佬,关注了