P6869 [COCI2019-2020#5] Putovanje(LCA)

 P6869 [COCI2019-2020#5] Putovanje(LCA)

P6869 [COCI2019-2020#5] Putovanje(LCA) 

P6869 [COCI2019-2020#5] Putovanje(LCA) 

const int N=2e5+5;

    int n,m,_;
    int i,j,k;
    int a[N];
    int son[N],sz[N],fa[N],dep[N];
    int top[N];
    struct Node
    {
        int to,c1,c2;
        Node(int to=0,int a=0,int b=0):to(to),c1(a),c2(b){}
    };
    vector<Node> G[N];
    ll c1[N],c2[N],d[N];

void dfs1(int u,int f)
{
    dep[u]=dep[f]+1;
    fa[u]=f;
    sz[u]=1;
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i].to;
        if(v==f) continue;
        c1[v]=G[u][i].c1;
        c2[v]=G[u][i].c2;
        dfs1(v,u);
        if(sz[son[u]]<sz[v]) son[u]=v;
    }
}

void dfs2(int u,int f)
{
    top[u]=f;
    if(son[u]) dfs2(son[u],f);
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i].to;
        if(v==son[u] || v==fa[u]) continue;
        dfs2(v,v);
    }
}

#define nx top[x]
#define ny top[y]

int LCA(int x,int y)
{
    while(nx!=ny){
        if(dep[nx]<dep[ny]) swap(x,y);
        x=fa[nx];
    }
    if(dep[x]>dep[y]) swap(x,y);
    return x;
}

void dfs(int u)
{
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i].to;
        if(v==fa[u]) continue;
        dfs(v);
        d[u]+=d[v];
    }
}

int main()
{
    while(~sd(n)){
        int x,y,o,oo;
        for(int i=1;i<n;i++){
            sdd(x,y);
            sdd(o,oo);
            G[x].pb(Node(y,o,oo));
            G[y].pb(Node(x,o,oo));
        }
        dfs1(1,0); dfs2(1,1);
        for(int i=1;i<n;i++){
            d[i]++; d[i+1]++;
            d[LCA(i,i+1)]-=2;
        }
        dfs(1);
        ll ans=0;
        for(int i=2;i<=n;i++){
            if(c1[i]*d[i]>c2[i]) ans+=c2[i];
            else ans+=c1[i]*d[i];
        }
        pll(ans);
    }
    //PAUSE;
    return 0;
}

 

上一篇:Golang 浮点类型、字符类型


下一篇:JVM的编译方式及运行效率