제출 #830758

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
830758		NeroZein	Catfish Farm (IOI22_fish)	C++17	26 / 100	1092 ms	2097152 KiB

fish

#include "fish.h"
#include <bits/stdc++.h>
using namespace std;

const int N = 302;
int n; 
int max_m, max_n;
vector<vector<long long>> a;
vector<vector<long long>> pref;
vector<vector<vector<long long>>> dp;

long long bt(int col, int last, int last2) {
  if (col > min(n, max_m + 1)) {
    return 0;
  }
  long long& ret = dp[col][last][last2];
  if (ret != -1) {
    return ret;
  }
  long long sum = 0; 
  long long ps = pref[col][last]; 
  for (int nlast = 0; nlast <= min(n, max_n + 1); ++nlast) {
    if (nlast > last2 && nlast > last) sum += a[col - 1][nlast];
    if (nlast <= last) ps -= a[col][nlast];
    ret = max(ret, bt(col + 1, nlast, last) + sum + ps); 
  }
  return ret;
}

long long max_weights(int N_, int M, vector<int> X, vector<int> Y, vector<int> W) {
  n = N_; 
  bool even = true;
  for (int i = 0; i < M; ++i) {
    max_m = max(max_m, X[i] + 1); 
    max_n = max(max_n, Y[i] + 1); 
    even &= (X[i] % 2 == 0);
  }
  if (even) {
    return accumulate(W.begin(), W.end(), 0LL);
  }
  dp.resize(max_m + 2); 
  for (int i = 0; i < max_m + 2; ++i) {
    dp[i].resize(max_n + 2);
    for (int j = 0; j < max_n + 2; ++j) {
      dp[i][j].assign(max_n + 2, -1);
    }
  }
  a.resize(max_m + 2);
  for (int i = 0; i < max_m + 2; ++i) {
    a[i].resize(max_n + 2); 
  }
  for (int i = 0; i < M; ++i) {
    a[X[i] + 1][Y[i] + 1] = W[i];
  }
  pref.resize(max_m + 2);
  for (int i = 0; i < max_m + 2; ++i) {
    pref[i].resize(max_n + 2); 
  }
  for (int i = 1; i < max_m + 2; ++i) {
    for (int j = 1; j < max_n + 2; ++j) {
      pref[i][j] = pref[i][j - 1] + a[i][j];      
    } 
  }
  return bt(1, 0, 0); 
}

• Subtask 1: 3 / 3
• Subtask 2: 0 / 6
• Subtask 3: 9 / 9
• Subtask 4: 14 / 14
• Subtask 5: 0 / 21
• Subtask 6: 0 / 17
• Subtask 7: 0 / 14
• Subtask 8: 0 / 16