/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 5ms 5.02 MiB
#2 Accepted 5ms 5.035 MiB
#3 Accepted 5ms 5.02 MiB
#4 Accepted 5ms 5.02 MiB
#5 Accepted 6ms 5.02 MiB
#6 Accepted 8ms 5.02 MiB
#7 Accepted 6ms 5.02 MiB
#8 Accepted 9ms 5.02 MiB
#9 Accepted 9ms 5.02 MiB
#10 Accepted 11ms 5.02 MiB
#11 Accepted 11ms 5.188 MiB
#12 Accepted 214ms 243.77 MiB
#13 Accepted 232ms 243.77 MiB
#14 Accepted 237ms 243.637 MiB
#15 Accepted 237ms 243.77 MiB
#16 Accepted 223ms 243.77 MiB
#17 Accepted 225ms 243.617 MiB
#18 Accepted 222ms 243.59 MiB
#19 Accepted 251ms 243.602 MiB
#20 Accepted 224ms 243.613 MiB
#21 Accepted 220ms 243.773 MiB
#22 Accepted 246ms 243.574 MiB
#23 Accepted 249ms 243.648 MiB
#24 Accepted 235ms 243.57 MiB
#25 Accepted 238ms 243.684 MiB
#26 Accepted 220ms 243.625 MiB
#27 Accepted 254ms 243.699 MiB
#28 Accepted 223ms 243.598 MiB
#29 Accepted 228ms 243.547 MiB
#30 Accepted 239ms 243.77 MiB
#31 Accepted 255ms 243.715 MiB
#32 Accepted 271ms 255.492 MiB
#33 Accepted 255ms 255.27 MiB
#34 Accepted 226ms 243.012 MiB
#35 Accepted 225ms 243.215 MiB
#36 Accepted 219ms 243.02 MiB
#37 Accepted 236ms 243.23 MiB
#38 Accepted 228ms 243.184 MiB
#39 Accepted 235ms 243.062 MiB
#40 Accepted 243ms 243.02 MiB
#41 Accepted 245ms 243.074 MiB
#42 Accepted 267ms 243.703 MiB
#43 Accepted 238ms 243.582 MiB
#44 Accepted 238ms 243.645 MiB
#45 Accepted 242ms 243.703 MiB
#46 Accepted 230ms 243.754 MiB
#47 Accepted 254ms 243.574 MiB
#48 Accepted 245ms 243.57 MiB
#49 Accepted 213ms 243.527 MiB
#50 Accepted 281ms 243.684 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];
ll dp[N][305],n,k;
vector<int> g[N],ch[N];

void dfs(int u,int p=0) {
    dp[u][1] = a[u];
    for(auto x: g[u]) {
        if(x == p) continue;
        dfs(x,u);
        ch[u].push_back(x);
        for(int i = 1; i < k; i++) {
            dp[u][i+1] = max(dp[u][i+1],dp[x][i]+a[u]);
        }
    }
}
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);
    ll ans = 0;
    for(int i = 1; i <= n; i++) {
        if(ch[i].size() <= 1) {
            ans = max(ans, dp[i][k]);
            continue;
        }
        for(int j = 0; j < k; j++) {
            ans = max(ans, dp[ch[i][0]][j] + dp[ch[i][1]][k - 1 - j] + a[i]);
        }
    }
    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
P1160 Max path sum (Easy Version)
Contest
Brain Booster #8
Language
C++17 (G++ 13.2.0)
Submit At
2025-02-17 15:42:45
Judged At
2025-02-17 15:42:45
Judged By
Score
100
Total Time
281ms
Peak Memory
255.492 MiB