AtCoder Beginner Contest 207 F Tree Patrolling题解

题意：
给定一棵树，共有n个节点，你可以放任意数量的保安在树的节点上面，保安能看守他放置的点和该点的直接父节点和直接子节点，问对于能看守0~n个节点的放置方案有多少？

思路：
树上背包优化。设dp方程为：$f[0/1/2][i][j]$为以1为根节点，没放置在i节点点，且i没被i的子节点看守 / 放置在i节点 / 没放置在i节点上，但i已经被子节点看守，同时在i节点的子树上看守了j个点的方案数,$sz[i]$为i节点子树的节点个数,转移方程为
f [ 0 ] [ i ] [ j ] = ∑ u ⊂ s o n ( i ) ∑ k = 0 s z [ u ] f [ 0 ] [ u ] [ k ] ∗ f [ 0 ] [ i ] [ j − k ] + f [ 2 ] [ u ] [ k ] ∗ f [ 0 ] [ i ] [ j − k ] f [ 1 ] [ i ] [ j ] = ∑ u ⊂ s o n ( i ) ∑ k = 0 s z [ u ] f [ 1 ] [ u ] [ k ] ∗ f [ 1 ] [ i ] [ j − k ] + f [ 2 ] [ u ] [ k ] ∗ f [ 1 ] [ i ] [ j − k ] + f [ 0 ] [ u ] [ k ] ∗ f [ 1 ] [ i ] [ j − k − 1 ] f [ 2 ] [ i ] [ j ] = ∑ u ⊂ s o n ( i ) ∑ k = 0 s z [ u ] f [ 1 ] [ u ] [ k ] ∗ f [ 2 ] [ i ] [ j − k ] + f [ 0 ] [ u ] [ k ] ∗ f [ 2 ] [ i ] [ j − k ] + f [ 2 ] [ u ] [ k ] ∗ f [ 2 ] [ i ] [ j − k ] + f [ 1 ] [ u ] [ k ] ∗ f [ 0 ] [ i