結果

問題 No.1765 While Shining
ユーザー cozy_saunacozy_sauna
提出日時 2022-03-10 17:12:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 201 ms / 2,000 ms
コード長 3,136 bytes
コンパイル時間 329 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 101,376 KB
最終ジャッジ日時 2024-09-14 17:57:14
合計ジャッジ時間 4,417 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 59 ms
64,384 KB
testcase_01 AC 61 ms
64,512 KB
testcase_02 AC 57 ms
64,640 KB
testcase_03 AC 58 ms
64,384 KB
testcase_04 AC 61 ms
64,512 KB
testcase_05 AC 58 ms
65,024 KB
testcase_06 AC 58 ms
64,640 KB
testcase_07 AC 60 ms
64,512 KB
testcase_08 AC 61 ms
65,152 KB
testcase_09 AC 103 ms
93,988 KB
testcase_10 AC 104 ms
99,024 KB
testcase_11 AC 98 ms
98,804 KB
testcase_12 AC 101 ms
101,376 KB
testcase_13 AC 95 ms
98,688 KB
testcase_14 AC 106 ms
99,928 KB
testcase_15 AC 104 ms
99,576 KB
testcase_16 AC 106 ms
99,668 KB
testcase_17 AC 104 ms
99,720 KB
testcase_18 AC 104 ms
99,684 KB
testcase_19 AC 102 ms
99,572 KB
testcase_20 AC 103 ms
99,580 KB
testcase_21 AC 106 ms
99,604 KB
testcase_22 AC 110 ms
99,440 KB
testcase_23 AC 106 ms
99,828 KB
testcase_24 AC 201 ms
99,692 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from sys import setrecursionlimit, stdin
from collections import defaultdict, deque
from itertools import permutations, combinations, product
from functools import lru_cache
from random import sample, choice, randint, random
from bisect import bisect_left, bisect_right
from heapq import heappush, heappop
from math import factorial, exp
from copy import copy, deepcopy
from time import time

setrecursionlimit(1 << 20)
readline = stdin.readline
# @lru_cache(maxsize = None)
INF = 10 ** 18
MOD = 998244353
# MOD = 1000000007
DYDX = [(-1, 0), (1, 0), (0, -1), (0, 1)]
ALP = 26
''' Input '''
def I(): return int(readline())
def ST(): return readline()[:-1]
def LI(): return list(map(int, readline().split()))
def LII(): return list(map(lambda x: int(x) - 1, readline().split()))
def SPI(): return map(int, readline().split())
def SPII(): return map(lambda x: int(x) - 1, readline().split())
def FIE(x): return [readline()[:-1] for _ in [0] * x]
def NODE(N, edge):
    node = [[] for _ in [0] * N]
    for _ in [0] * edge:
        a, b = map(int, input().split())
        node[a - 1].append(b - 1)
        node[b - 1].append(a - 1)
    return node 

def NODE_W(N, edge):
    node = [[] for _ in [0] * N]
    for _ in [0] * edge:
        a, b, c = map(int, input().split())
        node[a - 1].append((b - 1, c))
        node[b - 1].append((a - 1, c))
    return node

''' Array '''
def cmin(dp, i, x):
    if x < dp[i]: dp[i] = x
def cmax(dp, i, x):
    if x > dp[i]: dp[i] = x

''' Sorted Array '''
# x <= A[i] (minimum i meeting the requirement)
def binary(sortedA, x): return bisect_left(sortedA, x)
# x < A[i] (minimum i meeting the requirement)
def binary2(sortedA, x): return bisect_right(sortedA, x + 1)
# number of x in A
def quantity(sortedA, x): return bisect_right(sortedA, x) - bisect_left(sortedA, x)

''' Alphabet '''
def id_a(s): return ord(s) - ord('a')
def id_A(s): return ord(s) - ord('A')
def st_a(i): return chr(ord('a') + i)
def st_A(i): return chr(ord('A') + i)

''' Other'''
def ranges(*args): return [(i, j) for i in range(args[0]) for j in range(args[-1])]
def nynx(y, x, ly = INF, lx = INF): return [(y + dy, x + dx) for dy, dx in DYDX if 0 <= y + dy < ly and 0 <= x + dx < lx]
def gen(x, *args):
    ret = [x] * args[-1]
    for e in args[:-1][::-1]: ret = [deepcopy(ret) for _ in [0] * e]
    return ret
def is_prime(x):
    if x == 1 or x % 2 == 0: return x == 2
    f = 3
    while f * f <= x:
        if x % f == 0: return False 
        f += 2
    return True 

''' Output '''
def puts(E):
    for e in E: print(e)
def pprint(E): 
    print()
    for e in E: print(e)
def yn(x): print("Yes" if x else "No")
def blank(arr): print(" ".join(map(str, arr)))
def link(arr): print(''.join(map(str, arr)))
def debug(*args):
    if DEBUG:
        for e in args: print(e)
DEBUG = False
###############################################################################################

N = I()
A = LI()
M = gen(0, 2)

ans = 0
for i in range(N)[::-1]:
    if A[i] and i != N - 1:
        ans += M[0]
    if A[i]:
        M[1] = M[0] + 1
        M[0] = 1
    else:
        M[0] = M[1] + 1
        M[1] = 1
    
print(ans)
0