[Python]bit全探索 ABC045C,ARC061C

ABC045C

$2^n$通りを探索する場合は、bit全探索が便利である。

下は、けんちょんさんのビット演算のまとめ表

Python ビット演算 超入門

サンプルコード

s = input()
n = len(s)

ans = 0

for bit in range(1 << (n - 1)): # n-1左シフト
    f = s[0]

    for i in range(n - 1):
        # フラグが立っている所に+を挿入
        if bit & (1 << i):
            f += "+"
        f += s[i + 1]

    ans += sum(map(int, f.split("+")))

print(ans)