問題文がヒントになってたな…。 https://atcoder.jp/contests/arc115/tasks/arc115_f

まぁこれは定番かな…。 https://atcoder.jp/contests/abc197/tasks/abc197_f