# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
871727 | Nonoze | 메기 농장 (IOI22_fish) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#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 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<min(n, 8); i++) {
int temp=dp(empl+1, i, take);
if ((pretake&(1<<i))!=(1<<i) && (take&(1<<i))!=(1<<i) && empl>0) {
sum+=values[empl-1][i];
}
if (empl+1<n) sum+=values[empl+1][i];
if ((take&(1<<i))==(1<<i)) 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;
}
values.resize(n, vector<int>(8, 0));
for (int i=0; i<m; i++) {
values[x[i]][y[i]]=w[i];
}
return dp(0, 0, 0);
#undef int
}