問題一覧 > 通常問題

No.3407 Birds-of-Paradise' Christmas Live

レベル : / 実行時間制限 : 1ケース 2.000秒 / メモリ制限 : 512 MB / 標準ジャッジ問題
タグ : / 解いたユーザー数 12
作問者 : hirayuu_yc / テスター : highlighter
ProblemId : 12876 / 出題時の順位表 / 自分の提出
問題文最終更新日: 2025-12-14 10:02:38
コンテストの他の問題:

問題文

カンザシフウチョウとカタカケフウチョウが歌とダンスを披露するようです。かわいいですね。彼女らの名前が流石に長過ぎるので、以降カンザシ、カタカケと略します。

彼女らは、$X$ 単位時間踊ります。時刻 $i$ には、

  • $W_i=1$ のとき、またそのときに限りカンザシはカタカケの方を向いていて、
  • $S_i=1$ のとき、またそのときに限りカタカケはカンザシの方を向いています。

彼女らは、はじめすべての振り付けを覚えています。

あなたは、彼女らの振り付けを忘れさせる操作を好きなだけ行うことができます。具体的には、

  • $1\leq i\leq j\leq X$ なる整数 $i,j$ と、カンザシかカタカケの一方を指定する。ただし、指定したフレンズは時刻 $i,i+1,\dots,j$ の振り付けを覚えていなければならない。
  • 指定したフレンズの時刻 $i,i+1,\dots,j$ の振り付けを忘れさせ、$(j-i+1)^2$ の報酬を得る。

ただし、あなたにとってダンスが失敗するのは本望ではありません。ダンスは、以下の条件を満たす時刻が存在するとき、またそのときに限り失敗します。

  • 振り付けを忘れているフレンズがいて、なおかつ
    • そのフレンズがもう一方のフレンズの方を向いていない、または
    • もう一方のフレンズも振り付けを忘れている

ダンスが失敗しないようにしつつ、得ることのできる最大の報酬を求めてください。

ただし、$X,W,S$ は直接与えられず、代わりに長さ $n$ の非負整数列 $w$ 、長さ $m$ の非負整数列 $s$ が与えられます。

$X,W,S$ は以下のように復元することができます。

  • $X$ は $w$ の総和である(なお、$w$ の総和と $s$ の総和は等しい)。
  • $W$ は、$w_1$ 個の $1$ 、$w_2$ 個の $0$ 、$w_3$ 個の $1$ 、…、$w_n$ 個の $n\bmod 2$ をこの順につなげた長さ $X$ の数列である。
  • $S$ は、$s_1$ 個の $1$ 、$s_2$ 個の $0$ 、$s_3$ 個の $1$ 、…、$s_m$ 個の $m\bmod 2$ をこの順につなげた長さ $X$ の数列である。

制約

  • $1\leq n,m\leq 3\times 10^5$
  • $0\leq w_1,s_1$
  • $1\leq w_i$ $(2\leq i\leq n)$
  • $1\leq s_i$ $(2\leq i\leq m)$
  • $1\leq \sum_{i=1}^{n} w_i=\sum_{i=1}^{m} s_i\leq 10^9$
  • 入力はすべて整数

入力

$n$
$w_1$ $w_2$ $\dots$ $w_n$
$m$
$s_1$ $s_2$ $\dots$ $s_m$

出力

答えを出力せよ。

サンプル

サンプル1
入力
2
5 3
3
0 2 6
出力
40

$W=(1,1,1,1,1,0,0,0),S=(0,0,1,1,1,1,1,1)$ です。

時刻 $1,2$ のカンザシの振り付け、時刻 $3,4,\dots,8$ のカタカケの振り付けを忘れさせるのが最適で、適切に実行すると $40$ の報酬を得ることができます。

サンプル2
入力
4
0 1000000 1 1000000
4
0 1000000 1 1000000
出力
1

君に出会える毎日がずっと続くといいな♪

一瞬だけお互いに見つめ合いますから、そこでどちらかの振り付けを忘れさせるのが最適です。

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