結果
問題 | No.1765 While Shining |
ユーザー |
![]() |
提出日時 | 2021-11-26 22:36:35 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 99 ms / 2,000 ms |
コード長 | 498 bytes |
コンパイル時間 | 156 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 107,776 KB |
最終ジャッジ日時 | 2024-06-29 18:15:06 |
合計ジャッジ時間 | 2,835 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
n = int(input()) a = list(map(int, input().split())) b0 = a[:] b1 = a[:] for i in range(n): if i % 2 == 0: b0[i] ^= 1 else: b1[i] ^= 1 nxt0 = [n - 1] * n nxt1 = [n - 1] * n for b, nxt in ((b0, nxt0), (b1, nxt1)): x = n - 1 for i in range(n)[::-1]: nxt[i] = x if b[i] == 1: x = i ans = 0 for i in range(n): if a[i] == 0: continue if i % 2 == 0: ans += nxt0[i] - i else: ans += nxt1[i] - i print(ans)