CF 70E Information Reform

来源：互联网发布：五子棋最强软件编辑：程序博客网时间：2024/05/17 23:52

树形DP

因为博主蒟蒻变懒了，以后就只写一些不好想的(丧)题题解，而不会啥题解都写了吧。。。

注意到每一个点都要找一个点和自己对应，并产生代价，于是可以记f[i][j]表示i强制对应j时i子树的最小答案(包括建j的k代价)，再记一个best[i]表示j取到best[i]时，f[i][j]最小

f[i][j] = Sigma { min( f[s][j] - k, f[s][best[s]] ) }

这样第一问就做完了233333

考虑第二问，如果以1为根进行DP的话，1的对应点肯定就是best[i]，对于一个儿子s，s的对应点要么是best[s]要么是best[i]，比较一下即可。

#include<cstdio>#include<cstring>#include<algorithm>#define N 185using namespace std;int f[N][N], best[N], dis[N][N], cost[N], n, k, d[N][N], ans[N];void floyd(){    for(int i = 1; i <= n; i++)        dis[i][i] = 0;    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]);}void dp(int x, int fa){    for(int y = 1; y <= n; y++)if(d[x][y])        if(y!=fa)dp(y,x);    for(int i = 1, minn = 1<<30; i <= n; i++)    {        f[x][i] = cost[dis[x][i]] + k;        for(int y = 1; y <= n; y++)if(d[x][y])            if(y!=fa)f[x][i] += min(f[y][i]-k, f[y][best[y]]);        if(f[x][i] < minn)        {            minn = f[x][i];            best[x] = i;        }    }}void make(int x, int fa){    for(int y = 1; y <= n; y++)if(d[x][y])        if(y!=fa)        {            if(f[y][best[y]] < f[y][ans[x]]-k)                ans[y] = best[y];            else ans[y] = ans[x];            make(y,x);        }}int main(){    scanf("%d%d",&n,&k);    for(int i = 1; i < n; i++)        scanf("%d",&cost[i]);    memset(dis,63,sizeof(dis));    for(int i = 1, a, b; i < n; i++)    {        scanf("%d%d",&a,&b);        d[a][b] = d[b][a] = true;        dis[a][b] = dis[b][a] = 1;    }    floyd();    dp(1,0);    ans[1] = best[1];    make(1,0);    printf("%d\n",f[1][best[1]]);    for(int i = 1; i <= n; i++)        printf("%d ",ans[i]);}

0 0