-
Notifications
You must be signed in to change notification settings - Fork 184
/
Copy pathpshtrg.cc
89 lines (85 loc) · 1.84 KB
/
pshtrg.cc
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
// https://www.codechef.com/MARCH18A/problems/PSHTRG
#include <algorithm>
#include <cmath>
#include <iostream>
#include <vector>
using namespace std;
typedef long long ll;
typedef vector<ll> vi;
typedef vector<vi> vvi;
int main() {
int n, q;
cin >> n >> q;
vi a(n);
for (int i = 0; i < n; i++) cin >> a[i];
int k = sqrt(n);
int m = k * k < n ? (n + k - 1) / k : k;
vvi b(m);
for (int i = 0; i < m; i++) {
vi c;
for (int j = 0; j < k; j++) {
if (i * k + j == n) break;
c.push_back(a[i * k + j]);
}
sort(c.begin(), c.end(), greater<ll>());
int j = 0;
while (j < min(int(c.size()), 3)) b[i].push_back(c[j++]);
while (j < c.size()) {
if (c[j-1] + c[j-2] > c[j-3]) break;
b[i].push_back(c[j++]);
}
}
while (q--) {
int t, l, r;
cin >> t >> l >> r;
l--;
if (t == 1) {
int o = a[l];
int i = l / k;
a[l] = r;
if (o < b[i].back() && r < b[i].back()) continue;
vi c;
for (int j = 0; j < k; j++) {
if (i * k + j == n) break;
c.push_back(a[i * k + j]);
}
sort(c.begin(), c.end(), greater<ll>());
b[i].clear();
int j = 0;
while (j < min(int(c.size()), 3)) b[i].push_back(c[j++]);
while (j < c.size()) {
if (c[j-1] + c[j-2] > c[j-3]) break;
b[i].push_back(c[j++]);
}
} else {
r--;
vi c;
int i = l / k;
int ii = r / k;
if ((r + 1) % k != 0) {
if (ii > i)
for (int j = ii * k; j <= r; j++) c.push_back(a[j]);
ii--;
}
if (l % k != 0 || i > ii) {
i++;
for (int j = l; j < min(i * k, r + 1); j++) c.push_back(a[j]);
}
for (int j = i; j <= ii; j++)
for (auto z : b[j]) c.push_back(z);
sort(c.begin(), c.end(), greater<ll>());
ll z = 0;
if (c.size() >= 3) {
int j = 2;
while (j < c.size()) {
if (c[j-1] + c[j] > c[j-2]) {
z = c[j-2] + c[j-1] + c[j];
break;
}
j++;
}
}
cout << z << endl;
}
}
}