Submission #73236860

Source Code Expand

import sys
input = lambda: sys.stdin.readline().rstrip()
ii = lambda: int(input())
mii = lambda: map(int, input().split())
lii = lambda: list(mii())
mod = 10**9+7
from collections import Counter
def solve():
    n,k = mii()
    a = lii()
    pre = [0]
    for i in a:
        pre.append(pre[-1]+i)
    cnt = Counter(pre)
    ans = 0
    for i,x in enumerate(pre):
        cnt[x]-=1
        ans+=cnt[k+x]
    print(ans)

t = 1
# t = ii()
for _ in range(t):
    solve()

Submission Info

Submission Time
Task E - Sum of Intervals
User xiaoe
Language Python (PyPy 3.11-v7.3.20)
Score 433
Code Size 490 Byte
Status AC
Exec Time 148 ms
Memory 183608 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 433 / 433
Status
AC × 3
AC × 60
Set Name Test Cases
Sample sample01.txt, sample02.txt, sample03.txt
All sample01.txt, sample02.txt, sample03.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, in29.txt, in30.txt, in31.txt, in32.txt, in33.txt, in34.txt, in35.txt, in36.txt, in37.txt, in38.txt, in39.txt, in40.txt, in41.txt, in42.txt, in43.txt, in44.txt, in45.txt, in46.txt, in47.txt, in48.txt, in49.txt, in50.txt, in51.txt, in52.txt, in53.txt, in54.txt, in55.txt, in56.txt, in57.txt
Case Name Status Exec Time Memory
in01.txt AC 69 ms 93332 KiB
in02.txt AC 69 ms 93488 KiB
in03.txt AC 70 ms 93456 KiB
in04.txt AC 69 ms 93368 KiB
in05.txt AC 69 ms 93444 KiB
in06.txt AC 145 ms 176260 KiB
in07.txt AC 71 ms 93920 KiB
in08.txt AC 70 ms 93600 KiB
in09.txt AC 110 ms 144412 KiB
in10.txt AC 146 ms 177964 KiB
in11.txt AC 147 ms 183052 KiB
in12.txt AC 120 ms 149696 KiB
in13.txt AC 109 ms 143728 KiB
in14.txt AC 147 ms 183040 KiB
in15.txt AC 126 ms 175096 KiB
in16.txt AC 115 ms 152352 KiB
in17.txt AC 126 ms 175092 KiB
in18.txt AC 144 ms 183304 KiB
in19.txt AC 114 ms 144328 KiB
in20.txt AC 137 ms 178828 KiB
in21.txt AC 137 ms 178648 KiB
in22.txt AC 112 ms 144260 KiB
in23.txt AC 144 ms 183608 KiB
in24.txt AC 76 ms 102340 KiB
in25.txt AC 70 ms 93500 KiB
in26.txt AC 70 ms 93400 KiB
in27.txt AC 146 ms 179452 KiB
in28.txt AC 70 ms 94168 KiB
in29.txt AC 76 ms 102116 KiB
in30.txt AC 144 ms 183052 KiB
in31.txt AC 69 ms 93740 KiB
in32.txt AC 145 ms 183280 KiB
in33.txt AC 70 ms 93432 KiB
in34.txt AC 69 ms 93400 KiB
in35.txt AC 109 ms 143692 KiB
in36.txt AC 135 ms 169160 KiB
in37.txt AC 130 ms 163424 KiB
in38.txt AC 139 ms 169288 KiB
in39.txt AC 142 ms 183256 KiB
in40.txt AC 118 ms 149776 KiB
in41.txt AC 69 ms 93412 KiB
in42.txt AC 69 ms 93680 KiB
in43.txt AC 141 ms 182916 KiB
in44.txt AC 148 ms 183016 KiB
in45.txt AC 70 ms 93556 KiB
in46.txt AC 69 ms 93680 KiB
in47.txt AC 69 ms 93580 KiB
in48.txt AC 69 ms 93564 KiB
in49.txt AC 69 ms 93400 KiB
in50.txt AC 70 ms 93400 KiB
in51.txt AC 126 ms 175088 KiB
in52.txt AC 125 ms 175100 KiB
in53.txt AC 69 ms 93460 KiB
in54.txt AC 69 ms 93412 KiB
in55.txt AC 69 ms 93564 KiB
in56.txt AC 70 ms 93400 KiB
in57.txt AC 69 ms 93572 KiB
sample01.txt AC 69 ms 93528 KiB
sample02.txt AC 69 ms 93724 KiB
sample03.txt AC 69 ms 93500 KiB