Address
Solution
简版题意:在一棵 n n n 个节点的边带权树上删掉 k k k 条边再加上 k k k 条权为 0 0 0 的边组成一棵新树,最大化新树的直径。
容易发现,如果删掉的 k k k 条边给定,那么答案就是这 k + 1 k+1 k+1 个连通块的直径之和。
于是问题转化成在树上选出 k + 1 k+1 k+1 条不相交的路径的最大权值和。点视为退化的路径。
树形 DP : f [ u ] [ i ] [ 0 / 1 / 2 ] f[u][i][0/1/2] f[u][i][0/1/2] 表示 u u u 的子树内选出 i i i 条路径,根的度数为 0 0 0 或 1 1 1 或 2 2 2 的最大收益。
注:特殊情况:如果 u u u 没有被选出则度数为 0 0 0 ,如果 u u u 单独作为一条路径则度数为 1 1 1 。
转移即枚举子节点 v v v 进行各种讨论,设 f ′ f' f′ 为 v v v 之前的 DP 数组。下面我们先不考虑 u u u 单独作为一条路径的情况。
f [ u ] [ i + j ] [ s ] = max ( f [ u ] [ i + j ] [ s ] , f ′ [ u ] [ i ] [ s ] + f [ v ] [ j ] [ t ] ) f[u][i+j][s]=\max(f[u][i+j][s],f'[u][i][s]+f[v][j][t]) f[u][i+j][s]=max(f[u][i+j][s],f′[u][i][s]+f[v][j][t])
其中 s s s 和 t t t 为 [ 0 , 2 ] ∩ Z [0,2]\cap\Z [0,2]∩Z 内任意数。
f [ u ] [ i + j ] [ 1 ] = max ( f [ u ] [ i + j ] [ 1 ] , f ′ [ u ] [ i ] [ 0 ] + f [ v ] [ j ] [ 1 ] + l e n ( u , v ) ) f[u][i+j][1]=\max(f[u][i+j][1],f'[u][i][0]+f[v][j][1]+len(u,v)) f[u][i+j][1]=max(f[u][i+j][1],f′[u][i][0]+f[v][j][1]+len