atcoder#AGC047D. [AGC047D] Twin Binary Trees
[AGC047D] Twin Binary Trees
配点 : 点
問題文
テレビドラマシリーズ『ストレンジャー・シングス』に触発されたクマのリマクは、現実世界と鏡像世界を歩いて行き来することにしました。
二つの高さ の完全二分木があり、それぞれの頂点は から までの標準的な番号付けがなされています。 すなわち、根の番号は で、番号 の子の番号は と です。
一つの木が持つ葉の数を とします。すなわち、 です。
数列 の順列 が与えられます。これは、二つの木を結ぶ 本の 特殊辺 を表します。 すなわち、第一の木の番号 の頂点と第二の木の番号 の頂点が一本の特殊辺で結ばれています。
入力例 1 の図示。P = (2, 3, 1, 4) であり、緑の辺が特殊辺
サイクルの 積 を、それに含まれる頂点の番号の積と定義します。ちょうど二本の特殊辺を持つような すべての単純サイクルの積の和を で割った余りを求めてください。
なお、単純サイクルとは、長さが 以上であって頂点や辺の重複がないようなサイクルです。
制約
- (ただし )
- (すなわち、この数列は の順列である)
入力
入力は以下の形式で標準入力から与えられる (ここで、 である)。
出力
ちょうど二本の特殊辺を持つようなすべての単純サイクルの積の和を求め、それを で割った余りを出力せよ。
3
2 3 1 4
121788
下図に考慮すべきサイクルを二つ示します (他にも存在します)。一つめのサイクルの積は $2 \cdot 5 \cdot 6 \cdot 3 \cdot 1 \cdot 2 \cdot 5 \cdot 4 = 7200$、二つめのサイクルの積は $1 \cdot 3 \cdot 7 \cdot 7 \cdot 3 \cdot 1 \cdot 2 \cdot 5 \cdot 4 \cdot 2 = 35280$ です。三つめのサイクルは、特殊辺の数が二本でないため、考慮すべきサイクルではありません。
2
1 2
36
グラフ中の唯一のサイクルは確かに特殊辺を二本持ち、その頂点番号の積は です。
5
6 14 15 7 12 16 5 4 11 9 3 10 8 2 13 1
10199246