标签:顶点 树形背包 clu pac 存在 highlight 距离 ext 输入
题目描述
输入
输出
样例输入
10 7
1 2 35129
2 3 42976
3 4 24497
2 5 83165
1 6 4748
5 7 38311
4 8 70052
3 9 3561
8 10 80238
样例输出
184524
题解
树形背包dp
先考虑几个显而易见的性质:
1.选出的点一定是相邻的
2.对于选出的点,如果从$a_k$再走回$a_1$,那么就相当于每条边经过了两次
由于题目没有包含$dis(a_k,a_1)$,因此就相当于选出的点中的一条链可以只经过一次,其余的需要经过两次。
那我们就可以将选点转化为选边,然后考虑树形背包:
设$f[i][j][k]$表示以$i$为根的子树中选择点$i$,共选出$j$条边,且包含的链端点数目为$k$的最小代价。
这里解释一下:当$k=0$时,相当于要从根节点遍历一遍选出的边然后再从根节点出去;$k=1$时,相当于从根节点遍历一遍,到达某链端点后不出去;$k=2$时相当于从某端点遍历到根节点,然后出去再回来到另一端点。
对于根节点与子节点之间的边,显然当$k=0$或$2$时计算两遍,否则计算一遍。
这里第二维和第三维都满足背包性质,然后就可以树形背包了。
时间复杂度$\Theta(4.5n^2)$
#include <cstdio> #include <cstring> #include <algorithm> #define N 3010 using namespace std; int head[N] , to[N << 1] , len[N << 1] , next[N << 1] , cnt , si[N] , f[N][N][3]; inline void add(int x , int y , int z) { to[++cnt] = y , len[cnt] = z , next[cnt] = head[x] , head[x] = cnt; } void dfs(int x , int fa) { int i , j , k , l , m; si[x] = 1 , f[x][0][0] = f[x][0][1] = 0; for(i = head[x] ; i ; i = next[i]) { if(to[i] != fa) { dfs(to[i] , x); for(j = si[x] - 1 ; ~j ; j -- ) for(k = si[to[i]] - 1 ; ~k ; k -- ) for(l = 2 ; ~l ; l -- ) for(m = l ; ~m ; m -- ) f[x][j + k + 1][l] = min(f[x][j + k + 1][l] , f[x][j][l - m] + f[to[i]][k][m] + len[i] * (2 - (m & 1))); si[x] += si[to[i]]; } } } int main() { int n , k , i , j , x , y , z , ans = 1 << 30; scanf("%d%d" , &n , &k); for(i = 1 ; i < n ; i ++ ) scanf("%d%d%d" , &x , &y , &z) , add(x , y , z) , add(y , x , z); memset(f , 0x3f , sizeof(f)); dfs(1 , 0); for(i = 1 ; i <= n ; i ++ ) for(j = 0 ; j <= 2 ; j ++ ) ans = min(ans , f[i][k - 1][j]); printf("%d\n" , ans); return 0; }
标签:顶点 树形背包 clu pac 存在 highlight 距离 ext 输入
原文地址:http://www.cnblogs.com/GXZlegend/p/7559627.html