题意:有一棵树,每个节点有颜色,要找出最小的连通块使得其中的点至少有$k$种不同的颜色,只需输出这个最小连通块的大小
因为$k$很小,所以如果颜色只有$k$种,我们可以直接状压DP,设$f_{i,j}$表示在$i$的子树中包含颜色集合为$j$的最小连通块大小(这个连通块必须包含$i$),那么可以枚举$s$的子集$t$,转移即为$f_{x,s}\gets f_{x,t}+f_{u,s-t}\left(u\in son_x\right)$
但颜色最多有$n$种,这时候我们就可以trick一下:随机地把$n$种颜色映射到$k$种颜色再DP,随机多几次错误的概率就很低了
......但我第一次交还是WA了啊==同样的代码再交一次就A了(论一个非洲人的自我修养)
#include#include #include #include int min(int a,int b){return a