AtCoder Beginner Contest 187 [ D - Choose Me ] をPyhonで解く
AtCoder Problemsが🟫茶diffと判定した問題から重点的にやっておこう。
という方針を立てました。
このレベル帯の問題が出来ないことが多いと感じるんですよ。
問題に書いてある通りに実装するとTLEで、実は数学的に計算量減らせますよ。
系の問題が多い、このレベル帯。
とりあえず、この問題やります。
問題概要
青木氏と高橋氏が選挙をする。
市にはN個の町があり、i番目の街には青木派がAi人、高橋派がBi人います。
青木派 | 高橋派 | |
---|---|---|
町1 | A1 | B1 |
町2 | A2 | B2 |
…… | …… | …… |
町N | AN | BN |
- 高橋氏がある町で演説を行った場合、その町の高橋派も青木派も全員高橋氏に投票します。
- 高橋氏がある町で演説を行わなかった場合、その町の青木派は全員青木氏に投票し、高橋派は投票に行きません。
高橋氏が青木氏より多く票を獲得するためには、最小でいくつの町で演説をする必要があるでしょうか?
解法
高橋派の票数 青木派の票数
と定義します。
になると高橋氏の勝利。
高橋派と青木派の票数の差分で考えるのが、この問題のポイント。
それによって町ごとの高橋氏が演説した時の効率がシンプルに一つの数字で、あわらすことができるようになる。
あとは町ごとの効率を計算して、ソート。
効率が良い町から演説していって、何回演説すれば勝てるかを計算すればよい。
高橋氏が演説をしなかった時、青木派の票はすべて青木氏に投票され、高橋派の票はどちらにも投票されない。
ゆえに
Σが出てくると、難しい感じしますが数列Aの全合計というだけ。
町iで高橋氏が演説したとき
ゆえにXについて考えると
つまり高橋氏が町iで演説したときに変化するの値は
を町ごとに計算しておいてソート。
効率のよい町から順に演説を行なっていけば、演説の最小回数はわかる。
書いたPythonコード
解説の内容をPythonでコードに落としこんだモノが以下。
参考
公式の解説が、この問題は解説わかりやすい。