/ SeriousOJ /

Record Detail

Time Exceeded


  
# Status Time Cost Memory Cost
#1 Accepted 3ms 4.777 MiB
#2 Accepted 6ms 5.277 MiB
#3 Accepted 4ms 8.027 MiB
#4 Accepted 3ms 7.527 MiB
#5 Accepted 3ms 5.293 MiB
#6 Accepted 4ms 7.754 MiB
#7 Accepted 58ms 8.277 MiB
#8 Accepted 20ms 5.074 MiB
#9 Accepted 4ms 6.777 MiB
#10 Accepted 5ms 6.379 MiB
#11 Accepted 5ms 5.027 MiB
#12 Accepted 2ms 5.25 MiB
#13 Accepted 5ms 7.824 MiB
#14 Accepted 52ms 6.527 MiB
#15 Accepted 5ms 7.277 MiB
#16 Accepted 3ms 7.777 MiB
#17 Accepted 4ms 6.277 MiB
#18 Accepted 12ms 7.277 MiB
#19 Time Exceeded ≥4097ms ≥7.75 MiB
#20 Accepted 3861ms 6.883 MiB
#21 Accepted 7ms 7.875 MiB
#22 Accepted 13ms 6.57 MiB
#23 Accepted 4ms 6.566 MiB
#24 Accepted 8ms 7.812 MiB
#25 Time Exceeded ≥4099ms ≥4.742 MiB

Code

/*
 *   Copyright (c) 2025 Emon Thakur
 *   All rights reserved.
 */
#include<bits/stdc++.h>
using namespace std;
using ll = long long;
ll a[100005],n,k,dp[100005][501],dp2[100005][501];
vector<ll> g[100005];

void dpontree(int node,int par)
{
    dp[node][1] = a[node];
    dp[node][0] = 0;
    for(int e:g[node])
    {
        if(e==par) continue;
        dpontree(e,node);
    }

    for(int i=2;i<=k;i++)
    {
        for(int 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]);
        }
    }
}

void rerooting(int node,int par)
{
    for(int i=1;i<k;i++)
    {
        ll mx=0,mx2=0;
        for(auto e:g[node])
        {
            if(e==par || dp[e][i-1]==0) continue;
            if(dp[e][i-1]+a[node] >= mx)
            {
                mx2 = mx;
                mx = dp[e][i-1]+a[node];
            }
            else if(dp[e][i-1]+a[node] > mx2)
            {
                mx2 = dp[e][i-1]+a[node];
            }
        }
        if(dp2[node][i]>=mx) mx2 = mx, mx = dp2[node][i];
        else if(dp2[node][i]>mx2) mx2 = dp2[node][i];

        for(auto e:g[node])
        {
            if(e==par) continue;
            if(dp[node][1]+dp[e][i-1]==mx) dp2[e][i+1] = (mx2)? mx2 + a[e]: 0;
            else dp2[e][i+1] = (mx)? mx + a[e]: 0;

            rerooting(e,node);
        }
    }
}

int main()
{
    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);
    }
    dpontree(1,0);
    dp2[1][1] = a[1];
    rerooting(1,0);
    ll ans = 0;
    for(int i=1;i<=n;i++) ans = max({ans , dp[i][k], dp2[i][k]});
    cout<<ans<<endl;
}

Information

Submit By
Type
Submission
Problem
P1160 Max path sum (Easy Version)
Language
C++17 (G++ 13.2.0)
Submit At
2025-02-12 08:25:55
Judged At
2025-02-12 08:25:55
Judged By
Score
23
Total Time
≥4099ms
Peak Memory
≥8.277 MiB