エクサウィザーズ 2019 E - Black or White
問題
考察
よくわからない有理数がなんとかは後ろの方で説明。
ある状態でチョコを食べる時、チョコが十分にある場合はであるが、片方のチョコがなくなってしまう場合には or となっているためチョコがなくなっているケースを考慮する必要がある。
番目に黒のチョコを食べる確率は以下のようになる。ただし黒のチョコが存在しない確率と白のチョコが存在しない確率をそれぞれとおいた。
整理すると
となる。この式からを求めることで解が得られることになる。上の式はだいたいだけど食べきってしまう分の補正項が括弧の分だけありますよ、というのが表現されている。
の求め方
の求め方との求め方は同様なので、以下はの求め方を記述する。
の場合
明らかにである。(食べきることはないので)
の場合
の間にすべて黒を選んだ場合にのみ黒がなくなってしまうので、である。
の場合
本題である。黒が個、白が個の順列を考えると以下のような式が成立しそうな気がするが、これは過剰に低く見積もってしまっている。
これは例えば初めに前半にすべての黒のチョコを食べて後半に白のチョコを食べるというケースを考えることでわかる。このケースの生起確率は ではなくである。なぜなら黒のチョコを連続で個食べる確率はであるが、その後白のチョコを食べ続ける確率はであるためである。このケースから上のcombinationを用いた式は成り立っていなさそうなことがわかる。
ここでdpを考える。番目に個のチョコが食べられている確率は、番目に個のチョコが食べられている確率と個のチョコが食べられている確率をがわかると得られそうである。結果以下のdpが得られた。に添字を一つ追加して現在食べた黒のチョコの数を識別した。
2項目のは黒を食べる場合と白を食べる場合と2通りあるためである。はメモ化して順次求めていく。(は上の方で求めてある) の方はどちらかを食べきってしまうことはないので、普通にcombinationが使えるため以下の式が成り立ちが得られた。
についても同様である。
以上より回目に黒のチョコを食べる確率が求められた。
有理数を分数で表した場合のなんとかかんとか
こどふぉではよくある表現らしい。
要は普通に掛け算や割り算をmod取りながらやれば良い。出力例1をよく見るとの逆元とそれの倍の値が並んでいることがわかる。
の領域は1以下の分数になっている。1以上の値と同時に取り扱う場合には大変なことになりそう。
解答
記事中とPとQが反対なので参照する場合は注意。
atcoder.jp