Submission #72199761

Source Code Expand

from collections import defaultdict

n = int(input())
a = list(map(int,input().split()))
aftst = defaultdict(lambda:[0,0])
aftf = defaultdict(int)
for i in a:
  if i%7 == 0:
    num = i//7
    x,y = aftst[num]
    aftst[num] = [x+1,y]
  if i%3 == 0:
    num = i//3
    x,y = aftst[num]
    aftst[num] = [x,y+1]
  if i%5 == 0:
    aftf[i//5] += 1

bfrst = defaultdict(lambda:[0,0])
ans = 0
for i in a:
  if i%7 == 0:
    num = i//7
    x,y = bfrst[num]
    ans += y*aftf[num]
    bfrst[num] = [x+1,y]
    aftst[num][0] -= 1
  if i%3 == 0:
    num = i//3
    x,y = bfrst[num]
    ans += x*aftf[num]
    bfrst[num] = [x,y+1]
    aftst[num][1] -= 1
  if i%5 == 0:
    num = i//5
    ans += aftst[num][0]*aftst[num][1]
    aftf[num] -= 1

print(ans)

Submission Info

Submission Time
Task D - Kadomatsu Subsequence
User campane
Language Python (PyPy 3.11-v7.3.20)
Score 425
Code Size 782 Byte
Status AC
Exec Time 441 ms
Memory 225560 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 425 / 425
Status
AC × 3
AC × 33
Set Name Test Cases
Sample sample_01.txt, sample_02.txt, sample_03.txt
All 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, test_19.txt, test_20.txt, test_21.txt, test_22.txt, test_23.txt, test_24.txt, test_25.txt, test_26.txt, test_27.txt, test_28.txt, test_29.txt, test_30.txt
Case Name Status Exec Time Memory
sample_01.txt AC 73 ms 93948 KiB
sample_02.txt AC 74 ms 93932 KiB
sample_03.txt AC 74 ms 93740 KiB
test_01.txt AC 73 ms 93840 KiB
test_02.txt AC 73 ms 93904 KiB
test_03.txt AC 74 ms 93732 KiB
test_04.txt AC 73 ms 93904 KiB
test_05.txt AC 163 ms 162452 KiB
test_06.txt AC 174 ms 168360 KiB
test_07.txt AC 166 ms 168176 KiB
test_08.txt AC 163 ms 162080 KiB
test_09.txt AC 165 ms 162396 KiB
test_10.txt AC 161 ms 162476 KiB
test_11.txt AC 159 ms 162632 KiB
test_12.txt AC 148 ms 162764 KiB
test_13.txt AC 150 ms 162768 KiB
test_14.txt AC 171 ms 162296 KiB
test_15.txt AC 168 ms 161896 KiB
test_16.txt AC 162 ms 162660 KiB
test_17.txt AC 169 ms 162664 KiB
test_18.txt AC 178 ms 162692 KiB
test_19.txt AC 171 ms 162492 KiB
test_20.txt AC 259 ms 164940 KiB
test_21.txt AC 207 ms 163120 KiB
test_22.txt AC 271 ms 164804 KiB
test_23.txt AC 400 ms 188456 KiB
test_24.txt AC 441 ms 189204 KiB
test_25.txt AC 414 ms 178708 KiB
test_26.txt AC 287 ms 199244 KiB
test_27.txt AC 289 ms 200296 KiB
test_28.txt AC 330 ms 225560 KiB
test_29.txt AC 292 ms 200496 KiB
test_30.txt AC 289 ms 200364 KiB