結果
問題 |
No.377 背景パターン
|
ユーザー |
|
提出日時 | 2017-04-24 23:31:01 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,830 ms / 5,000 ms |
コード長 | 963 bytes |
コンパイル時間 | 329 ms |
コンパイル使用メモリ | 82,312 KB |
実行使用メモリ | 79,356 KB |
最終ジャッジ日時 | 2024-09-13 12:13:51 |
合計ジャッジ時間 | 5,779 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 14 |
ソースコード
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 1 < n and all( n % i for i in range( 2, int( n ** 0.5 ) + 1 ) ) 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 ) + 1 ) if i * i != n and n % i == 0 ] hf, wf = factors( H ), factors( W ) hp, wp = [ v for v in hf if is_prime( v ) ], [ v for v in wf if is_prime( v ) ] def phi( x, memo = {} ): if x in memo: return memo[ x ] memo[ x ] = reduce( lambda a, b: a // b * ( b - 1 ), [ x ] + [ p for p in ( hp if H % x == 0 else wp ) if x % p == 0 ] ) return memo[ x ] 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 ) % MOD if ans >= MOD: ans -= MOD ans = ans * pow( H * W, MOD - 2, MOD ) % MOD print( ans )