/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 5ms 5.02 MiB
#2 Accepted 5ms 5.02 MiB
#3 Accepted 6ms 5.02 MiB
#4 Accepted 6ms 5.02 MiB
#5 Accepted 6ms 5.191 MiB
#6 Accepted 9ms 5.02 MiB
#7 Accepted 9ms 5.035 MiB
#8 Accepted 12ms 5.02 MiB
#9 Accepted 12ms 5.035 MiB
#10 Accepted 6ms 5.02 MiB
#11 Accepted 5ms 4.965 MiB
#12 Accepted 225ms 244.508 MiB
#13 Accepted 262ms 244.438 MiB
#14 Accepted 240ms 244.633 MiB
#15 Accepted 241ms 244.52 MiB
#16 Accepted 228ms 244.5 MiB
#17 Accepted 230ms 244.438 MiB
#18 Accepted 253ms 244.461 MiB
#19 Accepted 232ms 244.477 MiB
#20 Accepted 260ms 244.473 MiB
#21 Accepted 217ms 244.457 MiB
#22 Accepted 230ms 244.648 MiB
#23 Accepted 247ms 244.52 MiB
#24 Accepted 254ms 244.438 MiB
#25 Accepted 238ms 244.629 MiB
#26 Accepted 217ms 244.641 MiB
#27 Accepted 259ms 244.52 MiB
#28 Accepted 258ms 244.551 MiB
#29 Accepted 221ms 244.629 MiB
#30 Accepted 234ms 244.512 MiB
#31 Accepted 242ms 244.551 MiB
#32 Accepted 278ms 255.812 MiB
#33 Accepted 270ms 255.664 MiB
#34 Accepted 226ms 244.27 MiB
#35 Accepted 251ms 244.188 MiB
#36 Accepted 220ms 244.328 MiB
#37 Accepted 221ms 244.27 MiB
#38 Accepted 231ms 244.297 MiB
#39 Accepted 242ms 244.312 MiB
#40 Accepted 255ms 244.375 MiB
#41 Accepted 229ms 244.191 MiB
#42 Accepted 260ms 244.539 MiB
#43 Accepted 246ms 244.504 MiB
#44 Accepted 273ms 244.648 MiB
#45 Accepted 273ms 244.625 MiB
#46 Accepted 235ms 244.449 MiB
#47 Accepted 254ms 244.562 MiB
#48 Accepted 242ms 244.52 MiB
#49 Accepted 225ms 244.625 MiB
#50 Accepted 281ms 244.52 MiB

Code

#include<bits/stdc++.h>
using namespace std;

#define int        long long int
#define pb         push_back
#define all(x)     x.begin(),x.end()
#define allr(x)    x.rbegin(),x.rend()
#define ii         pair<int,int>
#define endl       '\n'

const int N = 1e5 + 5;
int a[N];
int n, k;
int dp[N][305];
vector<int> g[N], ch[N];

void dfs(int u, int p) {
    dp[u][1] = a[u];
    for(auto v : g[u]) {
        if(v == p)
            continue;
        dfs(v, u);
        ch[u].pb(v);
        for(int i = 1 ; i < k ; i++) {
            dp[u][i + 1] = max(dp[u][i + 1], dp[v][i] + a[u]);
        }
    }
}

void pipra(int tc) {
    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].pb(v);
        g[v].pb(u);
    }
    for(int i = 1 ; i <= n ; i++) {
        for(int j = 1 ; j <= k ; j++) {
            dp[i][j] = -1e18;
        }
    }
    dfs(1, 0);
    int ans = 0;
    for(int i = 1 ; i <= n ; i++) {
        if(ch[i].size() < 2) {
            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 - j - 1] + a[i]);
        }
    }
    cout << ans << endl;
}

int32_t main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);

    int t = 1;
    // cin >> t;
    for(int i = 1 ; i <= t ; i++)
        pipra(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-04-12 08:56:18
Judged At
2025-04-12 08:56:18
Judged By
Score
100
Total Time
281ms
Peak Memory
255.812 MiB