結果

問題 No.1520 Zigzag Sum
ユーザー nephrologist
提出日時 2021-05-29 13:27:14
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,008 bytes
コンパイル時間 265 ms
コンパイル使用メモリ 82,120 KB
実行使用メモリ 89,772 KB
最終ジャッジ日時 2024-11-07 21:28:37
合計ジャッジ時間 4,873 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 2 TLE * 1 -- * 4
権限があれば一括ダウンロードができます

ソースコード

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)
        if h > w:
            ans += comb(h - 1, num) * comb(w - 1, num - 1) * (2 * num)
        else:
            ans += comb(h - 1, num - 1) * comb(w - 1, num) * (2 * num)
    print(ans % mod)
0