/ SeriousOJ /

Record Detail

Memory Exceeded


  
# Status Time Cost Memory Cost
#1 Accepted 4ms 7.777 MiB
#2 Accepted 3ms 7.0 MiB
#3 Accepted 3ms 7.754 MiB
#4 Accepted 3ms 7.777 MiB
#5 Accepted 4ms 8.047 MiB
#6 Accepted 3ms 7.047 MiB
#7 Accepted 3ms 5.027 MiB
#8 Accepted 4ms 6.277 MiB
#9 Accepted 4ms 6.727 MiB
#10 Accepted 3ms 5.027 MiB
#11 Accepted 4ms 7.297 MiB
#12 Accepted 3ms 6.527 MiB
#13 Accepted 3ms 6.277 MiB
#14 Accepted 4ms 6.328 MiB
#15 Accepted 4ms 6.277 MiB
#16 Accepted 3ms 6.777 MiB
#17 Accepted 4ms 7.527 MiB
#18 Accepted 3ms 5.027 MiB
#19 Accepted 4ms 8.027 MiB
#20 Accepted 4ms 8.004 MiB
#21 Accepted 3ms 4.777 MiB
#22 Accepted 4ms 8.027 MiB
#23 Accepted 3ms 6.297 MiB
#24 Accepted 3ms 5.027 MiB
#25 Accepted 3ms 7.277 MiB
#26 Accepted 4ms 7.777 MiB
#27 Accepted 4ms 5.324 MiB
#28 Accepted 4ms 6.574 MiB
#29 Accepted 5ms 7.57 MiB
#30 Accepted 5ms 7.57 MiB
#31 Accepted 5ms 7.719 MiB
#32 Accepted 5ms 8.062 MiB
#33 Accepted 4ms 4.824 MiB
#34 Accepted 5ms 7.574 MiB
#35 Accepted 5ms 7.57 MiB
#36 Accepted 5ms 6.348 MiB
#37 Accepted 4ms 4.82 MiB
#38 Accepted 5ms 6.316 MiB
#39 Accepted 5ms 6.426 MiB
#40 Accepted 5ms 7.312 MiB
#41 Accepted 5ms 6.316 MiB
#42 Accepted 5ms 6.418 MiB
#43 Accepted 5ms 7.777 MiB
#44 Accepted 3ms 6.777 MiB
#45 Accepted 4ms 6.277 MiB
#46 Accepted 2ms 5.027 MiB
#47 Accepted 3ms 7.836 MiB
#48 Accepted 3ms 8.324 MiB
#49 Accepted 3ms 5.027 MiB
#50 Accepted 4ms 8.027 MiB
#51 Accepted 4ms 7.867 MiB
#52 Accepted 3ms 7.027 MiB
#53 Accepted 3ms 7.777 MiB
#54 Accepted 3ms 7.121 MiB
#55 Accepted 3ms 7.027 MiB
#56 Accepted 3ms 6.574 MiB
#57 Accepted 4ms 7.508 MiB
#58 Accepted 3ms 7.277 MiB
#59 Accepted 4ms 7.777 MiB
#60 Accepted 4ms 8.027 MiB
#61 Accepted 3ms 5.277 MiB
#62 Accepted 3ms 5.027 MiB
#63 Accepted 3ms 6.574 MiB
#64 Accepted 3ms 7.527 MiB
#65 Accepted 3ms 7.277 MiB
#66 Accepted 5ms 7.957 MiB
#67 Accepted 5ms 7.777 MiB
#68 Accepted 5ms 7.777 MiB
#69 Accepted 3ms 7.539 MiB
#70 Accepted 3ms 7.316 MiB
#71 Accepted 3ms 6.277 MiB
#72 Accepted 3ms 7.5 MiB
#73 Accepted 3ms 8.027 MiB
#74 Accepted 3ms 7.074 MiB
#75 Accepted 3ms 4.82 MiB
#76 Accepted 3ms 6.027 MiB
#77 Accepted 3ms 6.027 MiB
#78 Accepted 3ms 6.527 MiB
#79 Accepted 3ms 6.07 MiB
#80 Accepted 3ms 7.027 MiB
#81 Accepted 3ms 7.777 MiB
#82 Accepted 3ms 7.527 MiB
#83 Accepted 3ms 6.301 MiB
#84 Accepted 3ms 6.754 MiB
#85 Accepted 3ms 7.574 MiB
#86 Accepted 3ms 4.859 MiB
#87 Accepted 3ms 5.945 MiB
#88 Accepted 3ms 7.777 MiB
#89 Accepted 3ms 6.793 MiB
#90 Accepted 3ms 5.277 MiB
#91 Accepted 2ms 6.527 MiB
#92 Accepted 295ms 773.102 MiB
#93 Accepted 298ms 773.09 MiB
#94 Accepted 297ms 773.133 MiB
#95 Accepted 296ms 773.152 MiB
#96 Accepted 302ms 773.121 MiB
#97 Accepted 252ms 773.273 MiB
#98 Accepted 291ms 773.27 MiB
#99 Accepted 279ms 801.52 MiB
#100 Memory Exceeded ≥780ms ≥1.0 GiB

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


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

int main()
{
    ios::sync_with_stdio(false); cin.tie(nullptr);
    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<<'\n';
}

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:51:20
Judged At
2025-02-12 08:51:20
Judged By
Score
99
Total Time
≥780ms
Peak Memory
≥1.0 GiB