結果

問題 No.377 背景パターン
コンテスト
ユーザー 0w1
提出日時 2017-04-24 22:31:36
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,251 bytes
コンパイル時間 242 ms
コンパイル使用メモリ 82,212 KB
実行使用メモリ 79,208 KB
最終ジャッジ日時 2024-09-13 12:08:42
合計ジャッジ時間 5,612 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 3
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

from functools import reduce

MOD = int( 1e9 ) + 7

H, W, K = map( int, input().split() )

def gcd( a, b ):
  if b == 0: return a
  return gcd( b, a % b )

def is_prime( n ):
  return n > 1 and all( n % i for i in range( 2, int( n ** 0.5 ) + 1 ) )

def get_prime( n ):
  return [ i for i in range( 1, int( n ** 0.5 ) + 1 ) if n % i == 0 and is_prime( i ) ] + [ n // i for i in range( 1, int( n ** 0.5 ) ) if n % i == 0 and is_prime( n // i ) ]

hp, wp = get_prime( H ), get_prime( W )

def phi( x, memo = {} ):
  if x in memo:
    return memo[ x ]
  if H % x == 0:
    res = x
    for p in hp:
      if p > x: break
      if x % p == 0:
        res = res // p * ( p - 1 )
    memo[ x ] = res
    return res
  else:
    res = x
    for p in wp:
      if p > x: break
      if x % p == 0:
        res = res // p * ( p - 1 )
    memo[ x ] = res
    return res

def factors( n ):
  return [ i for i in range( 1, int( n ** 0.5 ) + 1 ) if n % i == 0 ] + [ n // i for i in range( 1, int( n ** 0.5 ) ) if n % i == 0 ]

hf, wf = factors( H ), factors( W )

ans = 0
for a in hf:
  for b in wf:
    ans += pow( K, H // a * W // b * gcd( a, b ) % ( MOD - 1 ), MOD ) * phi( a ) * phi( b )
    ans %= MOD

ans = ans * pow( H * W, MOD - 2, MOD ) % MOD

print( ans )
0