結果

問題 No.1596 Distance Sum in 2D Plane
ユーザー prd_xxx
提出日時 2021-07-10 00:11:16
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 264 ms / 2,000 ms
コード長 716 bytes
コンパイル時間 382 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 117,496 KB
最終ジャッジ日時 2024-07-01 19:01:39
合計ジャッジ時間 4,913 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = sys.stdin.readline
N,M = map(int,input().split())
TXY = [tuple(map(int,input().split())) for i in range(M)]
MOD = 10**9+7

MAXN = N+N+5
fac = [1,1] + [0]*MAXN
finv = [1,1] + [0]*MAXN
inv = [0,1] + [0]*MAXN
for i in range(2,MAXN+2):
    fac[i] = fac[i-1] * i % MOD
    inv[i] = -inv[MOD%i] * (MOD // i) % MOD
    finv[i] = finv[i-1] * inv[i] % MOD

def comb(n,r):
    if n < r: return 0
    if n < 0 or r < 0: return 0
    return fac[n] * (finv[r] * finv[n-r] % MOD) % MOD

ans = (comb(2*N, N) * 2*N) % MOD

for t,x1,y1 in TXY:
    if t==1:
        x2,y2 = x1+1,y1
    else:
        x2,y2 = x1,y1+1
    c = comb(x1+y1, x1) * comb(N-x2+N-y2, N-x2)
    c %= MOD
    ans -= c
    ans %= MOD
print(ans)
0