# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
760951 | drdilyor | Catfish Farm (IOI22_fish) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
ll max_weights(
int n, int m, vector<int> x, vector<int> y, vector<int> w) {
vector arr(n+1, vector(n+1, 0ll));
for (int i = 0; i < m; i++)
arr[x[i]+1][y[i]] += w[i];
vector pref(n+1, vector(n+2, 0ll));
for (int i = 0; i <= n; i++) {
for (int j = 0; j <= n; j++)
pref[i][j+1] = pref[i][j] + arr[i][j];
auto sum = [&](int c, int l, int r) { // [l, r)
ll res = pref[c][r] - pref[c][l];
if (l <= r) return res;
else return 0ll;
};
vector memo(2, vector(n+1, vector(n+1, -1ll)));
auto dp = [&](auto& dp, bool inc, int c, int h2)->ll{
if (c == n+1) return 0;
ll& res = memo[inc][c][h2];
if (res!=-1) return res;
res = 0;
if (inc) {
for (int h3 = h2; h3 <= n; h3++) {
ll val = sum(c-1, h2, h3) + sum(c, h3, h2);
res = max(res, val + dp(dp, true, c+1, h3));
res = max(res, val + dp(dp, false, c+1, h3));
}
} else {
for (int h3 = 0; h3 <= h2; h3++) {
ll val = sum(c-1, h2, h3) + sum(c, h3, h2);
res = max(res, val + dp(dp, false, c+1, h3));
}
}
res = max(res, dp(dp, true, c+1, 0));
return res;
};
ll res = dp(dp, true, 1, 0);
return res;
}