#include "fish.h"
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
#define mp make_pair
#define fr first
#define sc second
#define all(x) x.begin(),x.end()
const int maxn = 1e5+10;
const int maxh = 12;
const int maxqh = 12;
const int inf = 1e18+10;
int n, m, dp[maxn][maxqh][maxqh];
vector<pair<int,int>> fish[maxn]; // fishs from column i
vector<int> hr[maxn]; // important heights from column i, the ones that can be chosen
long long max_weights(int32_t N, int32_t M, vector<int32_t> X, vector<int32_t> Y, vector<int32_t> W) {
n = N;
m = M;
for(int i = 0; i < m; i++) {
fish[X[i]+1].pb(mp(Y[i]+1,W[i]));
hr[X[i]+1].pb(Y[i]);
}
for(int i = 0; i <= n; i++) {
hr[i].pb(0);
hr[i].pb(maxh-1);
sort(all(hr[i]));
hr[i].erase(unique(all(hr[i])),hr[i].end());
}
// dp[i][h1][h] = until i and with h[i-1] = hr[i-1][h1] and h[i] = hr[i][h-1]
// dp[1][0][x] = 0;
for(int i = 0; i < hr[i].size(); i++) dp[1][0][i] = 0;
for(int i = 2; i <= n; i++) {
for(int h1 = 0; h1 < hr[i-1].size(); h1++) {
for(int h = 0; h < hr[i].size(); h++) {
dp[i][h1][h] = 0;
int plus10 = 0; // fishs that 1 catchs in 0
for(auto aux : fish[i]) {
int y = aux.fr;
int w = aux.sc;
if(y > hr[i][h] && y <= hr[i-1][h1]) plus10+= w;
}
for(int h2 = 0; h2 < hr[i-2].size(); h2++) {
int plus01 = 0; // fishs that 0 catchs in 1 and had not been caught by 2
for(auto aux : fish[i-1]) {
int y = aux.fr;
int w = aux.sc;
if(y > hr[i-1][h1] && y <= hr[i][h] && y > hr[i-2][h2]) plus01+= w;
}
dp[i][h1][h] = max(dp[i][h1][h],dp[i-1][h2][h1]+plus01 + plus10);
}
}
}
}
int ans = 0;
for(int h = 0; h < hr[n].size(); h++) {
for(int h1 = 0; h1 < hr[n-1].size(); h1++) {
ans = max(ans,dp[n][h1][h]);
}
}
// // dp[i][h1][h] = until i and with h[i-1] = h1 and h[i] = h
// // dp[1][0][h] = 0;
// for(int i = 0; i <= maxh-1; i++) dp[1][0][i] = 0;
// for(int i = 2; i <= n; i++) {
// for(int h1 = 0; h1 <= maxh-1; h1++) {
// for(int h = 0; h <= maxh-1; h++) {
// dp[i][h1][h] = 0;
// int plus10 = 0; // fishs that 1 catchs in 0
// for(auto aux : fish[i]) {
// int y = aux.fr;
// int w = aux.sc;
// if(y > h && y <= h1) plus10+= w;
// }
// for(int h2 = 0; h2 <= maxh-1; h2++) {
// int plus01 = 0; // fishs that 0 catchs in 1 and had not been caught by 2
// for(auto aux : fish[i-1]) {
// int y = aux.fr;
// int w = aux.sc;
// if(y > h1 && y <= h && y > h2) plus01+= w;
// }
// dp[i][h1][h] = max(dp[i][h1][h],dp[i-1][h2][h1]+plus01 + plus10);
// }
// }
// }
// }
// int ans = 0;
// for(int h = 0; h <= maxh-1; h++) {
// for(int h1 = 0; h1 <= maxh-1; h1++) {
// ans = max(ans,dp[n][h1][h]);
// }
// }
return ans;
}
Compilation message
fish.cpp: In function 'long long int max_weights(int32_t, int32_t, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:39:22: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
39 | for(int i = 0; i < hr[i].size(); i++) dp[1][0][i] = 0;
| ~~^~~~~~~~~~~~~~
fish.cpp:41:28: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
41 | for(int h1 = 0; h1 < hr[i-1].size(); h1++) {
| ~~~^~~~~~~~~~~~~~~~
fish.cpp:42:30: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
42 | for(int h = 0; h < hr[i].size(); h++) {
| ~~^~~~~~~~~~~~~~
fish.cpp:52:36: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
52 | for(int h2 = 0; h2 < hr[i-2].size(); h2++) {
| ~~~^~~~~~~~~~~~~~~~
fish.cpp:67:22: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
67 | for(int h = 0; h < hr[n].size(); h++) {
| ~~^~~~~~~~~~~~~~
fish.cpp:68:28: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
68 | for(int h1 = 0; h1 < hr[n-1].size(); h1++) {
| ~~~^~~~~~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1072 ms |
12028 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
3 ms |
4948 KB |
Output is correct |
2 |
Execution timed out |
1092 ms |
15596 KB |
Time limit exceeded |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
52 ms |
120780 KB |
Output is correct |
2 |
Correct |
55 ms |
120784 KB |
Output is correct |
3 |
Correct |
82 ms |
113632 KB |
Output is correct |
4 |
Correct |
77 ms |
123352 KB |
Output is correct |
5 |
Correct |
117 ms |
130980 KB |
Output is correct |
6 |
Correct |
113 ms |
131016 KB |
Output is correct |
7 |
Correct |
131 ms |
131112 KB |
Output is correct |
8 |
Correct |
128 ms |
130956 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
3 ms |
4948 KB |
Output is correct |
2 |
Correct |
2 ms |
4948 KB |
Output is correct |
3 |
Correct |
3 ms |
4948 KB |
Output is correct |
4 |
Correct |
3 ms |
4948 KB |
Output is correct |
5 |
Correct |
3 ms |
4948 KB |
Output is correct |
6 |
Correct |
3 ms |
4948 KB |
Output is correct |
7 |
Correct |
3 ms |
4948 KB |
Output is correct |
8 |
Correct |
3 ms |
4948 KB |
Output is correct |
9 |
Correct |
4 ms |
5204 KB |
Output is correct |
10 |
Correct |
9 ms |
5460 KB |
Output is correct |
11 |
Correct |
5 ms |
5204 KB |
Output is correct |
12 |
Correct |
5 ms |
5432 KB |
Output is correct |
13 |
Correct |
3 ms |
5076 KB |
Output is correct |
14 |
Correct |
6 ms |
5332 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
3 ms |
4948 KB |
Output is correct |
2 |
Correct |
2 ms |
4948 KB |
Output is correct |
3 |
Correct |
3 ms |
4948 KB |
Output is correct |
4 |
Correct |
3 ms |
4948 KB |
Output is correct |
5 |
Correct |
3 ms |
4948 KB |
Output is correct |
6 |
Correct |
3 ms |
4948 KB |
Output is correct |
7 |
Correct |
3 ms |
4948 KB |
Output is correct |
8 |
Correct |
3 ms |
4948 KB |
Output is correct |
9 |
Correct |
4 ms |
5204 KB |
Output is correct |
10 |
Correct |
9 ms |
5460 KB |
Output is correct |
11 |
Correct |
5 ms |
5204 KB |
Output is correct |
12 |
Correct |
5 ms |
5432 KB |
Output is correct |
13 |
Correct |
3 ms |
5076 KB |
Output is correct |
14 |
Correct |
6 ms |
5332 KB |
Output is correct |
15 |
Incorrect |
3 ms |
5332 KB |
1st lines differ - on the 1st token, expected: '299', found: '298' |
16 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
3 ms |
4948 KB |
Output is correct |
2 |
Correct |
2 ms |
4948 KB |
Output is correct |
3 |
Correct |
3 ms |
4948 KB |
Output is correct |
4 |
Correct |
3 ms |
4948 KB |
Output is correct |
5 |
Correct |
3 ms |
4948 KB |
Output is correct |
6 |
Correct |
3 ms |
4948 KB |
Output is correct |
7 |
Correct |
3 ms |
4948 KB |
Output is correct |
8 |
Correct |
3 ms |
4948 KB |
Output is correct |
9 |
Correct |
4 ms |
5204 KB |
Output is correct |
10 |
Correct |
9 ms |
5460 KB |
Output is correct |
11 |
Correct |
5 ms |
5204 KB |
Output is correct |
12 |
Correct |
5 ms |
5432 KB |
Output is correct |
13 |
Correct |
3 ms |
5076 KB |
Output is correct |
14 |
Correct |
6 ms |
5332 KB |
Output is correct |
15 |
Incorrect |
3 ms |
5332 KB |
1st lines differ - on the 1st token, expected: '299', found: '298' |
16 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
52 ms |
120780 KB |
Output is correct |
2 |
Correct |
55 ms |
120784 KB |
Output is correct |
3 |
Correct |
82 ms |
113632 KB |
Output is correct |
4 |
Correct |
77 ms |
123352 KB |
Output is correct |
5 |
Correct |
117 ms |
130980 KB |
Output is correct |
6 |
Correct |
113 ms |
131016 KB |
Output is correct |
7 |
Correct |
131 ms |
131112 KB |
Output is correct |
8 |
Correct |
128 ms |
130956 KB |
Output is correct |
9 |
Incorrect |
123 ms |
131008 KB |
1st lines differ - on the 1st token, expected: '99999', found: '99998' |
10 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1072 ms |
12028 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |