/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 2ms 2.777 MiB
#2 Accepted 3ms 2.777 MiB
#3 Accepted 2ms 2.777 MiB
#4 Accepted 3ms 2.777 MiB
#5 Accepted 3ms 2.777 MiB
#6 Accepted 3ms 2.777 MiB
#7 Accepted 3ms 2.777 MiB
#8 Accepted 3ms 2.777 MiB
#9 Accepted 2ms 2.777 MiB
#10 Accepted 2ms 3.027 MiB
#11 Accepted 2ms 2.777 MiB
#12 Accepted 45ms 6.141 MiB
#13 Accepted 65ms 6.094 MiB
#14 Accepted 61ms 6.52 MiB
#15 Accepted 62ms 6.27 MiB
#16 Accepted 54ms 6.094 MiB
#17 Accepted 51ms 6.16 MiB
#18 Accepted 49ms 6.27 MiB
#19 Accepted 57ms 6.23 MiB
#20 Accepted 47ms 6.117 MiB
#21 Accepted 46ms 6.227 MiB
#22 Accepted 52ms 6.281 MiB
#23 Accepted 51ms 6.27 MiB
#24 Accepted 59ms 6.488 MiB
#25 Accepted 58ms 6.27 MiB
#26 Accepted 43ms 6.191 MiB
#27 Accepted 63ms 6.223 MiB
#28 Accepted 44ms 6.41 MiB
#29 Accepted 47ms 6.09 MiB
#30 Accepted 59ms 6.27 MiB
#31 Accepted 64ms 6.312 MiB
#32 Accepted 649ms 246.188 MiB
#33 Accepted 458ms 169.801 MiB
#34 Accepted 51ms 6.273 MiB
#35 Accepted 54ms 6.52 MiB
#36 Accepted 47ms 6.52 MiB
#37 Accepted 45ms 6.152 MiB
#38 Accepted 55ms 6.062 MiB
#39 Accepted 44ms 6.164 MiB
#40 Accepted 53ms 6.27 MiB
#41 Accepted 50ms 6.27 MiB
#42 Accepted 72ms 6.395 MiB
#43 Accepted 64ms 6.27 MiB
#44 Accepted 60ms 6.27 MiB
#45 Accepted 65ms 6.27 MiB
#46 Accepted 51ms 6.27 MiB
#47 Accepted 76ms 6.27 MiB
#48 Accepted 64ms 6.27 MiB
#49 Accepted 47ms 6.254 MiB
#50 Accepted 99ms 6.27 MiB

Code

#include <bits/stdc++.h>

using namespace std;

const int N = 1e5 + 5;
int64_t ans, INF = -1e18;
int a[N], n, k;
vector<int> g[N];

vector<int64_t> dfs(int node, int pr = -1) {
  vector<int64_t> dp(k + 1, INF);
  dp[1] = a[node];
  for (auto &son : g[node]) {
    if (son == pr) continue;
    vector<int64_t> cdp = dfs(son, node);
    for (int i = 1; i < k; i++) {
      if (dp[i] != INF && cdp[k - i] != INF) {
        ans = max(ans, dp[i] + cdp[k - i]);
      }
    }
    for (int i = 2; i <= k; i++) {
      if (cdp[i - 1] != INF) {
        dp[i] = max(dp[i], cdp[i - 1] + dp[1]);
      }
    }
  }
  ans = max(ans, dp[k]);
  return dp;
}

void solve(int cs) {
  cin >> n >> k;
  for (int i = 1; i <= n; i++)
    cin >> a[i];
  for (int i = 0; i < n - 1; i++) {
    int u, v;
    cin >> u >> v;
    g[u].push_back(v);
    g[v].push_back(u);
  }
  ans = INF;
  dfs(1);
  cout << (ans == INF ? 0 : ans) << "\n";
}

int32_t main() {
  ios::sync_with_stdio(!cin.tie(0));
  int t = 1;
  // cin >> t;
  for (int i = 1; i <= t; ++i) {
    solve(i);
  }
  return 0;
}

Information

Submit By
Type
Submission
Problem
P1160 Max path sum (Easy Version)
Language
C++17 (G++ 13.2.0)
Submit At
2025-02-17 18:12:16
Judged At
2025-02-17 18:12:16
Judged By
Score
100
Total Time
649ms
Peak Memory
246.188 MiB