結果

問題 No.2954 Calculation of Exponentiation
コンテスト
ユーザー dp_ijk
提出日時 2024-11-08 22:26:01
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 162 ms / 2,000 ms
コード長 799 bytes
コンパイル時間 354 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 88,704 KB
最終ジャッジ日時 2024-11-08 22:26:08
合計ジャッジ時間 6,315 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

def factors(X: int):
   assert X >= 1
   F: list[tuple[int, int]] = []
   d = 2
   while d*d <= X:
      e = 0
      while X%d == 0:
         e += 1
         X //= d
      if e > 0:
         F.append((d, e))
      d += 1
   if X > 1:
      F.append((X, 1))
   return F


from fractions import Fraction

A, B = map(Fraction, input().split())


def solve(A, B):
   if B == 0: return True
   if B > 0:
      p, q = A.as_integer_ratio()
      if q != 1:
         return False
      x, y = B.as_integer_ratio()
      for _, e in factors(A):
         g = Fraction(e*x, y)
         if g.as_integer_ratio()[1] != 1 or g < 0:
            return False
   if B < 0:
      return solve(1/A, -B)
   return True


ans = solve(A, B)
if ans == True:
   print("Yes")
elif ans == False:
   print("No")
else:
   raise
0