結果

問題 No.1531 く2
ユーザー kozy
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

#k桁目まで見たとき、1の個数がiこ、今j個1が連続しているものの個数
N,K=map(int,input().split())
s=bin(N)[2:]
t=bin(K)[2:]
mod=998244353
if len(s)<len(t):
  s=(len(t)-len(s))*"0"+s
else:
  t=(len(s)-len(t))*"0"+t
M=len(s)
dp=[[[0]*(M+1) for i in range(M+1)] for j in range(M+1)]
dp[0][0][0]=1
for k in range(M):
  for i in range(M):
    for j in range(M):
      if t[k]=="1":
        a=1
      else:
        a=0
      if 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=0
for i in range(M):
  for j in range(M):
    ans+=pow(2,i,mod)*dp[M][i][j]
    ans%=mod
print(ans)
0