/ SeriousOJ /

Record Detail

Time Exceeded


  
# Status Time Cost Memory Cost
#1 Accepted 1ms 540.0 KiB
#2 Accepted 1ms 540.0 KiB
#3 Accepted 1ms 772.0 KiB
#4 Accepted 1ms 328.0 KiB
#5 Accepted 1ms 540.0 KiB
#6 Accepted 1ms 540.0 KiB
#7 Accepted 1ms 540.0 KiB
#8 Accepted 1ms 540.0 KiB
#9 Accepted 1ms 540.0 KiB
#10 Accepted 2ms 584.0 KiB
#11 Accepted 1ms 540.0 KiB
#12 Time Exceeded ≥3099ms ≥6.57 MiB
#13 Time Exceeded ≥3100ms ≥6.617 MiB

Code

#include <bits/stdc++.h>

using namespace std;

void solve(int cs) {
  int n, k;
  cin >> n >> k;
  vector<int64_t> a(n);
  for (int i = 0; i < n; i++) {
    cin >> a[i];
  }
  vector<vector<int>> g(n);
  for (int i = 0; i < n - 1; i++) {
    int u, v; cin >> u >> v;
    --u, --v;
    g[u].push_back(v);
    g[v].push_back(u);
  }
  int64_t ans = 0;
  auto dfs = [&](auto &&self, int node, int pr, int len, int64_t sum) -> void {
    if (len == k) {
      ans = max(ans, sum);
    } else {
      for (auto &son : g[node]) {
        if (son ^ pr) {
          self(self, son, node, len + 1, sum + a[son]);
        }
      }
    }
  };
  for (int i = 0; i < n; i++) {
    dfs(dfs, i, -1, 1, a[i]);
  }
  cout << 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
P1161 Max path sum (Hard Version)
Contest
Brain Booster #8
Language
C++17 (G++ 13.2.0)
Submit At
2025-02-17 15:50:42
Judged At
2025-02-17 15:50:42
Judged By
Score
22
Total Time
≥3100ms
Peak Memory
≥6.617 MiB