結果

問題 No.1685 One by One
ユーザー hitonanode
提出日時 2021-08-14 02:36:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,277 ms / 3,000 ms
コード長 1,867 bytes
コンパイル時間 139 ms
コンパイル使用メモリ 81,840 KB
実行使用メモリ 452,984 KB
最終ジャッジ日時 2024-06-29 19:28:35
合計ジャッジ時間 42,818 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

N, M = map(int, input().split())

md = 1000000007

ncrs = [[0] * (max(N, M) + 10) for _ in range(N + M + 1)]
ncrs[0][0] = 1
for i in range(1, len(ncrs)):
    ncrs[i][0] = 1
    for j in range(1, len(ncrs[i])):
        ncrs[i][j] = ncrs[i - 1][j - 1] + ncrs[i - 1][j]
        if ncrs[i][j] >= md:
            ncrs[i][j] -= md

precalc = [[0] * (M + 1) for _ in range(N + 1)]
precalc[0][0] = 1
for n in range(N + 1):
    for m in range(M + 1):
        if n + m - 1 >= 0:
            precalc[n][m] = ncrs[n + m - 1][n - 1]
        if m >= 2:
            precalc[n][m] += precalc[n][m - 2]
            if precalc[n][m] >= md:
                precalc[n][m] -= md

ret = 0

if N % 2 == 0:
    half = (N - 1) // 2
    for nnonzero in range(half * 2 + 1):
        coeff = 0
        for npos in range(half + 1):
            if nnonzero - npos < 0 or nnonzero - npos > half:
                continue
            coeff += ncrs[N - 1][npos] * ncrs[N - 1 - npos][nnonzero - npos]
            coeff %= md
        tmp = 0
        for b0 in range(M + 1):
            if M - nnonzero - b0 < 0:
                break
            tmp += precalc[nnonzero][M - nnonzero - b0] * (2 - (b0 == 0))
        ret += tmp % md * coeff
        ret %= md
else:
    for npos in range(N // 2 + 1):
        for nneg in range(N // 2 + 1):
            nnonzero = npos + nneg
            coeff = 0
            if M - (npos + nneg) >= 0:
                coeff += precalc[nnonzero][M - nnonzero]
            m = M - npos - (N - nneg)
            if m >= 0:
                coeff += precalc[nnonzero][m] * 2
            m = M - (N - nnonzero) - (npos if npos > nneg else nneg) * 2
            if m >= 0:
                coeff += md - precalc[nnonzero][m]
            ret += ncrs[N][nnonzero] * ncrs[nnonzero][npos] % md * coeff % md

print(ret % md)
0