結果

問題 No.1952 xooooooooooor
ユーザー ntuda
提出日時 2022-06-03 16:43:19
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 41 ms / 2,000 ms
コード長 1,042 bytes
コンパイル時間 162 ms
コンパイル使用メモリ 82,232 KB
実行使用メモリ 52,736 KB
最終ジャッジ日時 2024-09-21 02:20:10
合計ジャッジ時間 2,852 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

N, M = map(int, input().split())
n = N
NB = []
while n > 0:
    NB.append(n & 1)
    n >>= 1
x = len(NB)
if x > M:
    # 計算量が大きくならない理由
    # x <= 30、 x >= M
    # 　よって y < 60 なので下のfor文を回しても問題ない
    y = x + M - 1
    ans = 0
    b = 1
    for i in range(y):
        start = max(0, i - M + 1)
        end = min(i, x) + 1
        a = sum(NB[start:end]) & 1
        ans += a * b
        ans %= MOD
        #print(i,a,b,a*b)
        b <<= 1
else:
    # x-1桁目まで
    ans = 0
    b = 1
    for i in range(x-1):
        start = 0
        end = i + 1
        a = sum(NB[start:end]) & 1
        ans += a * b
        b <<= 1

    # x桁目～ M桁目まで
    ans += (sum(NB) & 1) * (pow(2, M - x+1, MOD)-1) * pow(2, x-1, MOD) % MOD

    # M+1桁 ～ x+M+1桁目まで
    b = pow(2,M,MOD)
    for i in range(x - 1):
        start = i + 1
        end = x + 1
        a = sum(NB[start:end]) & 1
        ans += a * b
        ans %= MOD
        b <<= 1

print(ans)
0