/ SeriousOJ /

Record Detail

Runtime Error


  
# Status Time Cost Memory Cost
#1 Accepted 3ms 4.277 MiB
#2 Accepted 3ms 3.777 MiB
#3 Accepted 3ms 5.027 MiB
#4 Accepted 4ms 5.504 MiB
#5 Accepted 3ms 3.527 MiB
#6 Accepted 3ms 3.527 MiB
#7 Accepted 3ms 2.777 MiB
#8 Accepted 3ms 4.527 MiB
#9 Accepted 3ms 3.777 MiB
#10 Accepted 3ms 4.527 MiB
#11 Accepted 3ms 3.527 MiB
#12 Accepted 510ms 86.824 MiB
#13 Accepted 792ms 86.891 MiB
#14 Accepted 503ms 86.855 MiB
#15 Accepted 558ms 86.809 MiB
#16 Accepted 301ms 86.609 MiB
#17 Accepted 478ms 86.523 MiB
#18 Accepted 451ms 86.77 MiB
#19 Accepted 693ms 86.797 MiB
#20 Accepted 312ms 86.551 MiB
#21 Accepted 791ms 86.879 MiB
#22 Accepted 144ms 86.551 MiB
#23 Accepted 616ms 86.781 MiB
#24 Accepted 435ms 86.742 MiB
#25 Accepted 261ms 86.695 MiB
#26 Accepted 126ms 86.52 MiB
#27 Accepted 245ms 86.82 MiB
#28 Accepted 115ms 86.52 MiB
#29 Accepted 722ms 86.805 MiB
#30 Accepted 376ms 86.828 MiB
#31 Accepted 638ms 87.02 MiB
#32 Accepted 726ms 86.844 MiB
#33 Accepted 863ms 86.828 MiB
#34 Accepted 274ms 86.672 MiB
#35 Accepted 490ms 86.703 MiB
#36 Accepted 359ms 86.695 MiB
#37 Accepted 905ms 86.82 MiB
#38 Accepted 241ms 86.52 MiB
#39 Accepted 493ms 86.535 MiB
#40 Accepted 306ms 86.77 MiB
#41 Accepted 723ms 86.844 MiB
#42 Accepted 193ms 86.52 MiB
#43 Accepted 226ms 86.57 MiB
#44 Accepted 478ms 92.535 MiB
#45 Accepted 485ms 92.492 MiB
#46 Accepted 469ms 92.469 MiB
#47 Runtime Error 257ms 343.262 MiB
#48 Accepted 737ms 326.375 MiB
#49 Accepted 1305ms 558.207 MiB
#50 Accepted 226ms 145.379 MiB

Code

#define _GLIBCXX_FILESYSTEM
#include<bits/stdc++.h>
using namespace std;
#define ll long long

const int N = 1e5+2;
const ll inf = 1e16;

int a[N],n,k;
ll dp[N][105],ans;
vector<int> g[N];

void dfs(int u,int p=0) {
    dp[u][1] = a[u];
    multiset<ll,greater<ll>> v[k+1];
    int ch = 0;
    for(auto x: g[u]) {
        if(x == p) continue;
        dfs(x,u);
        ch++;
        for(int i = 0; i <= k; i++) {
            v[i].insert(dp[x][i]);
        }
        for(int i = 1; i < k; i++) {
            dp[u][i+1] = max(dp[u][i+1],dp[x][i]+a[u]);
        }
    }
    ans = max(ans, dp[u][k]);
    if(ch <= 1) return;
    for(auto x: g[u]) {
        if(x == p) continue;
        for(int j = 0; j < k; j++) {
            v[k - 1 - j].erase(v[k - 1 - j].find(dp[x][k - 1 - j]));
            ll mx = *v[k - 1 - j].begin();
            ll now = dp[x][j] + mx + a[u];
            ans = max(ans, now);
            v[k - 1 - j].insert(dp[x][k - 1 - j]);
        }
    }
}
void solve() {
    cin >> n >> k;
    for(int i = 1; i <= n; i++)
        cin >> a[i];
    for(int i = 1; i < n; i++) {
        int u,v;
        cin >> u >> v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= k; j++) {
            dp[i][j] = -inf;
        }
    }
    dfs(1);
    cout << ans << '\n';
    return;
}

int32_t main() {
    ios_base::sync_with_stdio(false);cin.tie(NULL);
    int tc = 1;
    // cin >> tc;
    for(int kase = 1; kase <= tc; kase++) {
        //cout << "Case " << kase << ": ";
        solve();
    }
    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 16:09:58
Judged At
2025-02-17 16:09:58
Judged By
Score
98
Total Time
1305ms
Peak Memory
558.207 MiB