【线段树】Picks loves segment tree
2021-02-21 14:51:00 # ACM

题链

关于修改:
$min$修改:直接递归到节点($maxse < val < maxf$),转化为对节点最大值的加减;
$add$修改:划分为对 节点最大值 和 节点其余值 的加减;
$lazy$标记 最大值的加减$lamaxf$ 和 其余值的加减$laoth$
重点在于将 $min$修改 和 $add$修改 变为同一修改

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
#include <bits/stdc++.h>
#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#pragma GCC optimize(2)
using namespace std;
#define LL long long
#define ll long long
#define ULL unsigned long long
#define ls rt<<1
#define rs rt<<1|1
#define MS 500009
#define INF 20000009
#define mod 1000000007
#define Pi acos(-1.0)
#define Pair pair<LL,LL>
#define M 11431471
#define B 231
#define MM 37101101
#define BB 312

LL n,m,k;
LL sum[MS<<2]; // 区间和
LL maxf[MS<<2]; // 区间最大值
LL cntf[MS<<2]; // 区间最大值个数
LL maxse[MS<<2]; // 区间第二大值
LL lamaxf[MS<<2]; // la_区间最大值需加的数
LL laoth[MS<<2]; // la_区间非最大值需加的数

ll read(){
char c=getchar();ll s=0,f=1;
for(;!isdigit(c);c=getchar())if(c=='-')f=-1;
for(;isdigit(c);c=getchar())s=s*10+c-48;
return s*f;
}

void push_up(int rt){
sum[rt] = sum[ls] + sum[rs];
maxf[rt] = max(maxf[ls],maxf[rs]);
if(maxf[ls] == maxf[rs]){
cntf[rt] = cntf[ls] + cntf[rs];
maxse[rt] = max(maxse[ls],maxse[rs]);
}
else if(maxf[ls] > maxf[rs]){
cntf[rt] = cntf[ls];
maxse[rt] = max(maxse[ls],maxf[rs]);
}
else{
cntf[rt] = cntf[rs];
maxse[rt] = max(maxf[ls],maxse[rs]);
}
}

void build(int l,int r,int rt){
lamaxf[rt] = 0;
laoth[rt] = 0;
if(l == r){
LL x;
x = read();
sum[rt] = x;
maxf[rt] = x;
cntf[rt] = 1;
maxse[rt] = -1e18;
return;
}
int m = l+r>>1;
build(l,m,ls);
build(m+1,r,rs);
push_up(rt);
}

void update(int rt,int l,int r,LL k1,LL k2){
sum[rt] += cntf[rt]*k1+(r-l+1-cntf[rt])*k2;
maxf[rt] += k1;
if(maxse[rt] != -1e18) maxse[rt] += k2;
lamaxf[rt] += k1;
laoth[rt] += k2;
}

void push_down(int rt,int l,int r){
LL m = l+r>>1;
LL maxn = max(maxf[ls],maxf[rs]);
if(maxf[ls] == maxn) update(ls,l,m,lamaxf[rt],laoth[rt]);
else update(ls,l,m,laoth[rt],laoth[rt]);
if(maxf[rs] == maxn) update(rs,m+1,r,lamaxf[rt],laoth[rt]);
else update(rs,m+1,r,laoth[rt],laoth[rt]);
lamaxf[rt] = laoth[rt] = 0;
}

void update_add(int L,int R,int l,int r,int rt,LL val){
if(L <= l && r <= R){
update(rt,l,r,val,val);
return;
}
push_down(rt,l,r);
int m = l+r>>1;
if(m >= L) update_add(L,R,l,m,ls,val);
if(m < R) update_add(L,R,m+1,r,rs,val);
push_up(rt);
}

void update_min(int L,int R,int l,int r,int rt,LL val){
if(maxf[rt] <= val) return;
if(L <= l && r <= R && val > maxse[rt]){
update(rt,l,r,val-maxf[rt],0);
return;
}
push_down(rt,l,r);
int m = l+r>>1;
if(m >= L) update_min(L,R,l,m,ls,val);
if(m < R) update_min(L,R,m+1,r,rs,val);
push_up(rt);
}

LL get_sum(int L,int R,int l,int r,int rt){
if(L <= l && r <= R){
return sum[rt];
}
push_down(rt,l,r);
LL ans = 0;
int m = l+r>>1;
if(m >= L) ans += get_sum(L,R,l,m,ls);
if(m < R) ans += get_sum(L,R,m+1,r,rs);
return ans;
}

void solve(){
n = read() ,m = read();
build(1,n,1);
while(m--){
LL op,l,r,val;
op = read();
l = read() ,r = read();
if(op == 1){
val = read();
update_min(l,r,1,n,1,val);
}
else if(op == 2){
val = read();
update_add(l,r,1,n,1,val);
}
else printf("%lld\n",get_sum(l,r,1,n,1));
}
}

int main() {
//ios::sync_with_stdio(false);
LL te;
te = 1;
while(te--){
solve();
}


return 0;
}
Prev
2021-02-21 14:51:00 # ACM
Next