Submission #1055910

#	Time	Username	Problem	Language	Result	Execution time	Memory
1055910		mychecksedad	Catfish Farm (IOI22_fish)	C++17	35 / 100	1104 ms	2097152 KiB

fish

#include "fish.h"
#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define all(x) x.begin(),x.end()
#define ll long long int
#define en cout << '\n'
#define pi pair<int,int>
#define vi vector<int> 
#define ff first
#define ss second
const int N = 3005;

ll dp[N][N], pref[N][N], mx[N], dp2[N][N];
long long max_weights(int n, int m, std::vector<int> X, std::vector<int> Y, std::vector<int> W) {
  vector<vector<ll>> a(n + 1, vector<ll>(n + 1));
  for(int i = 0; i < m; ++i){
    a[X[i]][Y[i] + 1] = W[i];
  }
  for(int i = 0; i < n; ++i){
    pref[i][0] = a[i][0];
    for(int j = 1; j <= n; ++j) pref[i][j] = pref[i][j - 1] + a[i][j];
  }
  for(int i = 0; i <= n; ++i) dp[0][i] = pref[1][i], dp2[0][i] = pref[1][i];
  mx[0] = pref[1][n];
  for(int i = 1; i < n; ++i){
    mx[i] = 0;
    for(int j = 0; j <= n; ++j){
      ll add1 = pref[i + 1][j];
      ll add2 = pref[i - 1][j];
      dp[i][j] = add1 + add2;
      if(i > 2) dp[i][j] = max(dp[i][j], mx[i - 3] + add1 + add2);
      // cout << i << ' ' << j << ' ' << add1 << ' ' <<add2 << ' ';
      dp2[i][j] = dp[i][j];
      for(int j2 = 0; j2 <= n; ++j2){
        if(i > 1){
          dp[i][j] = max(dp[i][j], dp[i - 2][j2] - pref[i - 1][min(j, j2)] + add1 + add2);
          dp2[i][j] = max(dp2[i][j], dp[i - 2][j2] - pref[i - 1][min(j, j2)] + add1 + add2);
        }
        if(j2 >= j) dp[i][j] = max(dp[i][j], dp[i - 1][j2] + add1 + add2 - pref[i - 1][min(j, j2)] - pref[i][min(j, j2)]);
        else{
          dp[i][j] = max(dp[i][j], dp2[i - 1][j2] + add1 + add2 - pref[i - 1][min(j, j2)] - pref[i][min(j, j2)]);
          dp2[i][j] = max(dp2[i][j], dp2[i - 1][j2] + add1 + add2 - pref[i - 1][min(j, j2)] - pref[i][min(j, j2)]);
        }
      }
      mx[i] = max(mx[i], dp[i][j]);
      // cout << dp[i][j] << '\n'; 
    }
    // en;
  }
  ll ans = mx[n - 1];

  return ans;
}

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