AtCoder Beginner Contest 167 F - Bracket Sequencing
解法
「正しい括弧列」について
この問題だと単に括弧列という言葉が使われていますが、どちらかと言うと「正しい括弧列」みたいな呼ばれ方をされることが多いです。
正しい括弧列の条件は、言葉で書くと
- 開き括弧と閉じ括弧の数が等しい。
- それぞれの閉じ括弧について、それに対応する開き括弧がそれより前に出現している。
となりますが、これを (
を 、)
を として見たときの先頭からの累積和で考えるのが定石です。すなわち上記の条件はそれぞれ累積和によって
- 全体の和がちょうど である。
- 先頭から累積和を取った時に、その最小値が 以上である。(つまり、先頭からどの地点までの累積和の値も負でない)
と書くことができて、これらを両方を満たすことが正しい括弧列であることの必要十分条件です。
パーツを繋げられる条件を考える
与えられた「パーツ」である それぞれについて、累積和の最小値 と全体の和 を求めます。パーツ はこの2つの値の組 で特徴づけられます。
全てのパーツを連結した文字列が先ほどの条件を満たせるかどうかを考えます。まず条件1について、全体の和は繋げ方に依らず全てのパーツについての の和になります。なのでこれが であるかどうかを調べれば良いです。
条件2について考えます。パーツを前から順番に繋げて行って、それまでの累積和(つまりそれまでに使ったパーツの の和)が であるとします。初期状態では です。ここにパーツ をつなげると、
- パーツ の中での、先頭からの累積和の最小値:
- パーツ 終了地点での、先頭からの累積和:
となります。このとき が負になってしまうと条件2に違反するので、それまでの累積和が であるときにパーツ を付けられる条件は と表現できます。そして累積和 は に変化します。
言い換えると は「大きいほどパーツ 自身を付けられる条件が緩くなる」、 は「大きいほどそれ以降のパーツを付けられる条件が緩くなる」ような値だと考えることができます。
最適な連結順序を考える
以上の考察から、なるべく条件を満たせるようにするにはどのような順序でパーツを連結していけば良いかを考えます。
まず であるパーツは、付ければ付けるほど が増加して条件を満たしやすくなるので、付けられるようになったものから付けてしまって良いです。よって付けられる条件が緩いもの、つまり が大きいものから順番に使うことにします。
それから であるパーツを使っていきます。これは付けるほど後が厳しくなっていくので と の両方が絡んできます。「条件の厳しいものを早く消化したい」と思うと が小さいものから、「後の条件を厳しくしたくない」と思うと が大きいものから使いたくなります。
では結局どうすれば良いのかと言うと、 が大きいものから順番に使うのが最適になります。
証明1
実は過去に書いた記事に全く同じ証明があります。こちらを参照してください。場合分けが必要なのでちょっとややこしいです。
Codeforces Round #579 (Div. 3) F1. Complete the Projects (easy version) - ARMERIA
リンク先の記事における を本解説の に置き換えてください。
証明2
対称性を利用します。
先ほどとは逆に、文字列末尾から見て )
を 、(
を としたときの累積和を考えます。その最小値と全体の和を と書くことにします。
ここで なので、 であるパーツについては です。そのため先ほど であるパーツについて書いたことと同じように、これらのパーツは末尾から見て が大きいものから順番に使うとして良いです。
また図を見ると分かるように、末尾から見た累積和は先頭から見た累積和と比べて のぶんだけずれています。そのため が成り立ちます。よって末尾から見て が大きいものから、つまり先頭から見て が大きいものから順番に使うとして良いです。
まとめ
パーツとなる各文字列 について、括弧を に置き換えた時の先頭からの累積和を求め、 を計算します。
それらを以下の順序で連結し、累積和が途中で負にならないかどうか、合計が になるかどうかを確認します。
- まず であるパーツを、 が大きいものから順番に使う。
- それから であるパーツを、 が大きいものから順番に使う。