結果
問題 | No.1694 ZerOne |
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
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()