結果
問題 |
No.1683 Robot Guidance
|
ユーザー |
|
提出日時 | 2021-09-29 07:44:42 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 346 ms / 2,000 ms |
コード長 | 1,053 bytes |
コンパイル時間 | 182 ms |
コンパイル使用メモリ | 82,052 KB |
実行使用メモリ | 123,344 KB |
最終ジャッジ日時 | 2024-07-16 05:38:06 |
合計ジャッジ時間 | 9,468 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 38 |
ソースコード
class Combination(): def __init__(self, N, MOD): self.factorial=[1]*(N+1) self.fact_inv=[1]*(N+1) self.MOD=MOD for k in range(1,N): self.factorial[k+1]=(self.factorial[k]*(k+1))%MOD self.fact_inv[N]=pow(self.factorial[N],MOD-2,MOD) for k in range(N,0,-1): self.fact_inv[k-1]=(self.fact_inv[k]*k)%MOD def cmb(self, n, r): if n<0 or r<0 or n<r: return 0 else: return (self.factorial[n]*self.fact_inv[r]*self.fact_inv[n-r])%self.MOD A,B,X,Y=map(int,input().split()) if A%2!=(X+Y)%2: print(0) exit() x_plus=(B+4)//4 y_plus=(B+3)//4 x_minus=(B+2)//4 y_minus=(B+1)//4 # print(x_plus,x_minus,y_plus,y_minus) ans=0 mod=10**9+7 C=Combination(3*10**6,mod) for a in range(A+1): b=a-X c=(A+X+Y-2*a)//2 d=(A+X-Y-2*a)//2 # print(a,b,c,d) if b<0 or c<0 or d<0: continue res=1 if a>0: res=res*C.cmb(x_plus+a-1,a)%mod if b>0: res=res*C.cmb(x_minus+b-1,b)%mod if c>0: res=res*C.cmb(y_plus+c-1,c)%mod if d>0: res=res*C.cmb(y_minus+d-1,d)%mod ans+=res print(ans%mod)