小x游世界树

题目：

小x得到了一个(不可靠的)小道消息，传说中的神岛阿瓦隆在格陵兰海的某处,据说那里埋藏着亚瑟王的宝藏，这引起了小x的好奇，但当他想前往阿瓦隆时发现那里只有圣诞节时才能到达，然而现在已经春天了，不甘心的他将自己的目的地改成了世界树，他耗费了大量的时间，终于将自己传送到了世界树下。世界树是一棵非常巨大的树，它有着许许多多的枝条以及节点，每个节点上都有一个平台。好不容易来到传说中的世界树下，小x当然要爬上去看看风景。小x每经过一条边都会耗费体力值。然而世界树之主想给他弄（gáo）些（dǐan）麻（shì）烦（qíng），于是他在每条边上都设了一个魔法阵，当小x踏上那条边时会被传送回根节点，魔法阵只生效一次。这岂不是要累死小x？幸运的是，每个平台上都有无数个加速器，这些加速器可以让小x在当前节点所连的边上耗费的体力值减少，不同平台的加速器性能不一定相同，但同一个平台的加速器性能绝对相同。世界树之主给了小x一次“换根”的机会，他可以将世界树的任何一个节点变为根，但所有的边都不能改变。小x想问你，将根换为哪个节点能使小x爬到世界树上的每个节点耗费的体力值和最少。默认编号为1的点为初始根。

解题思路：

考虑先以1为根节点求出答案
然后往子节点转移
就是换根

Accepted code:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#define R(i,u) for(register int i=head[u];i;i=nxt[i])
#define F(i,a,b) for(register int i=a;i<=b;i++)
#define add(a,b,c) nxt[++cnt]=head[a],to[cnt]=b,cost[cnt]=c,head[a]=cnt
#define LL long long
using namespace std;

const int N=700010;

int n,cnt,root;
int da[N],nxt[N<<1],to[N<<1],cost[N<<1],head[N],siz[N];
LL sum,ans,f[N];
char buf[1<<20],*p1,*p2;

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

void dfs1(int u,int pre) {
    siz[u]=1;
    R(i,u) {
        int v=to[i];if(v==pre) continue;
        dfs1(v,u);
        siz[u]+=siz[v];
        f[1]+=(LL)max((LL)cost[i]-da[u],0ll)*(LL)siz[v];
    }
}

void dfs2(int u,int pre) {
    R(i,u) {
        int v=to[i]; if(v==pre) continue;
        f[v]=f[u]-(cost[i]-da[u])*siz[v]
             +(cost[i]-da[v])*(n-siz[v]);
        if(f[v]<=f[root]) root=f[v]<f[root]?v:min(root,v);
        dfs2(v,u);
    }
}

int main() {
    n=read();
    F(i,1,n) da[i]=read();
    F(i,1,n-1) {
        int u=read(),v=read(),w=read();
        add(u,v,w),add(v,u,w);
    }
    dfs1(1,0); root=1; dfs2(1,0);
    printf("%d\n%lld",root,f[root]);
    return 0;
}