Submission #336947

#	Time	Username	Problem	Language	Result	Execution time	Memory
336947		boykut	K blocks (IZhO14_blocks)	C++14	0 / 100	145 ms	81900 KiB

blocks

#include <map>
#include <vector>
#include <fstream>

using namespace std;

ifstream cin("triangles.in");
ofstream cout("triangles.out");

const int inf = 2e9;

int a[100001];
int dp[100001][101];

int32_t main() {
  ios::sync_with_stdio(0);
  cin.tie(0);
  int N, K;
  cin >> N >> K;
  
  for (int i = 1; i <= N; i++) {
    cin >> a[i];
  }
  
  for (int i = 0; i <= N; i++) {
    for (int j = 0; j <= K; j++) {
      dp[i][j] = inf;
    }
  }
  
  dp[0][0] = 0;
  dp[1][1] = a[1];
  for (int i = 2; i <= N; i++) {
    dp[i][1] = max(dp[i - 1][1], a[i]);
  }
  
  auto getmax = [&](int l, int r) -> int {
    int mx = a[l];
    for (int i = l; i <= r; i++) {
      mx = max(a[l], a[i]);
    }
    return mx;
  };
  
  for (int k = 2; k <= K; k++) {
    for (int i = 1; i <= N; i++) {
      int first = dp[i - 1][k - 1];
      for (int j = 1; j < i; j++) {
        dp[i][k] = min(dp[i][k], dp[j][k - 1] + getmax(j + 1, i));
      }
    }
  }
  
  cout << dp[N][K] << '\n';
  
  return 0;
}

Compilation message (stderr)

blocks.cpp: In function 'int32_t main()':
blocks.cpp:47:11: warning: unused variable 'first' [-Wunused-variable]
   47 |       int first = dp[i - 1][k - 1];
      |           ^~~~~

• Subtask 1: 0 / 14
• Subtask 2: 0 / 18
• Subtask 3: 0 / 21
• Subtask 4: 0 / 47