結果

問題 No.1684 Find Brackets
ユーザー KudeKude
提出日時 2021-09-17 23:19:50
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,137 bytes
コンパイル時間 897 ms
コンパイル使用メモリ 87,160 KB
実行使用メモリ 131,824 KB
最終ジャッジ日時 2023-09-12 09:25:21
合計ジャッジ時間 6,067 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 AC 116 ms
107,064 KB
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
権限があれば一括ダウンロードができます

ソースコード

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