#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
#define NL '\n'
#define EL cout << NL
#define FOR(i,n) for(int i = 0; i < (n); i++)
#define FORS(i,s,n) for(int i = (s); i < (n); i++)
#define PRINTV(v) for(auto a:v) {cout << a << " ";} EL;
#define PRINTVV(v) for(auto a:v) {PRINTV(a);}
#define f first
#define s second
#define all(v) (v).begin(),(v).end()
#define STRP(p) "{" << (p).f << "," << (p).s << "}"
#define PRINTVP(v) for(auto a:v) {cout << STRP(a) << " ";} EL;
#define PRINTVVP(v) for(auto a:v) {PRINTVP(a);}
typedef long long ll;
typedef vector<ll> vl;
typedef vector<vl> vvl;
typedef pair<ll,ll> pl;
typedef vector<pl> vpl;
typedef vector<vpl> vvpl;
long long max_weights(int N, int M, std::vector<int> X, std::vector<int> Y,
std::vector<int> W) {
bool even = true;
bool smallx = true;
bool smally = true;
FOR(i,M) {
if (X[i] % 2 == 1) even = false;
if (X[i] > 1) smallx = false;
if (Y[i] != 0) smally = false;
}
if (even) {
ll res = 0;
FOR(i,M) {
res += W[i];
}
return res;
}
if (smallx) {
vl left(N);
vl right(N);
FOR(i,M) {
if (X[i] == 0) left[Y[i]] = W[i];
else right[Y[i]] = W[i];
}
if (N == 2) {
return max(right[0]+right[1],left[0]+left[1]);
}
ll ans = 0;
FOR(i,N) {
ans += right[i];
}
ll psum = ans;
FOR(i,N) {
psum += left[i]-right[i];
ans = max(ans,psum);
}
return ans;
}
if (smally) {
vector<array<array<ll,2>,2>> dp(N);
vl a(N);
FOR(i,M) a[X[i]] = W[i];
dp[0][0] = {0,0};
dp[0][1] = {0,0};
FORS(i,1,N) {
dp[i][0][0] = max(dp[i-1][0][0],dp[i-1][1][0]);
dp[i][1][0] = max(dp[i-1][0][1],dp[i-1][1][1]) + a[i];
dp[i][0][1] = max(dp[i-1][1][0],dp[i-1][0][0] + a[i-1]);
dp[i][1][1] = max(dp[i-1][0][1],dp[i-1][1][1]);
}
return max(dp[N-1][0][0],max(dp[N-1][0][1],max(dp[N-1][1][0],dp[N-1][1][1])));
}
}
Compilation message
fish.cpp: In function 'long long int max_weights(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:78:1: warning: control reaches end of non-void function [-Wreturn-type]
78 | }
| ^
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
23 ms |
2140 KB |
Output is correct |
2 |
Correct |
24 ms |
2644 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
1 ms |
348 KB |
Output is correct |
5 |
Correct |
76 ms |
7252 KB |
Output is correct |
6 |
Correct |
67 ms |
7276 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
37 ms |
5456 KB |
Output is correct |
3 |
Correct |
45 ms |
6484 KB |
Output is correct |
4 |
Correct |
17 ms |
2136 KB |
Output is correct |
5 |
Correct |
21 ms |
2652 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 |
Correct |
0 ms |
348 KB |
Output is correct |
10 |
Correct |
1 ms |
348 KB |
Output is correct |
11 |
Correct |
0 ms |
348 KB |
Output is correct |
12 |
Correct |
18 ms |
3708 KB |
Output is correct |
13 |
Correct |
25 ms |
4188 KB |
Output is correct |
14 |
Correct |
19 ms |
3664 KB |
Output is correct |
15 |
Correct |
21 ms |
3924 KB |
Output is correct |
16 |
Correct |
18 ms |
3676 KB |
Output is correct |
17 |
Correct |
23 ms |
4060 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
344 KB |
Output is correct |
2 |
Correct |
2 ms |
4188 KB |
Output is correct |
3 |
Correct |
13 ms |
5212 KB |
Output is correct |
4 |
Correct |
10 ms |
5212 KB |
Output is correct |
5 |
Correct |
22 ms |
6484 KB |
Output is correct |
6 |
Correct |
19 ms |
6492 KB |
Output is correct |
7 |
Correct |
23 ms |
6484 KB |
Output is correct |
8 |
Correct |
34 ms |
6492 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '2' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '2' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '2' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
344 KB |
Output is correct |
2 |
Correct |
2 ms |
4188 KB |
Output is correct |
3 |
Correct |
13 ms |
5212 KB |
Output is correct |
4 |
Correct |
10 ms |
5212 KB |
Output is correct |
5 |
Correct |
22 ms |
6484 KB |
Output is correct |
6 |
Correct |
19 ms |
6492 KB |
Output is correct |
7 |
Correct |
23 ms |
6484 KB |
Output is correct |
8 |
Correct |
34 ms |
6492 KB |
Output is correct |
9 |
Incorrect |
22 ms |
6484 KB |
1st lines differ - on the 1st token, expected: '99999', found: '66666' |
10 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
23 ms |
2140 KB |
Output is correct |
2 |
Correct |
24 ms |
2644 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
1 ms |
348 KB |
Output is correct |
5 |
Correct |
76 ms |
7252 KB |
Output is correct |
6 |
Correct |
67 ms |
7276 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
37 ms |
5456 KB |
Output is correct |
9 |
Correct |
45 ms |
6484 KB |
Output is correct |
10 |
Correct |
17 ms |
2136 KB |
Output is correct |
11 |
Correct |
21 ms |
2652 KB |
Output is correct |
12 |
Correct |
0 ms |
348 KB |
Output is correct |
13 |
Correct |
0 ms |
348 KB |
Output is correct |
14 |
Correct |
0 ms |
348 KB |
Output is correct |
15 |
Correct |
0 ms |
348 KB |
Output is correct |
16 |
Correct |
1 ms |
348 KB |
Output is correct |
17 |
Correct |
0 ms |
348 KB |
Output is correct |
18 |
Correct |
18 ms |
3708 KB |
Output is correct |
19 |
Correct |
25 ms |
4188 KB |
Output is correct |
20 |
Correct |
19 ms |
3664 KB |
Output is correct |
21 |
Correct |
21 ms |
3924 KB |
Output is correct |
22 |
Correct |
18 ms |
3676 KB |
Output is correct |
23 |
Correct |
23 ms |
4060 KB |
Output is correct |
24 |
Correct |
1 ms |
344 KB |
Output is correct |
25 |
Correct |
2 ms |
4188 KB |
Output is correct |
26 |
Correct |
13 ms |
5212 KB |
Output is correct |
27 |
Correct |
10 ms |
5212 KB |
Output is correct |
28 |
Correct |
22 ms |
6484 KB |
Output is correct |
29 |
Correct |
19 ms |
6492 KB |
Output is correct |
30 |
Correct |
23 ms |
6484 KB |
Output is correct |
31 |
Correct |
34 ms |
6492 KB |
Output is correct |
32 |
Incorrect |
1 ms |
348 KB |
1st lines differ - on the 1st token, expected: '3', found: '2' |
33 |
Halted |
0 ms |
0 KB |
- |