UVALive 4614 Moving to Nuremberg

题目大意：给你一棵树，有一些点你需要去，并且有一定的次数，每去一个点，你都要回来一趟，让你选择一个点，使这个总和最小。

思路：一看数据范围，暴力是肯定不行的。树嘛，那就肯定是树形DP。考虑一个节点 u，它有一个儿子是 v，设 d[ v ] 为以 v 为选择点的总和，因为是考虑来回，那么就读入的时候将所有边权都*2 就好了，那么 d[ v ] = d[ u ] + (sum_time - time[ v ]) *len- time[ v ]*len，len为 u、v 之间的边权（*2的），sum_time为所有的次数和，time[ v ] 为以v为根节点的子树中 mark 点的总次数。若选择 1 号节点为根节点，+上 u 变为 v 要多走的路，-掉 u 变为 v 要少走的路，那么这里需要预处理出 d[ 1 ] 、sum_time和 time[ u ]，很明显，d[ 1 ] 和 time[  u ] 都只要一遍dfs下去，回溯上来加起来就好了，而 sum_time更简单了，把读入加起来，或者就是 sum[ 1 ] （算d[ 1 ] 的时候需要一个辅助数组 sum，表示以 u 为根节点的子树中所有mark 点的路径总和 ）。综上，只要两遍 dfs 即可。

一开始自己想，真的很想出来，因为一直有个思维定势，总想着用 子节点去更新 父节点，这样就搞不出来，子节点的信息不能用在父亲节点上，其实这道题目是用父节点去更新子节点，厉害呀，哈哈，又学习了~~ (*^__^*) 嘻嘻……

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

typedef __int64 lld;

const int MAXN = 50055 ;

struct Edge
{
    int t,next,len;
} edge[MAXN<<1];


int head[MAXN],tot;

void add_edge(int s,int t,int len)
{
    edge[tot].t = t;
    edge[tot].len = len*2;
    edge[tot].next = head[s];
    head[s] = tot ++ ;
}

lld sum[MAXN],time[MAXN];

void dfs1(int u,int fa)
{
    sum[u] = 0;
    for(int e = head[u];e!=-1;e = edge[e].next)
    {
        int v = edge[e].t;
        int len = edge[e].len;
        if(v == fa) continue;
        dfs1(v,u);
        sum[u] += time[v]*len + sum[v];
        time[u] += time[v];
    }
    //printf("u = %d,time = %d,sum = %d\n",u,time[u],sum[u]);
}

int sum_time;

lld d[MAXN];

void dfs2(int u,int fa)
{
    for(int e = head[u];e!=-1;e = edge[e].next)
    {
        int v = edge[e].t;
        int len = edge[e].len;
        if(v == fa) continue;
        d[v] = d[u] + (sum_time - time[v] - time[v]) * len;
        dfs2(v,u);
    }
}

int ans[MAXN];

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        tot=0;
        memset(head,-1,sizeof(head));
        for(int i = 1;i<n;i++)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            add_edge(a,b,c);
            add_edge(b,a,c);
        }
        int m;
        scanf("%d",&m);
        memset(time,0,sizeof(time));
        sum_time = 0;
        for(int i = 1;i<=m;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            time[a] = b;
            sum_time += b;
        }
        dfs1(1,-1);
        d[1] = sum[1];
        dfs2(1,-1);
        int c = 0;
        lld minn = sum[1] + 1;
        for(int i = 1;i<=n;i++)
        {
            if(d[i] == minn)
            {
                ans[c++] = i;
            }
            else if(d[i] < minn)
            {
                minn = d[i];
                c = 0;
                ans[c ++] = i;
            }
        }
        printf("%I64d\n",minn);

        for(int i = 0;i<c;i++)
            if(i == 0) printf("%d",ans[i]);
            else printf(" %d",ans[i]);
        puts("");
    }
    return 0;
}