ABC045 C - たくさんの数式
備忘録
問題
回答
import sys import os import math import bisect import collections import itertools import heapq import re import queue # import fractions ii = lambda: int(sys.stdin.buffer.readline().rstrip()) il = lambda: list(map(int, sys.stdin.buffer.readline().split())) fl = lambda: list(map(float, sys.stdin.buffer.readline().split())) iln = lambda n: [int(sys.stdin.buffer.readline().rstrip()) for _ in range(n)] iss = lambda: sys.stdin.buffer.readline().decode().rstrip() sl = lambda: list(map(str, sys.stdin.buffer.readline().decode().split())) isn = lambda n: [sys.stdin.buffer.readline().decode().rstrip() for _ in range(n)] lcm = lambda x, y: (x * y) // math.gcd(x, y) # lcm = lambda x, y: (x * y) // fractions.gcd(x, y) MOD = 10 ** 9 + 7 MAX = float('inf') def main(): if os.getenv("LOCAL"): sys.stdin = open("input.txt", "r") S = iss() N = len(S)-1 ret = 0 for n in range(1 << N): fml = '' for m in range(len(S)): fml += S[m] if n >> m & 1: fml += '+' ret += eval(fml) print(ret) if __name__ == '__main__': main()
考え方
文字列S
を構成する各数字の間に+
を入れるか入れないかの2択を全て試すことで、
回答が得られそうだったので、bit全探索を行った。
実装は1
を文字列Sの長さ - 1
だけ左にビットシフトした範囲を探索し、
文字列の数値を頭から順に別の文字列に結合。
数値の位置のbit
が立っている場合には文字列に+
を結合し、
eval
で式として処理した。