AtCoder Beginner Contest 373 F問題メモ

AtCoder Beginner Contest 373

F - Knapsack with Diminishing Values

• $N$ 種類の品物があり、 $i$ 種類目の品物の重みは $w_i$、価値は $v_i$ です。どの種類の品物も $10^{10}$ 個ずつあります。
• 品物をいくつか選んで、容量 $W$ のバッグに入れます。
• ただし、各品物につき、$k$ 個選ぶと $k^2$ のペナルティが科されます。
  • つまり、$i$ 種類目の品物を $k_i$ 個選んだとき、スコアは $k_i v_i - k_i^2$
• 選んだ品物の重さの総和を $W$ 以下にしつつ、達成できるスコアの総和の最大値 $\displaystyle \sum_{i=1}^{N}k_i v_i - k_i^2$ を求めてください。

• $1 \leq N \leq 3000$
• $1 \leq W \leq 3000$
• $1 \leq w_i \leq W$
• $1 \leq v_i \leq 10^9$
• 入力はすべて整数

以下の素直なナップサックDPを考える。

• $\mathrm{DP}[i,j]:=i$ 番目の品物まで考慮し、重さ合計が $j$ の場合の最大スコア

通常のナップサックDP（のうち、同じ品物を何個でも選んで良い版）では、 各 $i$ につき、$j$ が小さい方から $\mathrm{DP}[i,j+w_i] \xleftarrow[\max]{} \mathrm{DP}[i-1,j]+v_i$ と 更新していくことで $O(W)$、全体 $O(NW)$ で更新できる。

しかし、今回は加算されるスコア $v_i$ が、それまで何個の同じ品物を選んだかによって変化するので上手くいかない。
（そして、その個数別にDPを管理してたら、$O(NW^2)$ となり間に合わない）

以下の性質を利用する。

$\mathrm{DP}[i,j]$ は、もらうDPで考えて、以下の中の最大値である。

• $\mathrm{DP}[i-1,j-0w_i] + 0v_i - 0^2$
• $\mathrm{DP}[i-1,j-1w_i] + 1v_i - 1^2$
• $\mathrm{DP}[i-1,j-2w_i] + 2v_i - 2^2$
• $\mathrm{DP}[i-1,j-3w_i] + 3v_i - 3^2$
• …

            j-4w j-3w j-2w  j-w   j
DP[i-1]   ....x....x....x....x....x
               ↘   ↘   ↘   ↘  ↓
DP[i]                             *

$\mathrm{DP}[i,j]$ の更新にあたり最大値を取る更新元（$j-k_iw_i$）を $\mathrm{argmax}(j)$ とする。

ここで、以下の事実が成り立つ。

• ある $(j_1,j_2)$ の組について、$\mod{w_i}$ が等しく、$j_1 \lt j_2$ ならば、$\mathrm{argmax}(j_1) \le \mathrm{argmax}(j_2)$ である。

略証

|<-u個->|
s2     s1            j1   j2
        |<--- k1個 ---|    |
 |<---------- k2個 --------|

|<-u個->|
s2     s1            j1     j2
③     ①<--- k1個 ---|      |
④<----②---- k2個 ----------|

$\mod{w_i}$ が等しい同士にグループ分けすると、$j$ を増やすに従って $\mathrm{argmax}(j)$ は必ず広義単調増加する。

このような性質を Monotone といい、各 $j$ に対する最大値は Monotone Minima を使うことで $O(W \log{W})$ で求められる。

• monotone minima による高速化について | フューチャー技術ブログ

よって、全体で $O(NW \log{W})$ に高速化される。

Python3