提出 #6236077

---

ソースコード 拡げる

import sys
input = sys.stdin.readline
import numpy as np

# 0 の個数 - 1 の個数 で累積和を見る
# 最大値、最小値、現在地を持つ
# とパラメータが多いので、現在地が0になるように平行移動して持つ

N,K = map(int,input().split())
S = input().rstrip()

MOD = 10 ** 9 + 7
dp = np.zeros((K+1, K+1), dtype=np.int64)
dp[0,0] = 1

for s in S:
    prev = dp
    dp = np.zeros_like(prev)
    if s in '0?':
        # 最大値を更新しなかった場合
        # 最大値に近づいて最小値から遠のく
        dp[:-1,1:] += prev[1:,:-1]
        # 最大値を更新する場合
        # 最小値からは遠のく
        dp[0,1:] += prev[0,:-1]
    if s in '1?':
        dp[1:,:-1] += prev[:-1,1:]
        dp[1:,0] += prev[:-1,0]
    dp %= MOD

select = (np.arange(K+1)[:,None] + np.arange(K+1)[None,:] <= K)
answer = dp[select].sum() % MOD
print(answer)

提出情報

ジャッジ結果