はじめに
問題は以下です。
https://atcoder.jp/contests/arc171/tasks/arc171_c
この記事は本番中に通せなかった AtCoder Regular Contest 171 C問題 - Swap on Tree の解き方の解釈を与えたものです。
この問題は自分にとってかなり苦手で、なぜこんなにも正解者が多かったのかは今でも理解できません。とりあえず解説を読んでから数時間考えた末に思いついた方法を載せます。これよりもっと簡単に解く方法があれば知りたいです。
STEP 1
辺に注目
ある辺に注目して、それを と置きます。辺 を使用した場合、駒がスワップされた後に辺 は削除され、木は つの連結成分に別れます。
その後、異なる連結成分の頂点はもう交換されることはありません。
辺 を削除した後のそれぞれの連結成分に属する頂点を グループと考えてみます。
もし操作中に辺 を使用しない場合, 最初から辺 は無いものとして構わないでしょう。よって、すべての操作にわたって グループと グループの頂点が交換されることはありません。
もし操作中に辺 を使用する場合、 グループの何らかの頂点と、 グループの何らかの頂点を交換した後に辺 を削除します。
つまり、操作後の状態において、もとの グループの中に グループの頂点が つ入り込んでいる状態になります。もとの グループの中にも グループの頂点が つ入り込んでいます。
辺 を使用した場合と使用していない場合とで、もとの グループの中に グループの頂点が入り込んでいるかどうかが異なります。よって、確実に数列 が違うということが分かります。
問題の言い換え
辺 に注目するのをやめて、すべての辺のそれぞれについて「使用する / 使用しない」を決めたとします。
使用する辺の集合によって出来上がる数列 はすべて異なる、と感じることができます。
実際にこれは簡単に証明できます。使用する辺の集合を として、 とします。このとき、(一般性を失わず) とします。すると、 を つ取ることができます。 では辺 は使用しますが、 では辺 は使用しません。さきほど考察したように、ある辺を「使用した / 使用しない」で数列 は必ず異なります。よって、出来上がる数列 は で異なります。これが任意の で成り立つため、使用する辺の集合によって出来上がる数列 の集合はすべて異なります。
よって、使用する辺集合を固定して、その場合に解けばよいことが分かります。使用する辺集合によって連結成分が別れている場合、それらの連結成分が関わることはありません。よって、辺集合に対するありえる数列 の個数は、それぞれの連結成分について操作した場合の掛け算になります。
よって、使用する辺集合は連結、とくに木の「すべての辺を使用する」場合を考えればいいことが分かります。制約は一旦ほっといて、これによって問題が「すべての辺を使用する」場合に帰着されました。
STEP 2
すべての辺を使用する場合を考えます。まず、出来上がる数列は残念ながら 通りでも 通りでもありません。辺を操作する順番が違っていても、数列 が同じになる場合があります。
操作順序が違っていても得られる数列が同じになる場合があるというのは嬉しくありません。このとき、操作順序ではなく、「数列 に対して、それがありえるのはどういうときか?」を考えます。
駒の「流れ」
ある頂点 を固定して、駒 の行き先を とします。駒 を頂点 に運ぶには、 のパスをとって、パス上ではこの順に運ぶ必要があります。
また、頂点 から頂点 に駒を運んでいるときは、他の辺に邪魔されてはいけません。
これが「頂点 の駒を頂点 に運ぶ」ための必要十分条件です。
そして、「パス上ではこの順に運ぶ必要がある」「他の辺に邪魔されてはいけない」の条件をまとめると、たとえば、↓の頂点のまわりの辺では、「1stの直後に2ndが来ないといけない」となります。
始点 においては、「その周りで1stを最初に使う」、終点 においては、「その周りで3rdを最後に使う」となります。つまり、通る頂点それぞれに対して、その頂点の周りの相対順序を 個定めています。
1対1対応
ここで、頂点 の固定を外して、すべての流れを見てみます。
すべての頂点からの「流れ」が見えます。さきほどの必要十分条件に従って、ある頂点の周りに注目すると、その相対順序は必ず 通りに定まることが分かります。
「とりうる数列 」に対して、こういう現象が、すべての頂点の周りについて起こります。すなわち、すべての頂点の周りの辺について、使う順番が決まります。
ここからもとの「流れ」は復元できます。「始点」は1stの辺に流れます。流れ着いた点の周りに、その直後に使う辺があれば、そこに流れます。なければ、そこが終点になります。
よって、ありえる数列 →「それぞれの頂点の周りの使う辺の順序」→ ありえる数列 が一意に定まることが分かりました。
問題なのが、「その頂点の周りの使う辺の順序」を自由に決めると、それに対応する数列 は「ありえる数列」か?ということです。すなわち、「その頂点の周りの使う辺の順序」を自由に決めると、それに従う「使う辺の順序」が取れるか?ということです。直感的には取れそうです。
もしそうなら、ありえる数列 と「それぞれの頂点の周りの使う辺の順序」との一対一対応(全単射)が取れて、代わりに「それぞれの頂点の周りの使う辺の順序」を数え上げればいいことになります。
「その頂点の周りの使う辺の順序」に従う「使う辺の順序」が取れるか
「最初に使う辺」はどれがいいでしょうか?使う辺の両端の頂点において、その辺がどちらも 1st であればよいです。それぞれの頂点から 1st の辺のみを伸ばした有向グラフを考えます。
これは functional graph で、閉路は必ず存在します。しかし、もとのグラフが木の構造をしているので、閉路は必ず長さ となります。その閉路を取ればよいです。今回は つありますが、どれでもよいです。
その辺を使ったあとを考えます。まず、辺によって連結成分に分断します。ある頂点について、その頂点の周りのすべての辺を使い切った場合、その頂点を捨てるとします。
生きている頂点については、また1stを考えると functional graph となるのでこの操作を繰り返すことができます。最後はすべての辺を使い切ります。
よって、「それぞれの頂点の周りの使う辺の順序」→ ありえる!数列 が対応する、ということが分かりました。
答え
ありえる数列 と「それぞれの頂点の周りの使う辺の順序」との一対一対応(全単射)が取れたので、それぞれの頂点の周りの辺の順序を独立に定めることができます。
頂点 の次数を とすると, の周りの辺の順序の決め方は 通りあります。
よって、 が答えになります。
STEP 3
DP
「すべての辺を使う場合」は、 が答えです。実際に答える際は、使う辺の集合 通りを指定しないといけないように見えますが、この場合何らかの木DPができそうに見えます。
適当に木を順序付けて有向木とします。頂点 に対して「 の子を根とする部分木たちに対する情報」から「 の情報」を得たいです。
の子 について、その頂点を使う場合、 の次数が 個増えます。木DPをする場合、 という制約によって「二乗の木DP」が許されます。よって、情報として、「根で使っている辺の次数」を持っておいてもよいでしょう。
を, 「頂点 を根とする部分木であって、現在根で使っている辺の次数が であるとき、その答えの総和」とおきます。
各 の子 について、
- と の間の辺を使う場合、 の次数は 増え、 の次数は 増えて継承される。もとの の次数を , の次数を とした場合、STEP 2 で導いた に従って、その差分を計算すると、新しい の状態 に が加算される。
- と の間の辺を使わない場合、次数はどちらも増えず継承される。新しい の状態 に が加算される。
とすると、部分木 個のマージが完成しました。ここで、 や を掛けるのは、
と同じ現象です。木DP全体の計算量は二乗の木DPによる と同じくらいか、それより軽そうです。(実際は同じらしい?です)
コード
C++, 28 ms, 964 Byte https://atcoder.jp/contests/arc171/submissions/50027892