結果

問題 No.1596 Distance Sum in 2D Plane
ユーザー prd_xxxprd_xxx
提出日時 2021-07-10 00:11:16
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 261 ms / 2,000 ms
コード長 716 bytes
コンパイル時間 739 ms
コンパイル使用メモリ 87,280 KB
実行使用メモリ 117,988 KB
最終ジャッジ日時 2023-09-14 11:33:53
合計ジャッジ時間 5,558 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 100 ms
95,056 KB
testcase_01 AC 105 ms
94,904 KB
testcase_02 AC 259 ms
117,256 KB
testcase_03 AC 261 ms
117,280 KB
testcase_04 AC 259 ms
117,280 KB
testcase_05 AC 251 ms
117,404 KB
testcase_06 AC 252 ms
117,776 KB
testcase_07 AC 250 ms
117,612 KB
testcase_08 AC 249 ms
117,640 KB
testcase_09 AC 242 ms
117,688 KB
testcase_10 AC 249 ms
117,472 KB
testcase_11 AC 242 ms
117,988 KB
testcase_12 AC 246 ms
117,588 KB
testcase_13 AC 248 ms
117,608 KB
testcase_14 AC 69 ms
71,180 KB
testcase_15 AC 69 ms
71,416 KB
testcase_16 AC 70 ms
71,224 KB
testcase_17 AC 70 ms
71,420 KB
testcase_18 AC 70 ms
71,376 KB
testcase_19 AC 70 ms
71,356 KB
権限があれば一括ダウンロードができます

ソースコード

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