Submission #66533028


Source Code Expand

Copy
n,l=map(int,input().split())
a=list(map(int,input().split()))
from itertools import accumulate
if l%3:exit(print(0))
d=list(accumulate(a,initial=0))
cnt=[0]*l
for i in range(n):
d[i]%=l
cnt[d[i]]+=1
l3=l//3
ans=0
for i in range(l):
ans+=cnt[i]*cnt[(i+l3)%l]*cnt[(i+2*l3)%l]
print(ans//3)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
n,l=map(int,input().split())
a=list(map(int,input().split()))
from itertools import accumulate
if l%3:exit(print(0))
d=list(accumulate(a,initial=0))
cnt=[0]*l
for i in range(n):
    d[i]%=l
    cnt[d[i]]+=1
l3=l//3
ans=0
for i in range(l):
    ans+=cnt[i]*cnt[(i+l3)%l]*cnt[(i+2*l3)%l]
print(ans//3)

Submission Info

Submission Time
Task C - Equilateral Triangle
User juten
Language Python (PyPy 3.10-v7.3.12)
Score 300
Code Size 312 Byte
Status AC
Exec Time 108 ms
Memory 150332 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 300 / 300
Status
AC × 3
AC × 37
Set Name Test Cases
Sample 00_sample_00.txt, 00_sample_01.txt, 00_sample_02.txt
All 00_sample_00.txt, 00_sample_01.txt, 00_sample_02.txt, 01_test_00.txt, 01_test_01.txt, 01_test_02.txt, 01_test_03.txt, 01_test_04.txt, 01_test_05.txt, 01_test_06.txt, 01_test_07.txt, 01_test_08.txt, 01_test_09.txt, 01_test_10.txt, 01_test_11.txt, 01_test_12.txt, 01_test_13.txt, 01_test_14.txt, 01_test_15.txt, 01_test_16.txt, 01_test_17.txt, 01_test_18.txt, 01_test_19.txt, 01_test_20.txt, 01_test_21.txt, 01_test_22.txt, 01_test_23.txt, 01_test_24.txt, 01_test_25.txt, 01_test_26.txt, 01_test_27.txt, 01_test_28.txt, 01_test_29.txt, 01_test_30.txt, 01_test_31.txt, 01_test_32.txt, 01_test_33.txt
Case Name Status Exec Time Memory
00_sample_00.txt AC 55 ms 76760 KiB
00_sample_01.txt AC 55 ms 76620 KiB
00_sample_02.txt AC 55 ms 76516 KiB
01_test_00.txt AC 55 ms 76472 KiB
01_test_01.txt AC 59 ms 81496 KiB
01_test_02.txt AC 73 ms 106208 KiB
01_test_03.txt AC 101 ms 149088 KiB
01_test_04.txt AC 57 ms 81080 KiB
01_test_05.txt AC 60 ms 82036 KiB
01_test_06.txt AC 91 ms 131460 KiB
01_test_07.txt AC 101 ms 145732 KiB
01_test_08.txt AC 59 ms 82956 KiB
01_test_09.txt AC 61 ms 82684 KiB
01_test_10.txt AC 95 ms 133660 KiB
01_test_11.txt AC 107 ms 139672 KiB
01_test_12.txt AC 59 ms 83332 KiB
01_test_13.txt AC 64 ms 84092 KiB
01_test_14.txt AC 80 ms 110388 KiB
01_test_15.txt AC 108 ms 138852 KiB
01_test_16.txt AC 55 ms 76232 KiB
01_test_17.txt AC 57 ms 80656 KiB
01_test_18.txt AC 74 ms 107204 KiB
01_test_19.txt AC 91 ms 130884 KiB
01_test_20.txt AC 88 ms 126236 KiB
01_test_21.txt AC 88 ms 125856 KiB
01_test_22.txt AC 88 ms 126048 KiB
01_test_23.txt AC 88 ms 125864 KiB
01_test_24.txt AC 88 ms 125964 KiB
01_test_25.txt AC 88 ms 125960 KiB
01_test_26.txt AC 103 ms 134800 KiB
01_test_27.txt AC 91 ms 131000 KiB
01_test_28.txt AC 89 ms 125760 KiB
01_test_29.txt AC 88 ms 126184 KiB
01_test_30.txt AC 95 ms 150332 KiB
01_test_31.txt AC 95 ms 150108 KiB
01_test_32.txt AC 95 ms 149920 KiB
01_test_33.txt AC 54 ms 76804 KiB


2025-06-07 (Sat)
22:45:58 +09:00