HAOI2015[BZOJ2110] T1 【树形DP】

4033: [HAOI2015]树上染色

Time Limit: 10 Sec   Memory Limit: 256 MB
Submit: 1959   Solved: 840
[ Submit][ Status][ Discuss]

Description

有一棵点数为N的树，树边有边权。给你一个在0~N之内的正整数K，你要在这棵树中选择K个点，将其染成黑色，并

将其他的N-K个点染成白色。将所有点染色后，你会获得黑点两两之间的距离加上白点两两之间距离的和的收益。

问收益最大值是多少。

Input

第一行两个整数N,K。

接下来N-1行每行三个正整数fr,to,dis，表示该树中存在一条长度为dis的边(fr,to)。

输入保证所有点之间是联通的。

N<=2000,0<=K<=N

Output

输出一个正整数，表示收益的最大值。

Sample Input

5 2
1 2 3
1 5 1
2 3 1
2 4 2

Sample Output

17

HYSBZ - 4033

题意：见上文

分析：树形DP做少啦，开始一点思路也没有。后来看大佬博客，有个很关键的一点，用f[i][j]表示i节点中有j个黑点对答案的贡献。这样思路就很清晰啦。

          因为对于一条边len(设这条边连接着x和y，且x是y的父节点)，他对答案的贡献可以由y子树的黑点数量求解出来。

           对于一棵树上很多点的距离和，可以由边的“使用”次数*边的权值。边·的次数也比较方便进行dp操作。

           [(无视)dp[x][0]和dp[x][1]分别表示x点是否为黑点，十分方便]

ACcode

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <vector>
#include <queue>
#include <map>
#include <algorithm>
using namespace std;
typedef long long LL;
const int INF = 1e9 + 5;
const int N = 2020;
int head[N*2],n,m,sum[N],num;
LL dp[N][N];
struct edge
{
    int to,nex,len;
}e[N*2];
void add(int from,int to,int cap)
{
    e[num].to=to;
    e[num].len=cap;
    e[num].nex=head[from];
    head[from]=num++;
}
void dfs(int x,int fa)
{
    dp[x][0]=dp[x][1]=0;
    sum[x]=1;
    for(int i=head[x];i!=-1;i=e[i].nex)
    {
        int y=e[i].to;
        if(y==fa)continue;
        dfs(y,x);
        sum[x]+=sum[y];
        for(int v=min(m,sum[x]);v>=0;v--)
        {
            for(int j=0;j<=min(v,sum[y]);j++)
            {
                LL res=(LL)j*(m-j)+(LL)(sum[y]-j)*(n-sum[y]-m+j);
                res*=e[i].len;
                dp[x][v]=max(dp[x][v],dp[x][v-j]+dp[y][j]+res);
            }
        }
    }
}
int main()
{
    int x,y,z;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(head,-1,sizeof(head));
        num=1;
        for(int i=1;i<n;i++)
        {
            scanf("%d%d%d",&x,&y,&z);
            add(x,y,z);
            add(y,x,z);
        }
        memset(dp,128,sizeof(dp));
        dfs(1,-1);
        LL ans=dp[1][m];
        printf("%lld\n",ans);
    }
    return 0;
}