結果
問題 | No.1596 Distance Sum in 2D Plane |
ユーザー |
|
提出日時 | 2021-07-09 21:59:23 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 283 ms / 2,000 ms |
コード長 | 645 bytes |
コンパイル時間 | 169 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 82,816 KB |
最終ジャッジ日時 | 2024-07-01 16:20:21 |
合計ジャッジ時間 | 4,794 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
MOD = 10**9+7n, M = map(int, input().split())N = n*2fact = [1]*(N+1)rfact = [1]*(N+1)for i in range(1, N+1):fact[i] = fact[i-1]*i % MODrfact[N] = pow(fact[N], MOD-2, MOD)for i in range(N, 0, -1):rfact[i-1] = rfact[i] * i % MOD# nPk (mod MOD) を求めるdef perm(n, k):return fact[n] * rfact[n-k] % MOD# nCk (mod MOD) を求めるdef comb(n, k):return fact[n] * rfact[k] * rfact[n-k] % MODans = N*comb(N,n)%MODfor i in range(M):t,x,y = map(int, input().split())if t==1:ans -= comb(x+y,x)*comb(N-x-y-1, n-y)%MODelse:ans -= comb(x+y,x)*comb(N-x-y-1, n-x)%MODans %= MODprint(ans)