結果

問題 No.1694 ZerOne
ユーザー tamato
提出日時 2021-10-01 22:22:09
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 711 ms / 2,000 ms
コード長 844 bytes
コンパイル時間 309 ms
コンパイル使用メモリ 82,112 KB
実行使用メモリ 76,512 KB
最終ジャッジ日時 2024-07-19 12:07:20
合計ジャッジ時間 8,110 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff #

mod = 1000000007
eps = 10**-9


def main():
    import sys
    input = sys.stdin.readline

    S = input().rstrip('\n')
    N = len(S)
    M = 225
    X = []
    for i in range(N):
        if S[i] == "1":
            X.append(i+1)

    dp = [[0] * (M * 2 + 1) for _ in range(N+1)]
    dp[0][0] = 1
    for x in X:
        dp_new = [[0] * (M * 2 + 1) for _ in range(N + 1)]
        for j_prev in range(N+1):
            for j_new in range(j_prev + 1, N+1):
                for balance in range(-M, M+1):
                    balance_new = balance + (j_new - x)
                    if -M <= balance_new <= M:
                        dp_new[j_new][balance_new] = dp_new[j_new][balance_new] + dp[j_prev][balance]
        dp = dp_new
    ans = 0
    for j in range(N+1):
        ans += dp[j][0]
    print(ans)


if __name__ == '__main__':
    main()
0