#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
ll max_weights(int N, int M, vector<int> X, vector<int> Y, vector<int> W) {
int mask = 0, mx = 0;
for(int i=0; i<M; i++) mask |= (1 << (X[i] % 2));
for(int i=0; i<M; i++) mx = max(mx, Y[i]);
if(mask == 1) {
ll ans = 0;
for(int &x : W) ans += x;
return ans;
}
if(mx == 0) {
vector<ll> val(N+5), dp(N+5);
for(int i=0; i<M; i++) val[X[i]] = W[i];
ll ans = 0;
vector<ll> MX(N+5); MX[0] = val[1];
for(int i=1; i<N; i++) {
dp[i] = max(dp[i-1], val[i-1]);
if(i >= 3) dp[i] = max(dp[i], MX[i-3] + val[i-1]);
if(i >= 2) dp[i] = max(dp[i], dp[i-2] + val[i-1]);
MX[i] = max(MX[i-1], dp[i] + val[i+1]);
}
for(int i=0; i<N; i++) ans = max(ans, dp[i] + val[i+1]);
return ans;
}
if(mx <= 8) {
ll val[N+5][mx+5], dp[N+5][mx+5][mx+5];
memset(val, 0, sizeof(val)); memset(dp, 0, sizeof(dp));
for(int i=0; i<M; i++) val[X[i]][Y[i]] = W[i];
for(int i=0; i<N; i++)
for(int j=1; j<=mx; j++) val[i][j] += val[i][j-1];
for(int j=0; j<=mx+1; j++) {
for(int k=0; k<=mx+1; k++) {
if(j > k) dp[1][j][k] = val[0][j-1] - (k ? val[0][k-1] : 0);
else if(j < k) dp[1][j][k] = val[1][k-1] - (j ? val[1][j-1] : 0);
}
}
for(int i=2; i<N; i++) {
for(int j=0; j<=mx+1; j++) {
for(int k=0; k<=mx+1; k++) {
if(j > k) dp[i][j][k] = (j ? val[i-1][j-1] : 0) - (k ? val[i-1][k-1] : 0);
if(j < k) dp[i][j][k] = (k ? val[i][k-1] : 0) - (j ? val[i][j-1] : 0);
if(k) dp[i][j][k] += val[i-2][k-1];
for(int l=0; l<=mx+1; l++) {
if(l <= k && k <= j) {
dp[i][j][k] = max(dp[i][j][k], dp[i-1][k][l] + (j ? val[i-1][j-1] : 0) - (k ? val[i-1][k-1] : 0));
}
if(l >= k && k >= j) {
dp[i][j][k] = max(dp[i][j][k], dp[i-1][k][l] + (k ? val[i][k-1] : 0) - (j ? val[i][j-1] : 0));
}
if(l <= k && k >= j) {
dp[i][j][k] = max(dp[i][j][k], dp[i-1][k][l] + (k ? val[i][k-1] : 0) - (j ? val[i][j-1] : 0));
}
}
}
}
}
ll ans = 0;
// // cout << dp[3][3][2] << '\n';
// for(int i=0; i<N; i++) {
// for(int j=0; j<=mx+1; j++) {
// for(int k=0; k<=mx+1; k++) {
// cout << i << " " << j << " " << k << ": " << dp[i][j][k] << '\n';
// }
// }
// }
for(int i=0; i<N; i++)
for(int j=0; j<=mx+1; j++)
for(int k=0; k<=mx+1; k++) ans = max(ans, dp[i][j][k] + val[i+1][j]);
return ans;
}
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
11 ms |
2136 KB |
Output is correct |
2 |
Correct |
14 ms |
2652 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
46 ms |
7268 KB |
Output is correct |
6 |
Correct |
44 ms |
7280 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Incorrect |
25 ms |
6808 KB |
1st lines differ - on the 1st token, expected: '40604614618209', found: '0' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
344 KB |
Output is correct |
2 |
Correct |
1 ms |
2652 KB |
Output is correct |
3 |
Correct |
9 ms |
3836 KB |
Output is correct |
4 |
Correct |
7 ms |
3436 KB |
Output is correct |
5 |
Correct |
15 ms |
4948 KB |
Output is correct |
6 |
Correct |
13 ms |
4956 KB |
Output is correct |
7 |
Correct |
15 ms |
4956 KB |
Output is correct |
8 |
Correct |
17 ms |
4936 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Incorrect |
1 ms |
348 KB |
1st lines differ - on the 1st token, expected: '216624184325', found: '215402477511' |
10 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Incorrect |
1 ms |
348 KB |
1st lines differ - on the 1st token, expected: '216624184325', found: '215402477511' |
10 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Incorrect |
1 ms |
348 KB |
1st lines differ - on the 1st token, expected: '216624184325', found: '215402477511' |
10 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
344 KB |
Output is correct |
2 |
Correct |
1 ms |
2652 KB |
Output is correct |
3 |
Correct |
9 ms |
3836 KB |
Output is correct |
4 |
Correct |
7 ms |
3436 KB |
Output is correct |
5 |
Correct |
15 ms |
4948 KB |
Output is correct |
6 |
Correct |
13 ms |
4956 KB |
Output is correct |
7 |
Correct |
15 ms |
4956 KB |
Output is correct |
8 |
Correct |
17 ms |
4936 KB |
Output is correct |
9 |
Incorrect |
12 ms |
2648 KB |
1st lines differ - on the 1st token, expected: '99999', found: '0' |
10 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
11 ms |
2136 KB |
Output is correct |
2 |
Correct |
14 ms |
2652 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
46 ms |
7268 KB |
Output is correct |
6 |
Correct |
44 ms |
7280 KB |
Output is correct |
7 |
Correct |
0 ms |
344 KB |
Output is correct |
8 |
Incorrect |
25 ms |
6808 KB |
1st lines differ - on the 1st token, expected: '40604614618209', found: '0' |
9 |
Halted |
0 ms |
0 KB |
- |