Submission #1072501

#	Time	Username	Problem	Language	Result	Execution time	Memory
1072501		Boas	Catfish Farm (IOI22_fish)	C++17	53 / 100	1096 ms	2097152 KiB

fish

#include "fish.h"

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

#pragma GCC optimize("Ofast")

#define loop(x, i) for (int i = 0; i < (x); i++)
#define loop1(x, i) for (int i = 1; i <= (x); i++)
#define rev(x, i) for (int i = (int)(x) - 1; i >= 0; i--)
#define itloop(x) for (auto it = begin(x); x != end(x); it++)
#define itrev(x) for (auto it = rbegin(x); x != rend(x); it++)
#define int long long
#define INF ((int64_t)(4e18 + 1))
#define INF32 ((int32_t)(2e9 + 1))
#define ALL(x) begin(x), end(x)
#define RALL(x) rbegin(x), rend(x)
#define removeIn(x, l) l.erase(find(ALL(l), x))
#define pb push_back
#define sz(x) (int)(x).size()
#define F first
#define S second
#define var const auto &
#define foreach(l) for (var e : l)

typedef int8_t i8;
typedef int16_t i16;
typedef int32_t i32;
typedef int64_t i64;
typedef pair<int, int> ii;
typedef tuple<int, int, int> iii;
typedef tuple<int, int, int, int> iiii;
typedef vector<int> vi;
typedef vector<i32> vi32;
typedef vector<vi> vvi;
typedef vector<vvi> vvvi;
typedef vector<vvvi> vvvvi;
typedef vector<ii> vii;
typedef vector<iii> viii;
typedef vector<vii> vvii;
typedef vector<viii> vviii;
typedef set<int> si;
typedef set<ii> sii;
typedef set<iii> siii;
typedef vector<si> vsi;
typedef vector<sii> vsii;
typedef vector<vsi> vvsi;
typedef vector<string> vstr;
typedef vector<vector<string>> vvstr;
typedef vector<bool> vb;
typedef vector<vb> vvb;

int max_weightsB(i32 N, i32 M, vi32 X, vi32 Y, vi32 W)
{
  vi pSum0 = {0}, pSum1 = {0};
  vi at0(N), at1(N);
  loop(M, i)
  {
    if (X[i] == 0)
      at0[Y[i]] = W[i];
    else
      at1[Y[i]] = W[i];
  }
  loop(N, i)
  {
    pSum0.pb(pSum0.back() + at0[i]);
    pSum1.pb(pSum1.back() + at1[i]);
  }
  int res = 0;
  loop(N + 1, h1)
  {
    res = max(res, pSum1.back() - pSum1[h1] + pSum0[h1]);
  }
  if (N == 2)
    return max(pSum0.back(), pSum1.back());
  return res;
}

int max_weights(i32 N, i32 M, vi32 X, vi32 Y, vi32 W)
{
  if (M == 0)
    return 0;
  bool allEven = 1;
  for (int x : X)
    if (x % 2)
      allEven = 0;
  if (allEven)
    return accumulate(ALL(W), 0LL);

  if (*max_element(ALL(X)) <= 1)
    return max_weightsB(N, M, X, Y, W);

  int yCount = *max_element(ALL(Y)) + 2;
  vvi wSum(N, vi(yCount));
  {
    vvi wsAtX(N, vi(yCount));
    loop(M, i)
    {
      wsAtX[X[i]][Y[i]] = W[i];
    }
    loop(N, x)
    {
      loop1(yCount - 1, y)
      {
        wSum[x][y] = wSum[x][y - 1] + wsAtX[x][y - 1];
      }
    }
  }
  vvvi dp(vvvi(yCount, vvi(yCount, vi(2))));  // 0: adjacentL, 1: adjacentR
  vvvi dp2(vvvi(yCount, vvi(yCount, vi(2)))); // 0: adjacentL, 1: adjacentR
  vi maxAll(N);
  vvi maxOfHeight(N, vi(yCount));
  loop(N, x)
  {
    vvi pMax1perJ(yCount, vi(yCount)); // dp[<=j][i][1]
    if (x != 0)
    {
      loop(yCount, j)
      {
        loop(yCount, i)
        {
          pMax1perJ[j][i] = max(j == 0 ? 0 : pMax1perJ[j - 1][i], dp[j][i][1]);
        }
      }
    }
    loop(yCount, j)
    {
      int prevMax = 0;
      if (x != 0)
      {
        loop(yCount, i)
        {
          prevMax = max(prevMax, dp[j][i][0]);
        }
      }
      loop(yCount, i) // hoogte van deze steiger
      {
        if (j <= i)
        {
          dp2[i][j][0] = dp2[i][j][1] = (x == 0) ? 0 : maxOfHeight[x - 1][i];
        }
        else
        {
          int sum = wSum[x][j] - wSum[x][i];
          if (x == 0)
          {
            dp2[i][j][0] = -1;
            dp2[i][j][1] = sum;
          }
          else
          {
            dp2[i][j][0] = max(dp[j][i][1], prevMax) + sum;
            dp2[i][j][1] = max(pMax1perJ[j][i], maxAll[x - 1]) + sum;
          }
        }
        maxAll[x] = max({maxAll[x], dp2[i][j][0]});
        maxOfHeight[x][j] = max({maxOfHeight[x][j], dp2[i][j][0], dp2[i][j][1]});
      }
    }
    dp = dp2;
  }
  return maxAll.back();
}

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