#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define MP make_pair
long long max_weights(int N, int M, vector<int> X, vector<int> Y, vector<int> W) {
vector<vector<pair<int, int>>> fish(N);
vector<vector<ll>> sum(N);
for (int i = 0; i < M; i++){
fish[X[i]].emplace_back(Y[i], W[i]);
}
for (int i = 0; i < N; i++){
sort(fish[i].begin(), fish[i].end());
int k = (int)fish[i].size();
sum[i].resize(k);
for (int j = 0; j < k; j++){
sum[i][j] = fish[i][j].second;
if (j > 0) sum[i][j] += sum[i][j - 1];
}
}
function<ll(int, int, int)> get_sum = [&](int i, int l, int r){ // sum of fish with X = i, l <= Y <= r
int pl = lower_bound(fish[i].begin(), fish[i].end(), MP(l, 0)) - fish[i].begin();
int pr = lower_bound(fish[i].begin(), fish[i].end(), MP(r + 1, 0)) - fish[i].begin() - 1;
if (pl > pr) return 0ll;
return sum[i][pr] - (pl > 0 ? sum[i][pl - 1] : 0ll);
};
vector<vector<ll>> dp(N, vector<ll>(N + 1, 0));
for (int i = 1; i < N; i++){
for (int l = 0; l <= N; l++){
for (int j = 0; j <= N; j++){
dp[i][l] = max(dp[i][l], dp[i - 1][j] + get_sum(i - 1, j + 1, l));
}
for (int j = 0; j <= N; j++){
if (i >= 2) dp[i][l] = max(dp[i][l], dp[i - 2][j] + get_sum(i - 1, 0, l));
}
}
}
ll res = 0;
for (int l = 0; l < N; l++){
res = max(res, dp[N - 1][l]);
}
for (int l = 0; l < N; l++){
res = max(res, dp[N - 2][l] + get_sum(N - 1, 0, l));
}
return res;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1115 ms |
2097152 KB |
Time limit exceeded |
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 |
Runtime error |
715 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
212 KB |
1st lines differ - on the 1st token, expected: '3', found: '2' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
212 KB |
1st lines differ - on the 1st token, expected: '3', found: '2' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
212 KB |
1st lines differ - on the 1st token, expected: '3', found: '2' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
715 ms |
2097152 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1115 ms |
2097152 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |