結果

問題 No.1596 Distance Sum in 2D Plane
ユーザー NSNS
提出日時 2021-07-10 14:04:24
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,667 bytes
コンパイル時間 278 ms
コンパイル使用メモリ 87,064 KB
実行使用メモリ 89,368 KB
最終ジャッジ日時 2023-09-14 19:12:31
合計ジャッジ時間 6,126 ms
ジャッジサーバーID
(参考情報)
judge12 / judge11
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 125 ms
87,392 KB
testcase_02 AC 325 ms
88,912 KB
testcase_03 AC 284 ms
89,108 KB
testcase_04 AC 289 ms
89,136 KB
testcase_05 AC 284 ms
89,204 KB
testcase_06 AC 281 ms
89,368 KB
testcase_07 AC 283 ms
89,244 KB
testcase_08 AC 283 ms
89,056 KB
testcase_09 AC 283 ms
89,200 KB
testcase_10 AC 282 ms
89,140 KB
testcase_11 AC 254 ms
89,148 KB
testcase_12 AC 253 ms
89,020 KB
testcase_13 AC 252 ms
89,172 KB
testcase_14 AC 128 ms
87,432 KB
testcase_15 AC 126 ms
87,308 KB
testcase_16 AC 128 ms
87,540 KB
testcase_17 AC 131 ms
87,320 KB
testcase_18 AC 126 ms
87,464 KB
testcase_19 AC 128 ms
87,344 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

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