BJ模拟 tree【树形dp】

题目描述：

给一棵 $n$ 个节点的边带权树，要求在树上选出 $k$ 个点 $a_1,a_2,...,a_k$，使得 $\sum_{i=1}^{k-1}dis(a_i,a_{i+1})$最小。$1\le k\le n\le 3000$

解题思路：

首先答案一定是一个连通子树，且为2*子树边权和-子树直径长度。

这样就比较好dp了。

$g[x][i]$ 表示以 $x$ 为根的子树中选了$i$个点且包含$x$（下面都是这样）的最小边权和的两倍，$f0[x][i]$ 表示直径一端为 $x$，$f1[x][i]$ 表示 $x$不是直径一端的最优值，按定义转移即可，dp性质可以保证选的直径确实是真直径。

时间复杂度为$O(n^2)$

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')c=getchar(),f=-1;
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=3005,INF=0x3f3f3f3f;
int n,k,ans=INF,size[N],g[N][N],f0[N][N],f1[N][N];
int tot,first[N],nxt[N<<1],to[N<<1],w[N<<1];
void add(int x,int y,int z)
{
    nxt[++tot]=first[x],first[x]=tot,to[tot]=y,w[tot]=z;
}
void checkmin(int &x,int y){x=x<=y?x:y;}
void dfs(int u,int fa)
{
    size[u]=1;g[u][1]=f0[u][1]=f1[u][1]=0;
    for(int e=first[u];e;e=nxt[e])
    {
        int v=to[e];if(v==fa)continue;
        dfs(v,u);
        for(int i=size[u];i;i--)
            for(int j=size[v];j;j--)
            {
                checkmin(g[u][i+j],g[u][i]+g[v][j]+w[e]*2);
                checkmin(f0[u][i+j],f0[u][i]+g[v][j]+w[e]*2);
                checkmin(f0[u][i+j],g[u][i]+f0[v][j]+w[e]);
                checkmin(f1[u][i+j],f1[u][i]+g[v][j]+w[e]*2);
                checkmin(f1[u][i+j],g[u][i]+f1[v][j]+w[e]*2);
                checkmin(f1[u][i+j],f0[u][i]+f0[v][j]+w[e]);
            }
        size[u]+=size[v];
    }
}
int main()
{
    //freopen("lx.in","r",stdin);
    memset(g,INF,sizeof(g));
    memset(f0,INF,sizeof(f0));
    memset(f1,INF,sizeof(f1));
    n=getint(),k=getint();
    for(int i=1;i<n;i++)
    {
        int x=getint(),y=getint(),z=getint();
        add(x,y,z),add(y,x,z);
    }
    dfs(1,0);
    for(int i=1;i<=n;i++)if(size[i]>=k)ans=min(ans,min(f0[i][k],f1[i][k]));
    cout<<ans<<'\n';
    return 0;
}