結果
問題 | No.1531 く2 |
ユーザー |
![]() |
提出日時 | 2021-06-04 22:58:16 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 885 bytes |
コンパイル時間 | 450 ms |
コンパイル使用メモリ | 82,108 KB |
実行使用メモリ | 78,768 KB |
最終ジャッジ日時 | 2024-11-20 06:09:35 |
合計ジャッジ時間 | 5,112 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 WA * 3 |
other | AC * 36 WA * 8 |
ソースコード
#k桁目まで見たとき、1の個数がiこ、今j個1が連続しているものの個数N,K=map(int,input().split())s=bin(N)[2:]t=bin(K)[2:]mod=998244353if len(s)<len(t):s=(len(t)-len(s))*"0"+selse:t=(len(s)-len(t))*"0"+tM=len(s)dp=[[[0]*(M+1) for i in range(M+1)] for j in range(M+1)]dp[0][0][0]=1for k in range(M):for i in range(M):for j in range(M):if t[k]=="1":a=1else:a=0if s[k]=="0" and t[k]=="0":dp[k+1][i][0]+=dp[k][i][j]elif s[k]=="0" and t[k]=="1":dp[k+1][i+1][j+1]+=dp[k][i][j]elif s[k]=="1" and t[k]=="0":dp[k+1][i][0]+=dp[k][i][j]dp[k+1][i+1][j+1]+=dp[k][i][j]else:dp[k+1][i+1][j+1]+=dp[k][i][j]dp[k+1][i-j+1][0]+=dp[k][i][j]ans=0for i in range(M):for j in range(M):ans+=pow(2,i,mod)*dp[M][i][j]ans%=modprint(ans)