結果

問題 No.1684 Find Brackets
コンテスト
ユーザー Kude
提出日時 2021-09-17 23:19:50
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,137 bytes
コンパイル時間 190 ms
コンパイル使用メモリ 82,200 KB
実行使用メモリ 125,812 KB
最終ジャッジ日時 2024-06-29 22:09:07
合計ジャッジ時間 4,254 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other AC * 1 WA * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 10 ** 9 + 7

fact_range = 2 * 10**6 + 100
fact = [1] * (fact_range + 1)
for i in range(0, fact_range):
    fact[i+1] = fact[i] * (i + 1) % MOD

ifact = [1] * (fact_range + 1)
ifact[fact_range] = pow(fact[fact_range], MOD - 2, MOD)
for i in range(fact_range, 0, -1):
    ifact[i-1] = ifact[i] * i % MOD

def comb(n, k):
    if k == 0:
        return 1
    if k < 0 or n < k:
        return 0
    else:
        return fact[n] * ifact[n-k] % MOD * ifact[k] % MOD

n, m = map(int, input().split())
s = [0] * (n + 2)
for i in range(n + 1)[::-1]:
    s[i] = s[i + 1] + comb(n, i)
    s[i] %= MOD

ans = s[m] * n % MOD
# print(ans)
# touch y = x + k
for k in range(1, n - m + 1):
    ans += s[m + k] # m + k <= n,  k <= n - m + 1
    ans %= MOD
    # print(k, ans, s[m + k], m - k)

# print('aa', ans)
t = [0] * (n + 2)
for i in range(n + 1)[::-1]:
    t[i] = t[i + 1] + comb(n, i) * max(0, 2 * i - n)
    t[i] %= MOD
u = [0] * (n + 2)
for i in range(n + 1)[::-1]:
    u[i] = u[i + 1] + comb(n, i) * max(0, 2 * i - n - 1)
    u[i] %= MOD

for k in range(1, n - m + 1):
    ans += t[m + k] - max(0, u[m + k - 1])
    ans %= MOD

print(ans)
0