結果
問題 | No.1596 Distance Sum in 2D Plane |
ユーザー |
![]() |
提出日時 | 2023-04-18 22:41:01 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 161 ms / 2,000 ms |
コード長 | 1,152 bytes |
コンパイル時間 | 220 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 82,560 KB |
最終ジャッジ日時 | 2024-10-13 22:34:43 |
合計ジャッジ時間 | 3,573 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
from sys import stdin input=lambda :stdin.readline()[:-1] mod=10**9+7 table_size=4*10**5 fac=[1]*(table_size+1) finv=[1]*(table_size+1) for i in range(2,table_size+1): fac[i]=fac[i-1]*i%mod finv[table_size]=pow(fac[table_size],mod-2,mod) for i in range(table_size-1,-1,-1): finv[i]=finv[i+1]*(i+1)%mod def binom(n,k): if n<0 or k<0: return 0 if k>n: return 0 global table_size,fac,finv if n>table_size: fac+=[0]*(n-table_size) finv+=[0]*(n-table_size) for i in range(table_size+1,n+1): fac[i]=fac[i-1]*i%mod finv[n]=inv(fac[n]) for i in range(n-1,table_size,-1): finv[i]=finv[i+1]*(i+1)%mod table_size=n return (fac[n]*finv[k]%mod)*finv[n-k]%mod def fpow(x,k): res=1 while k: if k&1: res=res*x%mod x=x*x%mod k>>=1 return res def inv(a): if a<table_size: return fac[a-1]*finv[a]%mod return fpow(a,mod-2) n,m=map(int,input().split()) ans=0 for i in range(m): t,x,y=map(int,input().split()) if t==1: ans-=binom(x+y,x)*binom(2*n-(x+y+1),n-(x+1)) else: ans-=binom(x+y,x)*binom(2*n-(x+y+1),n-(y+1)) ans%=mod ans+=binom(2*n,n)*2*n print(ans%mod)