http://acm.xmu.edu.cn/JudgeOnline/problem.php?id=1344
厦门大学在翔安也有了个新校区,新校区有n栋宿舍楼。由于地处偏僻,道路有限,只有n-1条路,每条路都是双向的,连接两栋宿舍楼,任意两栋楼都有且仅有一条路径互相到达。开发商想在校区内建一座食堂,食堂不能建在道路上,必须是在某栋宿舍楼。由于食堂实在很好吃,所有学生都会走路来这儿吃饭或买饭。为了便民,避免不必要的多走路,开发商希望食堂所在的宿舍楼到其他各个宿舍楼的距离之和最小,求这个最小值。
经典问题。先建树,这里vector果然好用。g[i]保存点i所有子节点到i点的距离之和。s[i]保存以该点为根的树的节点数,然后
f[x]=f[fa]+l[ln].l1*(n-2*s[x]);
DFS利用父节点得到的数据计算子节点的数据。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<vector>
#include<algorithm>
#include<map>
#include<cmath>
using namespace std;
#define M 1000000000
#define N 10005
struct line
{
int a,b,l1;
}l[2*N];
vector<int> p[N],p2[N];
int g[N],f[N],n,s[N];
bool tp[N];
int dfs1(int x)
{
tp[x]=false;
int ans=0;
unsigned int i;
bool h=false;
for (i=0;i<p2[x].size();i++)
{
int bb=l[p2[x][i]].b;
if (tp[bb]==true)
{
g[bb]=dfs1(bb);
ans+=g[bb]+l[p2[x][i]].l1*s[bb];
s[x]+=s[bb];
h=true;
p[x].push_back(p2[x][i]);
}
}
if (h==false) s[x]=1;
else s[x]+=1;
return ans;
}
void dfs2(int fa,int x,int ln)
{
tp[x]=false;
unsigned int i;
if (x!=1) f[x]=f[fa]+l[ln].l1*(n-2*s[x]);
for (i=0;i<p[x].size();i++)
dfs2(x,l[p[x][i]].b,p[x][i]);
}
int main()
{
//freopen("a","r",stdin);
int i;
n=1;
scanf("%d",&n);
while(n!=0)
{
int h,k,ss;
for (i=1;i<=n;i++)
{
tp[i]=true;
s[i]=0;
p2[i].clear();
p[i].clear();
}
for (i=1;i<=n-1;i++)
{
scanf("%d%d%d",&h,&k,&ss);
p2[h].push_back(i);
l[i].a=h;
l[i].b=k;
l[i].l1=ss;
p2[k].push_back(i+n-1);
l[i+n-1].b=h;
l[i+n-1].a=k;
l[i+n-1].l1=ss;
}
g[1]=dfs1(1);
f[1]=g[1];
for (i=1;i<=n;i++) tp[i]=true;
tp[1]=false;
dfs2(0,1,0);
int ans=M;
for (i=1;i<=n;i++)
if (f[i]<ans) ans=f[i];
printf("%d\n",ans);
scanf("%d",&n);
}
return 0;
}