Submission #22413407

Source Code Expand

n=int(input())
a=list(map(int,input().split()))
for i in range(n):
    a[i]%=200
group_index = [-1 for j in range(n)]
group = []

index = 0
ans=0

for i in range(n):
    g_index = group_index[i]
    if g_index != -1:
        continue
    index+=1
    group_index[i] = index
    tmp = [i]
    a1 = a[i]
    for j in range(i+1,n):
        if group_index[j] != -1:
            continue
        if a1-a[j]==0:
            tmp.append(j)
            group_index[j] = index
    group.append(tmp)
for i in range(len(group)):
    l = len(group[i])
    ans += l*(l-1)//2
print(ans)

Submission Info

Submission Time
Task C - Ringo's Favorite Numbers 2
User issan
Language PyPy3 (7.3.0)
Score 300
Code Size 599 Byte
Status AC
Exec Time 373 ms
Memory 115676 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 300 / 300
Status
AC × 3
AC × 26
Set Name Test Cases
Sample sample_01.txt, sample_02.txt, sample_03.txt
All extest_01.txt, extest_02.txt, extest_03.txt, extest_04.txt, extest_05.txt, sample_01.txt, sample_02.txt, sample_03.txt, test_01.txt, test_02.txt, test_03.txt, test_04.txt, test_05.txt, test_06.txt, test_07.txt, test_08.txt, test_09.txt, test_10.txt, test_11.txt, test_12.txt, test_13.txt, test_14.txt, test_15.txt, test_16.txt, test_17.txt, test_18.txt
Case Name Status Exec Time Memory
extest_01.txt AC 152 ms 112212 KiB
extest_02.txt AC 130 ms 105024 KiB
extest_03.txt AC 133 ms 103652 KiB
extest_04.txt AC 139 ms 100696 KiB
extest_05.txt AC 139 ms 100880 KiB
sample_01.txt AC 50 ms 61764 KiB
sample_02.txt AC 54 ms 61804 KiB
sample_03.txt AC 48 ms 61784 KiB
test_01.txt AC 51 ms 61812 KiB
test_02.txt AC 51 ms 61672 KiB
test_03.txt AC 52 ms 63464 KiB
test_04.txt AC 55 ms 63040 KiB
test_05.txt AC 57 ms 64032 KiB
test_06.txt AC 56 ms 63492 KiB
test_07.txt AC 53 ms 63984 KiB
test_08.txt AC 69 ms 68540 KiB
test_09.txt AC 259 ms 90976 KiB
test_10.txt AC 73 ms 68160 KiB
test_11.txt AC 172 ms 81916 KiB
test_12.txt AC 311 ms 91152 KiB
test_13.txt AC 283 ms 94392 KiB
test_14.txt AC 69 ms 68464 KiB
test_15.txt AC 373 ms 100220 KiB
test_16.txt AC 370 ms 100280 KiB
test_17.txt AC 305 ms 100268 KiB
test_18.txt AC 126 ms 115676 KiB