Submission #1092446

# Submission time Handle Problem Language Result Execution time Memory
1092446 2024-09-24T06:20:01 Z juicy Biochips (IZhO12_biochips) C++17
100 / 100
293 ms 400208 KB
#include <bits/stdc++.h>

using namespace std;

#ifdef LOCAL
#include "debug.h"
#else
#define debug(...) 42
#endif

const int N = 2e5 + 5;

int n, m;
int a[N], L[N], R[N], pos[N], dp[N][502];
vector<int> g[N];

int timer;

void dfs(int u) {
  pos[L[u] = ++timer] = u;
  for (int v : g[u]) {
    dfs(v);
  }
  R[u] = timer;
}

int main() {
  ios::sync_with_stdio(false); cin.tie(nullptr);

  cin >> n >> m;
  int rt = -1;
  for (int i = 1; i <= n; ++i) {
    int pr; cin >> pr >> a[i];
    if (!pr) {
      rt = i;
    } else {
      g[pr].push_back(i);
    }
  }  
  dfs(rt);
  for (int i = 1; i <= n; ++i) {
    int u = pos[i];
    for (int j = 0; j <= m; ++j) {
      dp[R[u] + 1][j + 1] = max(dp[R[u] + 1][j + 1], dp[i][j] + a[u]);
      dp[i + 1][j] = max(dp[i + 1][j], dp[i][j]);
    } 
  }
  cout << dp[n + 1][m];
  return 0;
}
# Verdict Execution time Memory Grader output
1 Correct 2 ms 4956 KB Output is correct
2 Correct 2 ms 5208 KB Output is correct
3 Correct 3 ms 5212 KB Output is correct
4 Correct 11 ms 22848 KB Output is correct
5 Correct 13 ms 24924 KB Output is correct
6 Correct 12 ms 25020 KB Output is correct
7 Correct 187 ms 298332 KB Output is correct
8 Correct 185 ms 298324 KB Output is correct
9 Correct 239 ms 362676 KB Output is correct
10 Correct 293 ms 400208 KB Output is correct