#include<bits/stdc++.h>
using namespace std;
#define int int64_t
#define pb push_back
#define pii pair<int, int>
#define st first
#define nd second
#define sz(x) (int)(x).size()
#define all(x) (x).begin(), (x).end()
ostream& operator<<(ostream &os, vector<auto> v) {
os << "{";
for(int i=0;i<sz(v);i++) {
os << v[i];
if(i!=sz(v)-1) os << ", ";
}
os << "}";
return os;
}
vector<int> V[1<<19][4];
int tab[200009];
const int INF = 1e18;
vector<int> f(vector<int> a, vector<int> b) {
vector<int> ret(2*sz(a)-1, -INF);
for(int i=0;i<sz(a);i++) {
for(int j=0;j<sz(b);j++) {
ret[i+j] = max(ret[i+j], a[i]+b[j]);
}
}
// cerr << a << " + " << b << " = " << ret << "\n";
return ret;
}
int32_t main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int n;
cin >> n;
int N = 1;
while(N<n) N*=2;
// cerr << N << "\n";
for(int i=0;i<n;i++) {
cin >> tab[i];
}
for(int i=N;i<2*N;i++) {
for(int j=0;j<4;j++) {
V[i][j].resize(2);
fill(all(V[i][j]), -INF);
}
if(i-N<n) {
// cerr << "here\n";
V[i][3][1] = tab[i-N];
}
V[i][0][0] = 0;
}
for(int i=N-1;i;i--) {
for(int j=0;j<4;j++) {
V[i][j].resize(2*sz(V[2*i][j])-1);
fill(all(V[i][j]), -INF);
}
for(int j=0;j<4;j++) {
for(int k=0;k<4;k++) {
if((j&1)&&(k&2)) continue;
int type = (j&2)+(k&1);
vector<int> v = f(V[i*2][j], V[i*2+1][k]);
for(int a=0;a<sz(v);a++) {
V[i][type][a] = max(V[i][type][a], v[a]);
}
}
}
// cerr << "now: "<< i << ": ";
// for(int j=0;j<4;j++) {
// cerr << V[i][j] << "\n";
// }
}
for(int i=1;i<=(n+1)/2;i++) {
int ans = 0;
for(int j=0;j<4;j++) {
ans = max(ans, V[1][j][i]);
}
cout << ans << "\n";
}
}
Compilation message
candies.cpp:12:41: warning: use of 'auto' in parameter declaration only available with '-fconcepts-ts'
12 | ostream& operator<<(ostream &os, vector<auto> v) {
| ^~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
42 ms |
52060 KB |
Output is correct |
2 |
Correct |
35 ms |
52060 KB |
Output is correct |
3 |
Correct |
34 ms |
52060 KB |
Output is correct |
4 |
Correct |
40 ms |
52124 KB |
Output is correct |
5 |
Correct |
35 ms |
52056 KB |
Output is correct |
6 |
Correct |
34 ms |
52064 KB |
Output is correct |
7 |
Correct |
34 ms |
52056 KB |
Output is correct |
8 |
Correct |
35 ms |
52100 KB |
Output is correct |
9 |
Correct |
34 ms |
52056 KB |
Output is correct |
10 |
Correct |
35 ms |
52056 KB |
Output is correct |
11 |
Correct |
37 ms |
52128 KB |
Output is correct |
12 |
Correct |
34 ms |
52056 KB |
Output is correct |
13 |
Correct |
38 ms |
52108 KB |
Output is correct |
14 |
Correct |
35 ms |
52232 KB |
Output is correct |
15 |
Correct |
34 ms |
52060 KB |
Output is correct |
16 |
Correct |
40 ms |
52316 KB |
Output is correct |
17 |
Correct |
34 ms |
52056 KB |
Output is correct |
18 |
Correct |
34 ms |
52060 KB |
Output is correct |
19 |
Correct |
39 ms |
52064 KB |
Output is correct |
20 |
Correct |
35 ms |
52144 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
42 ms |
52060 KB |
Output is correct |
2 |
Correct |
35 ms |
52060 KB |
Output is correct |
3 |
Correct |
34 ms |
52060 KB |
Output is correct |
4 |
Correct |
40 ms |
52124 KB |
Output is correct |
5 |
Correct |
35 ms |
52056 KB |
Output is correct |
6 |
Correct |
34 ms |
52064 KB |
Output is correct |
7 |
Correct |
34 ms |
52056 KB |
Output is correct |
8 |
Correct |
35 ms |
52100 KB |
Output is correct |
9 |
Correct |
34 ms |
52056 KB |
Output is correct |
10 |
Correct |
35 ms |
52056 KB |
Output is correct |
11 |
Correct |
37 ms |
52128 KB |
Output is correct |
12 |
Correct |
34 ms |
52056 KB |
Output is correct |
13 |
Correct |
38 ms |
52108 KB |
Output is correct |
14 |
Correct |
35 ms |
52232 KB |
Output is correct |
15 |
Correct |
34 ms |
52060 KB |
Output is correct |
16 |
Correct |
40 ms |
52316 KB |
Output is correct |
17 |
Correct |
34 ms |
52056 KB |
Output is correct |
18 |
Correct |
34 ms |
52060 KB |
Output is correct |
19 |
Correct |
39 ms |
52064 KB |
Output is correct |
20 |
Correct |
35 ms |
52144 KB |
Output is correct |
21 |
Execution timed out |
5016 ms |
200168 KB |
Time limit exceeded |
22 |
Halted |
0 ms |
0 KB |
- |