Submission #1166113

#	Time	Username	Problem	Language	Result	Execution time	Memory
1166113		JelalTkm	Split the sequence (APIO14_sequence)	C++20	0 / 100	0 ms	324 KiB

sequence

#include <bits/stdc++.h>
#pragma GCC optimize ("O3")
#pragma GCC target ("sse4")

using namespace std;

#define int long long int

const int N = 1e5 + 10;
const int md = 1e9 + 7;
const int INF = 1e18;

int32_t main(int32_t argc, char *argv[]) {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);

  int T = 1;
  // cin >> T;
  while (T--) {
    int n, k;
    cin >> n >> k; k++;
    vector<int> a(n + 1), pref(n + 1);
    for (int i = 1; i <= n; i++) {
      cin >> a[i];
      pref[i] = pref[i - 1] + a[i];
    }

    vector<vector<pair<int, int>>> sm(n + 1, vector<pair<int, int>> (n + 1));
    for (int i = 1; i <= n; i++) {
      sm[i][i].first = a[i];
      for (int j = i + 1; j <= n; j++) {
        int l = i, r = j;
        while (r - l > 1) {
          int mid = (l + r) >> 1;
          if ((pref[mid] - pref[i - 1]) > (pref[j] - pref[mid]))
            r = mid;
          else l = mid;
        }

        sm[i][j].first = (pref[l] - pref[i - 1]) * (pref[j] - pref[l]);
        sm[i][j].second = l;
        if (sm[i][j].first < (pref[l + 1] - pref[i - 1]) * (pref[j] - pref[l + 1])) {
          sm[i][j].first = (pref[l + 1] - pref[i - 1]) * (pref[j] - pref[l + 1]);
          sm[i][j].second = l + 1;
        }
      }
    }

    vector<vector<int>> dp(n + 1, vector<int> (k + 1));
    for (int q = 1; q <= k; q++) {
      for (int i = 1; i <= n; i++) {
        for (int j = i + 1; j <= n; j++) {
          if (!((q - 2) == 0 && (i - 1) >= 1)) {
            dp[j][q] = max(dp[j][q], dp[i - 1][q - 2] + sm[i][j].first + (pref[i - 1] * (pref[j] - pref[i - 1])));
          }
        }
      }
    }

    cout << dp[n][k] << '\n';

    vector<int> ans;
    int i = n, q = k;
    int cnt = 0;
    while (q > 1) {
      cnt++;
      for (int j = (i - 1); j >= 1; j--) {
        if (!((q - 2) == 0 && (j - 1) >= 1)) {
          if (dp[i][q] == (dp[j - 1][q - 2] + sm[j][i].first + (pref[j - 1] * (pref[i] - pref[j - 1])))) {
            ans.push_back(sm[j][i].second);
            ans.push_back(j - 1);
            i = j - 1;
            q -= 2;
            break;
          }
        }
      }
    }

    reverse(ans.begin(), ans.end());
    for (int i = (ans[0] == 0 ? 1 : 0); i < (int) ans.size(); i++)
      cout << ans[i] << " ";
    cout << '\n';
  }

  return 0;
}

• Subtask 1: 0 / 11
• Subtask 2: 0 / 11
• Subtask 3: 0 / 11
• Subtask 4: 0 / 17
• Subtask 5: 0 / 21
• Subtask 6: 0 / 29