[dfs] Jzoj P1497 景点中心

Description

话说宁波市的中小学生在镇海中学参加计算机程序设计比赛，比赛之余，他们在镇海中学的各个景点参观。镇海中学共有n个景点，每个景点均有若干学生正在参观。这n个景点以自然数1至n编号，每两个景点的编号均不同。每两个景点之间有且只有一条路径。选择哪个景点集中的学生，才能使所有学生走过的路径之和最小呢？

 

Input

输入文件center.in中有若干行：
第一行只有一个正整数n，表示景点数。
第二行有n个1至1000间的整数，这n个整数间互相以一个空格分隔。其中第i个整数表示第i个景点处的学生数。
第三行至第n+1行，每行有三个整数I,j,k，表示景点i和景点j之间有一条长尾k的路径直接连接。其中i<>j,1≤i≤n,1≤j≤n;1≤k≤1000。

Output

输出文件center.out中有二行；
第一行只有一个整数i，表示在第i个景点处集中时，所有学生走过的路径之和最短。
第二行也只有一个整数，表示所有学生走过的路径之和的最小值。

 

Sample Input

4
3  2  4  1
1  2  5
3  1  6
2  4  4

Sample Output

1
43

 

Hint

【数据限制】
所有的数据均随机生成，且满足：
30%的数据，1≤n≤200。
60%的数据，1≤n≤3000。
100%的数据，1≤n≤100000。

题解

• 先可以定i为根，用dfs求出从i到根结点的价值和子树大小
• 那么求完之后就可以求出所有点到i的距离
• 设在A处集合时，所有人走的距离的和为B
• 若C是A的儿子则所有人在C处集合的走的距离是C=B+(该边权值)*(总人数-2*size[C])

代码

#include<cstdio>
#include<iostream>
using namespace std;
struct edge {int to,from,v; }e[100010*2];
int head[100010],n,cnt=1;
long long size[100010],dis[100010],mn,ans;
void insert(int x,int y,int v) { e[cnt].to=y; e[cnt].v=v; e[cnt].from=head[x]; head[x]=cnt++; }
void dfs(int x,int fa)
{
    for (int i=head[x];i;i=e[i].from)
        if (e[i].to!=fa)
        {
            dfs(e[i].to,x);
            size[x]+=size[e[i].to];
            dis[x]+=dis[e[i].to]+size[e[i].to]*e[i].v;
        }
}
void dfs1(int x,int fa)
{
    for (int i=head[x];i;i=e[i].from) 
        if (e[i].to!=fa)
        {
            dis[e[i].to]=dis[x]+e[i].v*(size[1]-2*size[e[i].to]);  
            dfs1(e[i].to,x);
        }
}
int main()    
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%lld",&size[i]);
    for (int i=1;i<=n-1;i++)
    {
        int x,y,v;
        scanf("%d%d%d",&x,&y,&v);
        insert(x,y,v); insert(y,x,v);
    }
    dfs(1,0);
    dfs1(1,0);
    mn=10000000000000;
    for (int i=1;i<=n;i++)
        if (mn>dis[i]) 
        {
            mn=dis[i];
            ans=i;
        }
    printf("%lld
%lld",ans,mn);
    return 0;
}