【模板】求以每个点出发的最长路径和最短路径

void dfs(int u,int fa){
    f[u]=f2[u]=g[u]=g2[u]=0;
    for(int i=head[u];i+1;i=e[i].nxt){
        int v=e[i].v,w=e[i].w;
        if(v==fa) continue;
        dfs(v,u);
        if(f[v]+w>=f[u]){
            g[u]=f[u];
            f[u]=f[v]+w;
        }
        else if(f[v]+w>g[u]) g[u]=f[v]+w;
        
        if(f2[v]+w<=f2[u]){
            g2[u]=f2[u];
            f2[u]=f2[v]+w;
        }
        else if(f2[v]+w<g2[u]) g2[u]=f2[v]+w;
    }
}

void dp(int u,int fa,ll dis,ll dis2,ll &d,ll &d2){
    f[u]=max(f[u],dis);
    f2[u]=min(f2[u],dis2);
    for(int i=head[u];i+1;i=e[i].nxt){
        if(cut[i]) continue;
        int v=e[i].v,w=e[i].w;
        if(v==fa) continue;
        if(f[u]==f[v]+w&&f2[u]==f2[v]+w)
            dp(v,u,w+max(dis,g[u]),w+min(dis,g2[u]),d,d2);
        else if(f[u]==f[v]+w)
            dp(v,u,w+max(dis,g[u]),w+min(dis,f2[u]),d,d2);
        else if(f2[u]==f2[v]+w)
            dp(v,u,w+max(dis,f[u]),w+min(dis,g2[u]),d,d2);
        else dp(v,u,w+max(dis,f[u]),w+min(dis,f2[u]),d,d2);
    }
}

转载于:https://www.cnblogs.com/White-star/p/11603319.html