結果
問題 |
No.498 ワープクリスタル (給料日編)
|
ユーザー |
![]() |
提出日時 | 2016-03-16 22:02:36 |
言語 | PyPy2 (7.3.15) |
結果 |
AC
|
実行時間 | 164 ms / 2,000 ms |
コード長 | 1,302 bytes |
コンパイル時間 | 1,946 ms |
コンパイル使用メモリ | 76,736 KB |
実行使用メモリ | 104,548 KB |
最終ジャッジ日時 | 2024-10-15 19:45:28 |
合計ジャッジ時間 | 5,797 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 21 |
ソースコード
from math import log def solve(): p = 10**9+7; F = [1]; FI = [0]*16 for i in xrange(75): F.append((i+1)*F[i]%p) FI[15] = pow(F[15],p-2,p) for i in xrange(15,0,-1): FI[i-1] = i*FI[i]%p Gx,Gy,K = map(int,raw_input().split()) G = (Gx,Gy); dp = [None]*(1<<K*4); dp[0] = (0,0) used = set() for i in xrange(K): x,y,N = map(int,raw_input().split()) mul = 1; shift = i*4 for j in xrange(int(log(N,2))): mx,my = x*mul,y*mul for k in xrange((1<<j+shift)-1,-1,-1): if dp[k] is None: continue nk = k+(mul<<shift) nxt = (dp[k][0]+mx,dp[k][1]+my) dp[nk]=nxt if nxt == G: used.add(nk) N -= mul; mul *= 2 if N > 0: mx,my = x*N,y*N for k in xrange((1<<4+shift)-1,-1,-1): if dp[k] is None: continue nk = k+(N<<shift) nxt = (dp[k][0]+mx,dp[k][1]+my) dp[nk]=nxt if nxt == G: used.add(nk) ans = 0 if G != (0,0) else 1 for k in used: total = 0; deno = 1 for j in xrange(K): dup_n = k%16; k>>=4 total += dup_n; deno=deno*FI[dup_n]%p ans = (ans + F[total]*deno)%p print ans solve()