結果
問題 | No.1596 Distance Sum in 2D Plane |
ユーザー |
![]() |
提出日時 | 2021-07-10 15:54:34 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 1,778 ms / 2,000 ms |
コード長 | 865 bytes |
コンパイル時間 | 282 ms |
コンパイル使用メモリ | 12,928 KB |
実行使用メモリ | 81,408 KB |
最終ジャッジ日時 | 2024-07-02 02:33:31 |
合計ジャッジ時間 | 22,788 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
N, M = map(int, input().split())TXY = [list(map(int, input().split())) for _ in range(M)]# 逆元を利用してcomb(nCk), perm(nPk)をO(1)計算MOD = 10 ** 9 + 7num = 2 * N + 1fac = [1] * (num + 1)inv = [1] * (num + 1)for j in range(1, num + 1):fac[j] = (fac[j - 1] * j) % MODinv[num] = pow(fac[num], MOD-2, MOD)for j in range(num - 1, -1, -1):inv[j] = (inv[j + 1] * (j + 1)) % MODdef comb(n, r):if r > n or n < 0 or r < 0:return 0return (fac[n] * inv[n - r] * inv[r]) % MODdef perm(n, r):if r > n or n < 0 or r < 0:return 0return comb(n, r) * fac[r] % MODans = comb(2 * N, N) * 2 * Nfor t, x, y in TXY:if t == 1:ans -= comb(x + y, x) * comb(2 * N - (x + y) - 1, N - y)else:ans -= comb(x + y, x) * comb(2 * N - (x + y) - 1, N - x)ans %= MODprint(ans % MOD)