问题：

一棵树，每个点有一个权值 $a_i$ ( $a_i$ 是0/1)

求：以每个点为根时，大小最小的权值为1（$<=>$子树的根节点权值为1，其他点权值为0）的子树大小

感觉应该能换根，挺典的但我太菜没换出来，复杂度尽量O(n),感谢高手帮助！！！