committee - 委員会 (Committee) Editorial by Mitsubachi
人を頂点に見立て、頂点 $s_i$ から頂点 $i$ へ辺を張ったグラフを考えます。制約より、これは頂点 $1$ を根とする根付き木となります。
ここで、以下のような $dp$ を考えます。
頂点 $i$ が葉の場合、頂点 $i$ の子は存在しない(人 $i$ の部下はいない)ので $dp[i]=a_i$ となります。 そうでない場合、 $dp[i]$ は以下の式のようになります。
DFSをすることによりこの $dp$ は $O(n)$ で解くことができます。
posted:
last update: