Submission #69686654


Source Code Expand

import sys
input = sys.stdin.readline

def sieve(n):
  is_p = [True]*(n+1)
  is_p[0]=is_p[1]=False
  for i in range(2,int(n**0.5)+1):
    if is_p[i]:
      step = i
      s = i*i
      is_p[s:n+1:step] = [False]*(((n - s)//step) + 1)
  return [i for i,v in enumerate(is_p) if v]

def build_ep_fact(maxn, primes):
  ep = {}
  for p in primes:
    a = [0]*(maxn+1)
    for x in range(1, maxn+1):
      t = x
      c = 0
      while t % p == 0:
        c += 1
        t //= p
      a[x] = a[x-1] + c
    ep[p] = a
  return ep

t, mod = map(int, input().split())

primes = sieve(5005)
e = build_ep_fact(5005, primes)

print(len(primes))
for _ in range(t):
  n = int(input())
  C = list(map(int, input().split()))
  sum_ = sum(C)
  ans = 1
  for p in primes:
    if p>sum_: break
    t = e[p][sum_]
    for c in C:
      t -= e[p][c]
    ans *= pow(p, t, mod)
    ans %= mod
  print(ans)

Submission Info

Submission Time
Task E - Count Sequences 2
User uparupaaa
Language Python (PyPy 3.10-v7.3.12)
Score 0
Code Size 927 Byte
Status WA
Exec Time 2215 ms
Memory 112144 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 450
Status
WA × 2
WA × 19
TLE × 32
Set Name Test Cases
Sample 00_sample_00.txt, 00_sample_01.txt
All 00_sample_00.txt, 00_sample_01.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, 01_test_34.txt, 01_test_35.txt, 01_test_36.txt, 01_test_37.txt, 01_test_38.txt, 01_test_39.txt, 01_test_40.txt, 01_test_41.txt, 01_test_42.txt, 01_test_43.txt, 01_test_44.txt, 01_test_45.txt, 01_test_46.txt, 01_test_47.txt, 01_test_48.txt
Case Name Status Exec Time Memory
00_sample_00.txt WA 85 ms 100212 KiB
00_sample_01.txt WA 88 ms 100652 KiB
01_test_00.txt WA 1578 ms 109824 KiB
01_test_01.txt TLE 2120 ms 111196 KiB
01_test_02.txt TLE 2212 ms 112144 KiB
01_test_03.txt WA 1610 ms 110232 KiB
01_test_04.txt TLE 2213 ms 110112 KiB
01_test_05.txt TLE 2213 ms 110552 KiB
01_test_06.txt WA 1576 ms 109616 KiB
01_test_07.txt TLE 2212 ms 110788 KiB
01_test_08.txt TLE 2212 ms 110944 KiB
01_test_09.txt WA 1563 ms 110172 KiB
01_test_10.txt TLE 2212 ms 110596 KiB
01_test_11.txt TLE 2212 ms 111160 KiB
01_test_12.txt WA 1573 ms 110024 KiB
01_test_13.txt TLE 2212 ms 110956 KiB
01_test_14.txt TLE 2212 ms 111132 KiB
01_test_15.txt WA 1599 ms 110252 KiB
01_test_16.txt TLE 2213 ms 110516 KiB
01_test_17.txt TLE 2212 ms 110996 KiB
01_test_18.txt WA 1601 ms 109916 KiB
01_test_19.txt TLE 2212 ms 110916 KiB
01_test_20.txt TLE 2212 ms 111068 KiB
01_test_21.txt WA 1575 ms 110048 KiB
01_test_22.txt TLE 2212 ms 110612 KiB
01_test_23.txt TLE 2212 ms 110824 KiB
01_test_24.txt WA 1561 ms 110336 KiB
01_test_25.txt TLE 2215 ms 110684 KiB
01_test_26.txt TLE 2212 ms 110752 KiB
01_test_27.txt WA 1578 ms 110088 KiB
01_test_28.txt TLE 2212 ms 111244 KiB
01_test_29.txt TLE 2212 ms 110792 KiB
01_test_30.txt WA 1564 ms 110024 KiB
01_test_31.txt TLE 2128 ms 110400 KiB
01_test_32.txt TLE 2215 ms 111184 KiB
01_test_33.txt WA 1571 ms 109924 KiB
01_test_34.txt TLE 2117 ms 110296 KiB
01_test_35.txt TLE 2212 ms 110956 KiB
01_test_36.txt WA 1574 ms 109876 KiB
01_test_37.txt TLE 2061 ms 110796 KiB
01_test_38.txt TLE 2212 ms 112104 KiB
01_test_39.txt WA 1560 ms 110096 KiB
01_test_40.txt TLE 2130 ms 110432 KiB
01_test_41.txt TLE 2212 ms 111308 KiB
01_test_42.txt WA 1538 ms 109832 KiB
01_test_43.txt TLE 2076 ms 110568 KiB
01_test_44.txt TLE 2212 ms 111212 KiB
01_test_45.txt WA 1549 ms 109772 KiB
01_test_46.txt WA 1904 ms 110324 KiB
01_test_47.txt TLE 2212 ms 111488 KiB
01_test_48.txt TLE 2213 ms 111452 KiB