/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 3ms 5.527 MiB
#2 Accepted 3ms 5.035 MiB
#3 Accepted 3ms 5.027 MiB
#4 Accepted 3ms 5.527 MiB
#5 Accepted 3ms 5.254 MiB
#6 Accepted 3ms 5.027 MiB
#7 Accepted 3ms 4.527 MiB
#8 Accepted 3ms 4.277 MiB
#9 Accepted 3ms 4.824 MiB
#10 Accepted 3ms 4.027 MiB
#11 Accepted 3ms 4.078 MiB
#12 Accepted 129ms 236.617 MiB
#13 Accepted 180ms 236.734 MiB
#14 Accepted 227ms 236.664 MiB
#15 Accepted 190ms 236.598 MiB
#16 Accepted 195ms 236.68 MiB
#17 Accepted 132ms 236.766 MiB
#18 Accepted 151ms 236.645 MiB
#19 Accepted 154ms 236.52 MiB
#20 Accepted 121ms 236.715 MiB
#21 Accepted 114ms 236.52 MiB
#22 Accepted 135ms 236.52 MiB
#23 Accepted 135ms 236.562 MiB
#24 Accepted 150ms 236.77 MiB
#25 Accepted 160ms 236.52 MiB
#26 Accepted 103ms 236.77 MiB
#27 Accepted 190ms 236.543 MiB
#28 Accepted 105ms 236.699 MiB
#29 Accepted 116ms 236.523 MiB
#30 Accepted 162ms 236.535 MiB
#31 Accepted 179ms 236.531 MiB
#32 Accepted 219ms 248.867 MiB
#33 Accepted 187ms 248.859 MiB
#34 Accepted 136ms 236.77 MiB
#35 Accepted 144ms 236.723 MiB
#36 Accepted 119ms 236.734 MiB
#37 Accepted 112ms 236.641 MiB
#38 Accepted 144ms 236.77 MiB
#39 Accepted 108ms 236.555 MiB
#40 Accepted 130ms 236.742 MiB
#41 Accepted 133ms 236.582 MiB
#42 Accepted 223ms 236.77 MiB
#43 Accepted 184ms 236.52 MiB
#44 Accepted 171ms 236.77 MiB
#45 Accepted 193ms 236.57 MiB
#46 Accepted 140ms 236.582 MiB
#47 Accepted 241ms 236.598 MiB
#48 Accepted 183ms 236.684 MiB
#49 Accepted 117ms 236.648 MiB
#50 Accepted 345ms 236.605 MiB

Code

/*
 *   Copyright (c) 2025 Emon Thakur
 *   All rights reserved.
 */
#include<bits/stdc++.h>
using namespace std;
#define ll long long
//#define cout file

vector<int> g[100005];
ll a[100005],n,k,dp[100005][301],ans=0;
void dfs(int node,int par)
{
    dp[node][1] = a[node];
    dp[node][0] = 0;
    for(int e:g[node])
    {
        if(e==par) continue;
        dfs(e,node);
    }
    for(int i=2;i<=k;i++)
    {
        for(auto e:g[node])
        {
            if(e==par) continue;
            if(dp[e][i-1]) dp[node][i] = max(dp[node][i] , dp[e][i-1]+a[node]);
        }
    }
    if(g[node].size()==3 || (node==1 && g[1].size()==2))
    {
        vector<int>nodes;
        for(auto e:g[node])
        {
            if(e==par) continue;
            nodes.push_back(e);
        }
        int node1=nodes[0],node2 = nodes[1];
        for(int i=1;i<k-1;i++)
        {
            if(dp[node1][i] && dp[node2][k-i-1])
                dp[node][k] = max(dp[node][k] , dp[node1][i] + a[node] + dp[node2][k-i-1]);
        }
    }
    ans = max(ans , dp[node][k]);
}

int height(int node,int par)
{
    int mx = 0;
    for(auto e:g[node])
    {
        if(e==par) continue;
        mx = max(mx , height(e,node));
    }
    return mx+1;
}

void solve(int tc)
{
    
    // string outp = "output"+to_string(tc)+".txt";
    // string inp = "input"+to_string(tc)+".txt";
    // ofstream file(outp);
    // freopen(inp.c_str(),"r",stdin);
    // memset(dp,0,sizeof(dp));

    ans = 0;
    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);
    }
    dfs(1,0);
    //for(int i=1;i<=n;i++) cout<<dp[i][k]<<endl;
    //cout<<height(1,0)<<endl;
    cout<<ans<<'\n';
    //file<<ans<<'\n';
    //for(int i=1;i<=n;i++) g[i].clear();
}

int main()
{
    ios::sync_with_stdio(false); cin.tie(nullptr);
    // ll x = 32;
    // for(int i=x;i<=x;i++) solve(i);
    solve(0);
}

Information

Submit By
Type
Submission
Problem
P1160 Max path sum (Easy Version)
Language
C++17 (G++ 13.2.0)
Submit At
2025-02-12 09:48:24
Judged At
2025-02-12 09:48:24
Judged By
Score
100
Total Time
345ms
Peak Memory
248.867 MiB