atcoder#ARC150D. [ARC150D] Removing Gacha
[ARC150D] Removing Gacha
配点 : 点
問題文
頂点に から の番号がついた 頂点の根付き木があります。頂点 はこの木の根であり、頂点 の親頂点は頂点 です。
各頂点は白、黒の色を持っています。はじめすべて頂点の色は白です。
根付き木において、頂点 を結ぶ唯一の単純パス上の頂点 (頂点 含む) の色がすべて黒であるとき、頂点 を「よい頂点」といいます。また、「よい頂点」ではない頂点を「わるい頂点」といいます。
すべての頂点の色が黒になるまで「『わるい頂点』から一様ランダムに頂点を つ選び、その頂点を黒色で上塗りする」という操作を行います。
操作を行う回数の期待値を で求めてください。
期待値 $\text{mod }{998244353}$ の定義
求める期待値は必ず有理数になることが証明できます。 また、この問題の制約のもとでは、その値を既約分数 で表した時、 となることも証明できます。 よって、$R \times Q \equiv P \pmod{998244353}, 0 \leq R < 998244353$ を満たす整数 が一意に定まります。 この を答えてください。
制約
- 入力される値はすべて整数
入力
入力は以下の形式で標準入力から与えられます。
出力
答えを出力してください。
4
1 1 3
831870300
例えば 回目の操作で順に頂点 が選ばれた場合を考えます。このとき、頂点 は「よい頂点」ですが、頂点 は祖先である頂点 が白色であるため「わるい頂点」です。よって 回目の操作で頂点を選ぶ際は頂点 の中から一様ランダムに選びます。
操作を行う回数の期待値は になります。
15
1 2 1 1 4 5 3 3 5 10 3 6 3 13
515759610