かしのブログ

競技プログラミングとか

AtCoder ABC 163 F - path pass i

問題

atcoder.jp

考察

その他のある色 Cについての解を、 全パス - Cを通らないパスとして計算する。  Cを通らないパスとは、色が Cでないノード v, uの単純パス P_{u,v}のうち、 P_{u,v}上に色が Cのノードが存在しないようなものである。 このようなノード v, uのペアの数は色 Cを含まない部分木のうち、色 Cを含まないようにノードを追加することができないようなものを用いて

頂点 rを根とする部分木に含まれるノードの数を |S_r|とする。 色 Cのあるノード vを根とする部分木において、

実装

atcoder.jp