結果

問題 No.1596 Distance Sum in 2D Plane
ユーザー NSNS
提出日時 2021-07-10 14:05:02
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 267 ms / 2,000 ms
コード長 1,676 bytes
コンパイル時間 173 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 88,320 KB
最終ジャッジ日時 2024-07-02 02:12:34
合計ジャッジ時間 5,166 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 100 ms
70,400 KB
testcase_01 AC 96 ms
70,656 KB
testcase_02 AC 262 ms
88,320 KB
testcase_03 AC 267 ms
88,264 KB
testcase_04 AC 258 ms
87,732 KB
testcase_05 AC 262 ms
87,664 KB
testcase_06 AC 256 ms
87,640 KB
testcase_07 AC 256 ms
88,012 KB
testcase_08 AC 254 ms
87,712 KB
testcase_09 AC 254 ms
87,624 KB
testcase_10 AC 257 ms
87,936 KB
testcase_11 AC 224 ms
88,116 KB
testcase_12 AC 231 ms
88,272 KB
testcase_13 AC 225 ms
87,784 KB
testcase_14 AC 91 ms
70,784 KB
testcase_15 AC 96 ms
70,656 KB
testcase_16 AC 93 ms
70,784 KB
testcase_17 AC 93 ms
70,400 KB
testcase_18 AC 95 ms
70,400 KB
testcase_19 AC 97 ms
70,912 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

class Combination:
    def __init__(self, n_max=10**6, mod=10**9+7):
        # self._n_max = n_max
        self._fac, self._finv, self._inv = [0]*n_max, [0]*n_max, [0]*n_max
        self._fac[0], self._fac[1] = 1, 1
        self._finv[0], self._finv[1] = 1, 1
        self._inv[1] = 1
        self._mod = mod
        for i in range(2, n_max):
            self._fac[i] = self._fac[i - 1] * i % self._mod
            self._inv[i] = self._mod - self._inv[self._mod%i] * (self._mod // i) % self._mod
            self._finv[i] = self._finv[i - 1] * self._inv[i] % self._mod
    def com(self, n, r):
        if n < r: return 0
        if n < 0 or r < 0: return 0
        return self._fac[n] * (self._finv[r] * self._finv[n - r] % self._mod) % self._mod

    def perm(self,n,r):
        if n < r: return 0
        if n < 0 or r < 0: return 0
        return self._fac[n] * (self._finv[n-r] % self._mod) % self._mod

    def lucas(self, n, r): # nCr (mod self._mod(prime)) 
        if n < r: return 0 
        res = 1
        while n > 0:
            nq, rq = n//self._mod, r//self._mod
            nr, rr = n-nq*self._mod, r-rq*self._mod
            res *= self.com(nr, rr)
            res %= self._mod
            n, r = nq, rq
        return res

MOD=10**9+7
comb=Combination(5*10**5+10,MOD)

n,m=map(int, input().split())
dist=n*2
pat=comb.com(2*n,n)
ans=dist*pat
ans%=MOD

for _ in range(m):
    t,x,y=map(int, input().split())
    if t==1:
        pat1=comb.com(x+y,x)
        pat2=comb.com(2*n-(x+y+1),n-y)
        ans-=pat1*pat2
        ans%=MOD
    if t==2:
        pat1=comb.com(x+y,x)
        pat2=comb.com(2*n-(x+y+1),n-x)
        ans-=pat1*pat2
        ans%=MOD
print(ans)
0