Submission #22315608

Source Code Expand

import sys
import numpy as np

read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines

def main(A):
    A[::2] *= -1
    A_cum = np.append(0, np.cumsum(A))
    count = {}
    ans = 0
    for x in A_cum:
        ans += count.get(x, 0)
        count[x] = count.get(x, 0) + 1
    return ans

A = np.array(read().split(), np.int64)[1:]

print(main(A))

Submission Info

Submission Time
Task C - ARC Wrecker 2
User maspy
Language Python (3.8.2)
Score 500
Code Size 416 Byte
Status AC
Exec Time 299 ms
Memory 57452 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 500 / 500
Status
AC × 4
AC × 40
Set Name Test Cases
Sample sample_01.txt, sample_02.txt, sample_03.txt, sample_04.txt
All corner_01.txt, corner_02.txt, corner_03.txt, corner_04.txt, corner_05.txt, corner_06.txt, corner_07.txt, corner_08.txt, in01.txt, in02.txt, in03.txt, in04.txt, in05.txt, in06.txt, in07.txt, in08.txt, in09.txt, in10.txt, in11.txt, in12.txt, in13.txt, in14.txt, in15.txt, in16.txt, in17.txt, in18.txt, in19.txt, in20.txt, in21.txt, in22.txt, in23.txt, in24.txt, in25.txt, in26.txt, in27.txt, in28.txt, sample_01.txt, sample_02.txt, sample_03.txt, sample_04.txt
Case Name Status Exec Time Memory
corner_01.txt AC 119 ms 27228 KiB
corner_02.txt AC 112 ms 27032 KiB
corner_03.txt AC 113 ms 27004 KiB
corner_04.txt AC 111 ms 26908 KiB
corner_05.txt AC 112 ms 26964 KiB
corner_06.txt AC 111 ms 26912 KiB
corner_07.txt AC 114 ms 26620 KiB
corner_08.txt AC 114 ms 27204 KiB
in01.txt AC 254 ms 33088 KiB
in02.txt AC 253 ms 33612 KiB
in03.txt AC 265 ms 41284 KiB
in04.txt AC 273 ms 44984 KiB
in05.txt AC 287 ms 54804 KiB
in06.txt AC 290 ms 54784 KiB
in07.txt AC 286 ms 54524 KiB
in08.txt AC 292 ms 57452 KiB
in09.txt AC 296 ms 33372 KiB
in10.txt AC 264 ms 39176 KiB
in11.txt AC 270 ms 45704 KiB
in12.txt AC 284 ms 45868 KiB
in13.txt AC 278 ms 46428 KiB
in14.txt AC 288 ms 46192 KiB
in15.txt AC 270 ms 44688 KiB
in16.txt AC 292 ms 46160 KiB
in17.txt AC 277 ms 44592 KiB
in18.txt AC 272 ms 45736 KiB
in19.txt AC 283 ms 46016 KiB
in20.txt AC 276 ms 55220 KiB
in21.txt AC 277 ms 55080 KiB
in22.txt AC 299 ms 57220 KiB
in23.txt AC 267 ms 55128 KiB
in24.txt AC 273 ms 54848 KiB
in25.txt AC 296 ms 54824 KiB
in26.txt AC 291 ms 46256 KiB
in27.txt AC 285 ms 46324 KiB
in28.txt AC 297 ms 46020 KiB
sample_01.txt AC 108 ms 27036 KiB
sample_02.txt AC 114 ms 27044 KiB
sample_03.txt AC 113 ms 26964 KiB
sample_04.txt AC 112 ms 26916 KiB