#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 100010;
int n;
ll s[maxn][5];
ll maxSuffix[maxn];
ll dp[maxn][5][5];
long long max_weights(int N, int M, vector<int> X, vector<int> Y, vector<int> W)
{
n = N;
for(int i = 0 ; i < M ; i++)
{
X[i]++; Y[i]++;
s[ X[i] ][ Y[i] ] += W[i];
}
for(int i = 1 ; i <= n ; i++)
for(int j = 3 ; j >= 0 ; j--)
s[i][j] += s[i][j + 1];
for(int hI = 0 ; hI <= 1 ; hI++)
{
for(int hB = 0 ; hB <= 1 ; hB++)
{
if( hB <= hI )
dp[n][hI][hB] = 0;
else
dp[n][hI][hB] = s[n][hI + 1] - s[n][hB + 1];
}
}
for(int i = n - 1 ; i > 0 ; i--)
{
for(int hI = 0 ; hI <= 1 ; hI++)
{
// Calculating dp[i][hI][0]
for(int x = 0 ; x <= 1 ; x++)
{
if( x <= hI )
dp[i][hI][0] = max( dp[i][hI][0] , dp[i + 1][x][hI] );
else
dp[i][hI][0] = max( dp[i][hI][0] , dp[i + 1][x][hI] + s[i][hI + 1] - s[i][x + 1] );
}
// Calculating dp[i][hI][hB] (hB > 0)
ll maxFirst = dp[i + 1][0][hI];
for(int x = 1 ; x >= 0 ; x--)
maxSuffix[x] = max( maxSuffix[x + 1] , dp[i + 1][x][hI] + s[i][hI + 1] - s[i][x + 1] );
for(int hB = 1 ; hB <= 1 ; hB++)
{
ll& ans = dp[i][hI][hB];
maxFirst = max( maxFirst , dp[i + 1][hB][hI] );
if( hB <= hI )
{
ans = dp[i][hI][0];
continue;
}
ans = max( maxFirst + s[i][hI + 1] - s[i][hB + 1] , maxSuffix[hB + 1] );
}
}
}
ll ans = 0;
for(int i = 0 ; i <= 1 ; i++)
ans = max( ans , dp[1][i][0] );
return ans;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
40 ms |
23284 KB |
1st lines differ - on the 1st token, expected: '40313272768926', found: '6462113953940' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
212 KB |
1st lines differ - on the 1st token, expected: '2', found: '1' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
14 ms |
23764 KB |
Output is correct |
2 |
Correct |
13 ms |
23788 KB |
Output is correct |
3 |
Correct |
27 ms |
22680 KB |
Output is correct |
4 |
Correct |
25 ms |
24708 KB |
Output is correct |
5 |
Correct |
44 ms |
26060 KB |
Output is correct |
6 |
Correct |
46 ms |
26848 KB |
Output is correct |
7 |
Correct |
43 ms |
27372 KB |
Output is correct |
8 |
Correct |
42 ms |
27408 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
212 KB |
1st lines differ - on the 1st token, expected: '3', found: '1' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
212 KB |
1st lines differ - on the 1st token, expected: '3', found: '1' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
212 KB |
1st lines differ - on the 1st token, expected: '3', found: '1' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
14 ms |
23764 KB |
Output is correct |
2 |
Correct |
13 ms |
23788 KB |
Output is correct |
3 |
Correct |
27 ms |
22680 KB |
Output is correct |
4 |
Correct |
25 ms |
24708 KB |
Output is correct |
5 |
Correct |
44 ms |
26060 KB |
Output is correct |
6 |
Correct |
46 ms |
26848 KB |
Output is correct |
7 |
Correct |
43 ms |
27372 KB |
Output is correct |
8 |
Correct |
42 ms |
27408 KB |
Output is correct |
9 |
Runtime error |
28 ms |
5864 KB |
Execution killed with signal 11 |
10 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
40 ms |
23284 KB |
1st lines differ - on the 1st token, expected: '40313272768926', found: '6462113953940' |
2 |
Halted |
0 ms |
0 KB |
- |