Description
给出一棵 个节点的树,边权均为 ,要求在其中一些点建立信息中转站,距离中转站 的点收到信息的代价为 ,设置一个中转站的代价是 ,问如何设置可以使得总代价最小
Input
第一行两个整数 表示点数和设置一个中转站的代价,之后输入 个整数 表示不同距离接受信息的代价,最后输入 条树边
Output
输出最小代价以及最小代价下每点获取信息的中转站位置
Sample Input
8 10
2 5 9 11 15 19 20
1 4
1 3
1 7
4 6
2 8
2 3
3 5
Sample Output
38
3 3 3 4 3 4 3 3
Solution
以 表示在 点建立一个中转站之后 点以及子树得到信息的代价最小值,令 表示使得 最小的 ,首先跑遍最短路得到任意两点距离最小值 ,那么在不考虑 子树时有 ,之后考虑 儿子节点 ,那么对于 而言,要么从 获取信息,代价为 ,要么从 获取信息,这样的话省去了多建立一个中转站的代价,此时代价为 ,两者选较小值累加到 中,在求出所有 后即得到 ,最后 即为最小代价和,具体方案只需比较 和 的大小就可以知道 点从哪一点获取的信息
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=185;
int n,K,d[maxn],dis[maxn][maxn],m[maxn][maxn],dp[maxn][maxn],ans[maxn],path[maxn];
void dfs(int u,int fa)
{
for(int i=1;i<=n;i++)dp[u][i]=d[dis[u][i]]+K;
for(int v=1;v<=n;v++)
if(m[u][v]&&v!=fa)
{
dfs(v,u);
for(int i=1;i<=n;i++)dp[u][i]+=min(dp[v][ans[v]],dp[v][i]-K);
}
ans[u]=1;
for(int i=2;i<=n;i++)
if(dp[u][i]<dp[u][ans[u]])ans[u]=i;
}
void output(int u,int fa,int i)
{
path[u]=i;
for(int v=1;v<=n;v++)
if(m[u][v]&&v!=fa)
{
if(dp[v][ans[v]]<dp[v][i]-K)output(v,u,ans[v]);
else output(v,u,i);
}
}
int main()
{
scanf("%d%d",&n,&K);
for(int i=1;i<n;i++)scanf("%d",&d[i]);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(i!=j)dis[i][j]=INF;
else dis[i][j]=0;
for(int i=1;i<n;i++)
{
int u,v;
scanf("%d%d",&u,&v);
m[u][v]=m[v][u]=1;
dis[u][v]=dis[v][u]=1;
}
for(int k=1;k<=n;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
dfs(1,0);
printf("%d\n",dp[1][ans[1]]);
output(1,0,ans[1]);
for(int i=1;i<=n;i++)printf("%d%c",path[i],i==n?'\n':' ');
return 0;
}