Official

D - Reverse Brackets Editorial by PCTprobability


予め SS( +S++ S + ) に置き換えておきます。この操作によって答えが変わらないことは簡単に確認できます。

ここで、SS と子の順番を区別する根付き木の間に以下のように一対一対応を作ります。

SS が正しい括弧列であることにより、対応する開き括弧と閉じ括弧の組を N2\frac{N}{2} 個作ることが出来る。この組が 11 個の頂点に対応する。ここで、根となる頂点は SS の左端の ( と右端の ) の組からなる頂点とする。 それ以外の頂点の親は、その括弧の組を含む括弧の組の中で、最も幅が小さいものとする。そして、ある頂点の子の順番は SS での括弧の組の位置関係を保って配置する。(括弧列の性質により括弧の組が交差することはなく、親が同じ 22 個の括弧の組同士の間では左右関係が定義出来る。)

根付き木からも両端の括弧が組となるような括弧列へ変換することが出来ます。(木 TT の根の子を u1,u2,...,uku_1,u_2,...,u_k としたとき、各 ii に対して uiu_i を根とする部分木から得られる括弧列 tit_i とすると、TT( +t1+t2+...+tk++ t_1 + t_2 + ... + t_k + ) が対応します。)

このとき、操作は以下のように言い換えられます。

  • ある頂点 vv を選ぶ。vv の子が u1,u2,...,uku_1,u_2,...,u_k の順番で並んでいるとする。ここから連続する部分列を選びその部分列を反転する。そして、その部分列に含まれる頂点のいずれかを祖先に持つ頂点全てについて、その頂点の子の順番を反転する。(ここでの「反転」は問題文で定義されている「反転」とは異なり、列 (a1,a2,,an)(a_1,a_2,\dots,a_n)(an,an1,...,a1)(a_n,a_{n-1},...,a_1) に置き換える操作を指します。)

この操作を組み合わせることにより、以下の操作が実現できます。

  • ある頂点 vv を選ぶ。vv の子が u1,u2,...,uku_1,u_2,...,u_k の順番で並んでいるとする。ここから連続する部分列を選び反転する。

また、新しいこの操作を用いても前の操作を行うことが出来るため、以降はこの操作を使い作ることの出来る木の個数を数え上げることにします。

この操作を用いることで、任意の頂点 vv の子を好きな順番に並び替えることが出来ます。ですが、vv の子を根とする部分木として同じものが存在する場合を考慮する必要があります。 つまり、問題は以下のように言い換えられます。

各頂点 vv について、vv を根とする部分木を tvt_v とする。t1,t2,,tnt_1,t_2,\dots,t_n を、同型かどうかで区別するとき、これらの度数分布を求めよ。
ここで、根付き木 A,BA,B が同型であるとは以下のいずれかを満たすことです。

  • A,BA,B が共に頂点数 11 の木である。
  • A,BA,B の子の個数が一致していて、それらを組にする方法であって「全ての組 (a,b)(a,b) について aa を根とする部分木と bb を根とする部分木が同型である」を満たすものが存在する。

これは部分木の頂点数の小さい方から判定することが可能です。具体的には、各部分木に整数のラベルを付けることを考えます。頂点数 11 の木にはラベルとして 11 を付けます。ここで、木 TT を根の子 u1,u2,,uku_1,u_2,\dots,u_k それぞれについてその頂点を根とする部分木につけられたラベルの多重集合と見ます。この多重集合が今までに現れていれば木 TT のラベルはその多重集合となる木のラベルと等しくなります。今までに現れていなければ木 TT のラベルはまだラベルとしてつけてない最小の正整数とします。

上記をそのまま実装するだけで O(NlogN)\mathrm{O}(N \log N) でこの問題を解くことが出来ます。正整数でなく木を表す括弧列を正規化して持つ、NN 個の部分木それぞれについて木の hash を愚直に求めるなどの O(N2)\mathrm{O}(N^2) の解法でも許容されます。

posted:
last update:



2025-03-13 (Thu)
15:24:05 +00:00