#include "fish.h"
//#include "grader.cpp"
#include <bits/stdc++.h>
using namespace std;
#define int long long
map<pair<int, pair<int, int>>, int> memo;
vector<vector<int>> values;
int n;
int mini;
int dp(int empl, int take, int pretake) {
if (empl>=n) return 0;
if (memo.count({empl, {take, pretake}})) return memo[{empl, {take, pretake}}];
int ans=dp(empl+1, 0, take);
int sum=0;
for (int i=0; i<mini; i++) {
int temp=dp(empl+1, i+1, take);
if (pretake<i+1 && take<i+1 && empl>0) {
sum+=values[empl-1][i];
}
if (empl+1<n) sum+=values[empl+1][i];
if (take>=i+1) sum-=values[empl][i];
ans=max(ans, temp+sum);
}
return memo[{empl, {take, pretake}}]=ans;
}
#undef int
long long max_weights(int N, int m, vector<int> x, vector<int> y, vector<int> w) {
#define int long long
n=N;
bool tacha=true, tachb=true, tachc=true;
for (int i = 0; i < m; ++i)
{
if (x[i]%2) tacha=false;
if (x[i]>1) tachb=false;
if (y[i]) tachc=false;
}
if (tacha)
{
int ans=0;
for (int i = 0; i < m; ++i) ans+=w[i];
return ans;
}
if (tachb)
{
vector<vector<int>> farm(2, vector<int> (n, 0));
for (int i=0; i<m; i++) {
farm[x[i]][y[i]]=w[i];
}
vector<int> suml, sumr;
suml.push_back(farm[0][0]);
sumr.push_back(farm[1][0]);
for (int i=1; i<n; i++) {
suml.push_back(suml.back()+farm[0][i]);
sumr.push_back(sumr.back()+farm[1][i]);
}
if (n==2) return max(suml.back(), sumr.back());
int ans=sumr.back();
for (int i=0; i<n; i++) {
ans=max(ans, suml[i]+sumr.back()-sumr[i]);
}
return ans;
}
for (int i=0; i<n; i++) {
mini=max(mini, (long long) y[i]);
}
values.resize(n, vector<int>(mini, 0));
for (int i=0; i<m; i++) {
values[x[i]][y[i]]=w[i];
}
return dp(0, 0, 0);
#undef int
}
Compilation message
fish.cpp: In function 'long long int max_weights(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:34:31: warning: variable 'tachc' set but not used [-Wunused-but-set-variable]
34 | bool tacha=true, tachb=true, tachc=true;
| ^~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
17 ms |
2140 KB |
Output is correct |
2 |
Correct |
24 ms |
2652 KB |
Output is correct |
3 |
Correct |
0 ms |
600 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
65 ms |
7272 KB |
Output is correct |
6 |
Correct |
66 ms |
7288 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Correct |
38 ms |
8044 KB |
Output is correct |
3 |
Correct |
46 ms |
9536 KB |
Output is correct |
4 |
Correct |
17 ms |
2140 KB |
Output is correct |
5 |
Correct |
21 ms |
2640 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 |
0 ms |
344 KB |
Output is correct |
11 |
Correct |
0 ms |
348 KB |
Output is correct |
12 |
Correct |
19 ms |
6044 KB |
Output is correct |
13 |
Correct |
23 ms |
7240 KB |
Output is correct |
14 |
Correct |
21 ms |
6068 KB |
Output is correct |
15 |
Correct |
22 ms |
6988 KB |
Output is correct |
16 |
Correct |
20 ms |
6040 KB |
Output is correct |
17 |
Correct |
27 ms |
6980 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Runtime error |
1 ms |
348 KB |
Execution killed with signal 11 |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
0 ms |
344 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 |
0 ms |
344 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 |
0 ms |
344 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 |
0 ms |
348 KB |
Output is correct |
2 |
Runtime error |
1 ms |
348 KB |
Execution killed with signal 11 |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
17 ms |
2140 KB |
Output is correct |
2 |
Correct |
24 ms |
2652 KB |
Output is correct |
3 |
Correct |
0 ms |
600 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
65 ms |
7272 KB |
Output is correct |
6 |
Correct |
66 ms |
7288 KB |
Output is correct |
7 |
Correct |
0 ms |
344 KB |
Output is correct |
8 |
Correct |
38 ms |
8044 KB |
Output is correct |
9 |
Correct |
46 ms |
9536 KB |
Output is correct |
10 |
Correct |
17 ms |
2140 KB |
Output is correct |
11 |
Correct |
21 ms |
2640 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 |
0 ms |
344 KB |
Output is correct |
17 |
Correct |
0 ms |
348 KB |
Output is correct |
18 |
Correct |
19 ms |
6044 KB |
Output is correct |
19 |
Correct |
23 ms |
7240 KB |
Output is correct |
20 |
Correct |
21 ms |
6068 KB |
Output is correct |
21 |
Correct |
22 ms |
6988 KB |
Output is correct |
22 |
Correct |
20 ms |
6040 KB |
Output is correct |
23 |
Correct |
27 ms |
6980 KB |
Output is correct |
24 |
Correct |
0 ms |
348 KB |
Output is correct |
25 |
Runtime error |
1 ms |
348 KB |
Execution killed with signal 11 |
26 |
Halted |
0 ms |
0 KB |
- |