結果
問題 |
No.498 ワープクリスタル (給料日編)
|
ユーザー |
![]() |
提出日時 | 2016-07-14 21:08:36 |
言語 | Python2 (2.7.18) |
結果 |
AC
|
実行時間 | 557 ms / 2,000 ms |
コード長 | 1,040 bytes |
コンパイル時間 | 239 ms |
コンパイル使用メモリ | 6,912 KB |
実行使用メモリ | 148,828 KB |
最終ジャッジ日時 | 2024-10-15 20:03:35 |
合計ジャッジ時間 | 7,950 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 21 |
ソースコード
def solve(): p = 10**9+7 Comb = [[0]*76 for i in xrange(76)] Comb[0][0] = 1 Gx,Gy,K = map(int,raw_input().split()) assert -10**5 <= Gx <= 10**5 assert -10**5 <= Gy <= 10**5 assert 1 <= K <= 5 G = (Gx,Gy) for i in xrange(K*15): for j in xrange(i+2): Comb[i+1][j] = (Comb[i][j] + (Comb[i][j-1] if j > 0 else 0)) % p dp = [None]*(1<<(4*K)) dp[0] = (0,0) ans = 0 check_set = set() for i in xrange(K): x,y,N = map(int,raw_input().split()) assert -10**4 <= x <= 10**4 assert -10**4 <= y <= 10**4 assert 1 <= N <= 15 assert (x,y) not in check_set check_set.add((x,y)) shift = 4*i for j in xrange(1,N+1): mx = j*x; my = j*y bit = j<<shift for k in xrange(1<<shift): if dp[k] is None: continue dp[k+bit] = (dp[k][0]+mx,dp[k][1]+my) for k in xrange(1<<(4*K)): if dp[k] != G: continue total = sum((k>>(i*4))%16 for i in xrange(K)) tmp = 1 for i in xrange(K): dup_n = (k>>(i*4))%16 tmp = tmp * Comb[total][dup_n] % p total -= dup_n ans = (ans + tmp) % p print ans solve()