結果

問題 No.3274 Arithmetic Right Shift
コンテスト
ユーザー 👑 loop0919
提出日時 2024-09-30 11:24:09
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 182 ms / 1,500 ms
コード長 532 bytes
コンパイル時間 421 ms
コンパイル使用メモリ 81,980 KB
実行使用メモリ 77,568 KB
最終ジャッジ日時 2024-09-30 11:24:12
合計ジャッジ時間 2,133 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

T = int(input())
for _ in range(T):
    K = int(input())
    S = input().strip()
    N = len(S)
    if S[0] == '0':
        val = int(S, 2)
        if K >= N:
            val_shifted = 0
        else:
            val_shifted = val >> K
        S_result = bin(val_shifted)[2:].zfill(N)
    else:
        val = int(S, 2) - (1 << N)
        if K >= N:
            val_shifted = -1
        else:
            val_shifted = val >> K
        val_shifted &= (1 << N) - 1
        S_result = bin(val_shifted)[2:].zfill(N)
    print(S_result)
0