Codeforces Global Round 7 E. Bombs
問題概要
から までの順列 が与えられる。
爆弾が仕掛けられているインデックスの集合 を考える。 に対して以下のように計算される値を と定義する。
- を空集合とする。
- の順に以下の操作を行う。
- 集合 に を追加する。その後 ならば爆弾が発動し、 の中で最大の要素が除外される。
- この操作が完了した後、 に含まれる最大の要素が である。
から までの順列 が与えられる。初期状態の を空集合として、 について順番に以下を処理せよ。
- 現時点での について を出力せよ。その後、 に を追加せよ。
制約
- は順列
- は順列
解法
まず「答えはある値 より小さいか?」という判定問題を考えます。これはつまり、 の全ての値が除外されるか?ということです。
この判定問題を解く際には、操作は以下のように単純化できます。
- の順に以下を行う。
- が 以上ならば、値のストックを 個増やす。
- であり、値のストックが 個以上ならば、値のストックを 個減らす。
つまり具体的にどの値が に追加され除外されたかを考えなくても良くなります。この操作の結果として値のストックが 個であれば、 の全ての値が除外されたことになります。
ただし が変更されるたびにこの操作を毎回やっていては間に合わないので、もう少し扱いやすい形にしましょう。
キーとなる考え方は累積和です。 以上の値が追加されることを 、爆弾が発動することを として、先頭から累積和を取ることを考えます。
この累積和において、「最小値」と「最後の値」に注目します。最小値よりも最後の値が大きい場合、最小値を取る時点より後に発生した であって、 によって相殺されなかったものが存在する、ということになります。逆も然りで、最小値と最後の値が等しい場合は全ての が で相殺されています。
よってこれら2つの値を見るだけで、「答えはある値 より小さいか?」という判定問題が解けることになります。
そしてこの形にしてしまえば、 の要素や判定問題の の値が変わっても対応できます。 に が追加されたときには、その位置に を追加すれば良いです。また に要素が追加されることで答えの値は単調減少するので、判定問題の も1つずつ減らしていくことになります。 を に変えた時に、 である位置に を追加すれば良いです。
数列の値を変更しながらその累積和の最小値を管理したいので、これは区間加算と区間最小値取得ができる遅延伝播セグメント木などで実現することができます。
ACコード
Submission #73701092 - Codeforces
この実装では累積和の管理において、先頭に を入れて各位置の 部分と 部分を分けて扱う長さ のセグメント木を用いていますが、 と は分けなくても大丈夫だと思います。先頭の はあったほうが良いです。