ZJOI2007 时态同步

BZOJ

这道题其实就是一个简单的树形DP。

我们可以发现，如果被使用道具的边离\(root\)越近，其影响范围就越大。所以，我们可以以\(DP[i]\)来表示以\(i\)节点为根的子树全部时态同步的最小花费。从底下来看，如果多点为根的子树都已经完成时态同步，这时要将这几棵子树时态同步，又自然要让他们的边权一样，直接统计一下即可。但实际代码中并不需要这个\(DP\)数组，可以直接求值。

#include <bits/stdc++.h>

typedef int mainint;
#define int long long

const int max_n=500000+4;

int N,Ans,root,tot;
int dis[max_n],first_edge[max_n];

struct Edge
{
    int to,w,next_edge;
}edge[max_n<<1];

inline int read()
{
    register int x=0,v=1;
    register char ch=getchar();
    while(!isdigit(ch))
    {
        if(ch=='-') v=-1;
        ch=getchar();
    }
    while(isdigit(ch))
    {
        x=(x<<1)+(x<<3)+ch-'0';
        ch=getchar();
    }
    return x*v;
}

inline void add_edge(int x,int y,int z)
{
    edge[++tot].to=y;
    edge[tot].next_edge=first_edge[x];
    edge[tot].w=z;
    first_edge[x]=tot;
}

void dfs1(int x,int fa)
{
    for(int k=first_edge[x];k;k=edge[k].next_edge)
    {
        if(edge[k].to!=fa)
        {
            dfs1(edge[k].to,x);
            dis[x]=std::max(dis[x],dis[edge[k].to]+edge[k].w);
        }
    }
}

void dfs2(int x,int fa)
{
    for(int k=first_edge[x];k;k=edge[k].next_edge)
    {
        if(edge[k].to!=fa)
        {
            dfs2(edge[k].to,x);
            Ans+=dis[x]-dis[edge[k].to]-edge[k].w;
        }
    }
}

mainint main()
{
    int x,y,z;
    N=read(),root=read();
    for(int i=1;i<N;++i)
    {
        x=read(),y=read(),z=read();
        add_edge(x,y,z),add_edge(y,x,z);    
    }   
    dfs1(root,0),dfs2(root,0);
    printf("%lld\n",Ans);
    return 0;
}

转载于:https://www.cnblogs.com/zcdhj/p/8380026.html