問題一覧 > 通常問題

No.3050 Prefix Removal

レベル : / 実行時間制限 : 1ケース 2.000秒 / メモリ制限 : 512 MB / 標準ジャッジ問題
タグ : / 解いたユーザー数 55
作問者 : suisen / テスター : 37zigen 👑 rin204
2 ProblemId : 9025 / 出題時の順位表 / 自分の提出
問題文最終更新日: 2025-02-08 13:48:15

問題文

正整数 $N, K$ および長さ $N$ の整数列 $A = (A _ 1, A _ 2, \ldots, A _ N)$ が与えられます。

suisen 君の現在の満足度は $0$ です。suisen 君はこれから $i = 1, 2, \ldots, K$ の順に次を行います。

  • 数列 $A$ の長さを $\vert A\vert$ とする。
    • $\vert A\vert = 0$ の場合: suisen 君の満足度は $-10 ^ {100}$ となる。

    • $\vert A\vert \geq 1$ の場合: $1\leq p\leq \vert A\vert$ を満たす整数 $p$ を選んで、$A$ の先頭から $p$ 個の要素を取り除く。このとき取り除いた要素の総和を $S$ として、suisen 君の満足度は $S\times i$ だけ増加する。

      例えば $A=(-3,2,1)$ に対して $p=2$ を選択すると先頭から $2$ つの要素 $-3,2$ が取り除かれ、$A = (1)$ となります。このとき取り除いた要素の総和は $(-3)+2=-1$ なので、suisen 君の満足度は $(-1)\times i$ だけ増加します。

全ての操作を終えた後の suisen 君の満足度の最大値を求めてください。

入力

$N\ K$
$A _ 1\ A _ 2 \ \cdots \ A _ N$
  • 入力は全て整数で与えられる。
  • $1\leq K \leq N\leq 5\times 10 ^ 5$
  • $\vert A _ i \vert \leq 10 ^ 7$

出力

答えを $1$ 行に出力してください。最後に改行してください。

サンプル

サンプル1
入力
5 3
3 2 0 1 -1
出力
10

$N = 5, K = 3, A = (3,2,0,1,-1)$ です。以下のように操作を行うことで、最終的な満足度は $10$ となります。

  • $i = 1$: $A = (3,2,0,1,-1)$ で数列 $A$ の長さは $5$ です。$p=1$ を選択すると、先頭から $1$ つの要素 $3$ が取り除かれ $A = (2,0,1,-1)$ となります。このとき取り除いた要素の総和は $3$ なので、満足度に $3 \times 1 = 3$ が加算されて $3$ となります。
  • $i = 2$: $A = (2,0,1,-1)$ で数列 $A$ の長さは $4$ です。$p=1$ を選択すると、先頭から $1$ つの要素 $2$ が取り除かれ $A = (0,1,-1)$ となります。このとき取り除いた要素の総和は $2$ なので、満足度に $2 \times 2 = 4$ が加算されて $7$ となります。
  • $i = 3$: $A = (0,1,-1)$ で数列 $A$ の長さは $3$ です。$p=2$ を選択すると、先頭から $2$ つの要素 $0,1$ が取り除かれ $A = (-1)$ となります。このとき取り除いた要素の総和は $1$ なので、満足度に $1 \times 3 = 3$ が加算されて $10$ となります。

満足度が $10$ を超えるような操作の方法は存在しないので、答えとして $10$ を出力します。

サンプル2
入力
5 1
-5 1 -4 3 4
出力
-1

満足度は負の値を取り得ます。

提出するには、Twitter 、GitHub、 Googleもしくは右上の雲マークをクリックしてアカウントを作成してください。