思路:因为树是特殊的图,故树上距离可以用图的方法求解。
本题数据量小(1<=n<=100),直接使用Floyed即可。当然也可以BFS。方法为:进行n次BFS,求出以每个点为起点的最短路,再计算以每个点为终点的最短路。代码如下:
int dist[M][M];
inline void bfs(int st) {
bool *vis=new bool[n+1]();
queue<int> q;
q.push(st);
while(!q.empty()) {
int now=q.front();
q.pop();
vis[now]=1;
for(int i=fir[now]; ~i; i=G[i].next) {
int to=G[i].to;
if(!vis[to]) {
dist[st][to]=min(dist[st][to],dist[st][now]+1);
q.push(to);
}
}
}
}
inline int calc_sum(int ter) {
int res=0;
for(int i=1; i<=n; i++) {
if(i!=ter) res+=dist[i][ter];
}
return res;
}
int main() {
static int k=0,wei[M],res[M]; //有数组,应设为static
scanf("%d",&n);
fill(fir+1,fir+n+1,-1);
//省略建图代码
for(int i=1; i<=n; i++) bfs(i);
for(int i=1; i<=n; i++) {
for(int j=1; j<=n; j++) {
dist[i][j]*=wei[i];
}
}
for(int i=1; i<=n; i++) res[i]=calc_sum(i);
printf("%d",*min_element(res+1,res+n+1));
return 0;
}