結果

問題 No.2782 メルセンヌ数総乗
ユーザー katonyonkokatonyonko
提出日時 2024-06-14 21:30:29
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 51 ms / 2,000 ms
コード長 3,442 bytes
コンパイル時間 293 ms
コンパイル使用メモリ 82,256 KB
実行使用メモリ 62,404 KB
最終ジャッジ日時 2024-06-14 21:30:32
合計ジャッジ時間 2,360 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 47 ms
55,800 KB
testcase_01 AC 44 ms
56,728 KB
testcase_02 AC 45 ms
56,476 KB
testcase_03 AC 49 ms
61,616 KB
testcase_04 AC 46 ms
55,428 KB
testcase_05 AC 46 ms
56,240 KB
testcase_06 AC 45 ms
56,920 KB
testcase_07 AC 45 ms
56,860 KB
testcase_08 AC 51 ms
56,360 KB
testcase_09 AC 47 ms
57,316 KB
testcase_10 AC 46 ms
56,348 KB
testcase_11 AC 46 ms
56,776 KB
testcase_12 AC 49 ms
56,680 KB
testcase_13 AC 44 ms
56,348 KB
testcase_14 AC 46 ms
56,092 KB
testcase_15 AC 46 ms
55,564 KB
testcase_16 AC 44 ms
57,024 KB
testcase_17 AC 50 ms
62,404 KB
testcase_18 AC 44 ms
57,056 KB
testcase_19 AC 48 ms
62,076 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
1
2
5
31
"""

#素因数分解はここからコピーするのが多分速い
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)

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

def solve(test):
  N=int(input())
  f=primeFactor(2**(N+1)-1)
  for n in range(2,N+1):
    f2=primeFactor(2**n-1)
    for k in f2:
      if k in f:
        f[k]-=f2[k]
  if len([k for k in f if f[k]>0])==0:
    print('Yes')
  else:
    print('No')

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