結果

問題 No.2896 Monotonic Prime Factors
ユーザー katonyonkokatonyonko
提出日時 2024-09-21 20:51:01
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 625 ms / 2,000 ms
コード長 3,638 bytes
コンパイル時間 454 ms
コンパイル使用メモリ 82,236 KB
実行使用メモリ 246,784 KB
最終ジャッジ日時 2024-09-21 20:51:15
合計ジャッジ時間 10,329 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 306 ms
246,588 KB
testcase_01 AC 286 ms
246,784 KB
testcase_02 AC 283 ms
246,528 KB
testcase_03 AC 282 ms
246,656 KB
testcase_04 AC 625 ms
246,528 KB
testcase_05 AC 443 ms
246,400 KB
testcase_06 AC 461 ms
246,528 KB
testcase_07 AC 553 ms
246,784 KB
testcase_08 AC 608 ms
246,656 KB
testcase_09 AC 436 ms
246,400 KB
testcase_10 AC 420 ms
246,656 KB
testcase_11 AC 385 ms
246,592 KB
testcase_12 AC 350 ms
246,400 KB
testcase_13 AC 415 ms
246,400 KB
testcase_14 AC 473 ms
246,656 KB
testcase_15 AC 324 ms
246,400 KB
testcase_16 AC 388 ms
246,720 KB
testcase_17 AC 374 ms
246,784 KB
testcase_18 AC 471 ms
246,400 KB
testcase_19 AC 356 ms
246,592 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

import io
import sys
from collections import defaultdict, deque, Counter
from itertools import permutations, combinations, accumulate
from heapq import heappush, heappop
from bisect import bisect_right, bisect_left
from math import gcd
import math

_INPUT = """\
6
3
12 2
15 3
7 7
16
10 1
3465 13
121 4
2450 1
4620 3
2310 2
40425 7
245 13
11550 13
98 15
16940 3
20 13
9091 5
49 20
33 28
84 31415
"""

def gcd(a, b):
  while b: a, b = b, a % b
  return a
def lcm(a, b):
  return a // gcd(a, b) * b
def isPrimeMR(n):
  d = n - 1
  d = d // (d & -d)
  L = [2, 7, 61] if n < 1<<32 else [2, 3, 5, 7, 11, 13, 17] if n < 1<<48 else [2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37]
  for a in L:
    t = d
    y = pow(a, t, n)
    if y == 1: continue
    while y != n - 1:
      y = y * y % n
      if y == 1 or t == n - 1: return 0
      t <<= 1
  return 1
def findFactorRho(n):
  m = 1 << n.bit_length() // 8
  for c in range(1, 99):
      f = lambda x: (x * x + c) % n
      y, r, q, g = 2, 1, 1, 1
      while g == 1:
          x = y
          for i in range(r):
              y = f(y)
          k = 0
          while k < r and g == 1:
              ys = y
              for i in range(min(m, r - k)):
                  y = f(y)
                  q = q * abs(x - y) % n
              g = gcd(q, n)
              k += m
          r <<= 1
      if g == n:
          g = 1
          while g == 1:
              ys = f(ys)
              g = gcd(abs(x - ys), n)
      if g < n:
          if isPrimeMR(g): return g
          elif isPrimeMR(n // g): return n // g
          return findFactorRho(g)
def primeFactor(n):
  i = 2
  ret = {}
  rhoFlg = 0
  while i * i <= n:
      k = 0
      while n % i == 0:
          n //= i
          k += 1
      if k: ret[i] = k
      i += i % 2 + (3 if i % 3 == 1 else 1)
      if i == 101 and n >= 2 ** 20:
          while n > 1:
              if isPrimeMR(n):
                  ret[n], n = 1, 1
              else:
                  rhoFlg = 1
                  j = findFactorRho(n)
                  k = 0
                  while n % j == 0:
                      n //= j
                      k += 1
                  ret[j] = k
  if n > 1: ret[n] = 1
  if rhoFlg: ret = {x: ret[x] for x in sorted(ret)}
  return ret
def divisors(N):
  pf = primeFactor(N)
  ret = [1]
  for p in pf:
      ret_prev = ret
      ret = []
      for i in range(pf[p]+1):
          for r in ret_prev:
              ret.append(r * (p ** i))
  return sorted(ret)

mod=998244353
F=[1]
N=2*10**6+1
for i in range(N):
    F.append(F[-1]*(i+1)%mod)
I=[pow(F[-1],mod-2,mod)]
for i in range(N):
    I.append(I[-1]*(N-i)%mod)
I=I[::-1]

def input():
  return sys.stdin.readline()[:-1]

def solve(test):
  Q=int(input())
  x=0
  for i in range(Q):
    A,B=map(int,input().split())
    p=primeFactor(A)
    for k in p:
      x+=p[k]
    if x>=B:
      print(F[x-1]*I[B-1]*I[x-B]%mod)
    else:
      print(0)

def random_input():
  from random import randint,shuffle
  N=randint(1,10)
  M=randint(1,N)
  A=list(range(1,M+1))+[randint(1,M) for _ in range(N-M)]
  shuffle(A)
  return (" ".join(map(str, [N,M]))+"\n"+" ".join(map(str, A))+"\n")*3

def simple_solve():
  return []

def main(test):
  if test==0:
    solve(0)
  elif test==1:
    sys.stdin = io.StringIO(_INPUT)
    case_no=int(input())
    for _ in range(case_no):
      solve(0)
  else:
    for i in range(1000):
      sys.stdin = io.StringIO(random_input())
      x=solve(1)
      y=simple_solve()
      if x!=y:
        print(i,x,y)
        print(*[line for line in sys.stdin],sep='')
        break

#0:提出用、1:与えられたテスト用、2:ストレステスト用
main(0)
0