結果

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

ソースコード

diff # 

mod = 10 ** 9 + 7


def setup(n):
    fact = [0] * n
    revFact = [0] * 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
    return fact, revFact


def solve(n):
    fact, revFact = setup(n * 2 + 1)

    def P(n, r): return fact[n] * revFact[n - r] % mod if n >= r else 0
    def C(n, r): return P(n, r) * revFact[r] % mod
    def catalan(n): return (C(2 * n, n) - C(2 * n, n + 1) + mod) % mod

    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


n = int(input())
print(solve(n))
0