結果

問題 No.1520 Zigzag Sum
ユーザー nephrologistnephrologist
提出日時 2021-05-29 13:23:53
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 909 bytes
コンパイル時間 192 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 88,320 KB
最終ジャッジ日時 2024-04-25 10:47:56
合計ジャッジ時間 4,787 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 226 ms
87,680 KB
testcase_01 WA -
testcase_02 AC 340 ms
83,328 KB
testcase_03 TLE -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 10 ** 9 + 7
N = 4 * 10 ** 5 + 5
bikkuri = [0] * N
bikkuri[0] = 1
gyaku = [0] * N
for i in range(1, N):
    bikkuri[i] = (i * bikkuri[i - 1]) % mod
gyaku[N - 1] = pow(bikkuri[N - 1], mod - 2, mod)
for i in range(N)[::-1]:
    gyaku[i - 1] = (gyaku[i] * i) % mod


def comb(n, r):
    if r < 0 or n - r < 0:
        return 0
    return (bikkuri[n] * gyaku[r] * gyaku[n - r]) % mod


t = int(input())
for _ in range(t):
    h, w = map(int, input().split())
    h -= 1
    w -= 1

    ans = 0
    num1 = min(h, w)
    for i in range(1, num1 + 1):
        j = i - 1
        ans += 2 * comb(h - 1, j) * comb(w - 1, j) * (2 * j + 1)
        ans += comb(h - 1, j) * comb(w - 1, j - 1) * (2 * j)
        ans += comb(h - 1, j - 1) * comb(w - 1, j) * (2 * j)
        ans %= mod
    if h != w:
        num = min(h, w)
        ans += comb(num + 1 - 1, num) * comb(num - 1, num - 1) * (2 * num)
    print(ans % mod)
0