結果

問題 No.660 家を通り過ぎないランダムウォーク問題
ユーザー しらっ亭
提出日時 2017-09-11 08:08:04
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 673 ms / 2,000 ms
コード長 994 bytes
コンパイル時間 267 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 71,808 KB
最終ジャッジ日時 2024-11-07 16:47:40
合計ジャッジ時間 8,038 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 45
権限があれば一括ダウンロードができます

ソースコード

diff #

mod = 10 ** 9 + 7

MAX_N = 500000
MAX_N2 = MAX_N * 2 + 1

fact = [0] * MAX_N2
revFact = [0] * MAX_N2
inv = [0] * MAX_N2


def setup(n):
    fact[0] = 1
    for i in range(1, n):
        fact[i] = fact[i - 1] * i % mod

    revFact[n - 1] = pow(fact[n - 1], mod - 2, mod)
    for i in range(n - 2, -1, -1):
        revFact[i] = revFact[i + 1] * (i + 1) % mod

    inv[1] = 1
    for i in range(2, n):
        inv[i] = (mod - mod // i) * inv[mod % i] % mod


def C(n, r):
    return fact[n] * revFact[n - r] % mod * revFact[r] % mod


def catalan(n):
    return (C(2 * n, n) - C(2 * n, n + 1) + mod) % mod


def solve(n):
    ans = 0
    m = n // 2

    for i in range(m+1):
        ans += C(n + 2 * i, i)
    ans %= mod

    csum = 0

    for i in range(1, m + 1):
        j = m - i
        csum += catalan(i - 1)
        csum %= mod
        ans += mod - C(n + 2 * j, j) * csum * 2 % mod

    return ans % mod


def main():
    n = int(input())
    setup(n * 2 + 1)
    print(solve(n))


main()
0