西瓜の備忘録

競プロとかで気づいたこととか考察を書き留めるためのブログ

AtCoderのTDC-Fを解いた

tdpc.contest.atcoder.jp

タイトル通り

TLでDP頑張ってる人に触発されて解けそうなDP問題探してたらいい感じのを見つけたのでがんばって解いた

内容はフィボナッチ数列の拡張(直前K項の和)みたいな感じ

累積和取ってi<K-1なら直前の総和

i=K-1なら総和-1

i=Kなら総和

i>Kなら総和からK項前の和を引いた数ってDPを更新していった

i=K-1,Kの時の部分もう少しきれいに場合分けできるかもしれないけど今は思いつかなかったので未来の自分に投げる