CF 70E Information Reform

树形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 185
using 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]);
}