teratailのやつ

hoge.swift

func setBit(_ x: Int, _ n: Int) -> Int { ... }
func unsetBit(_ x: Int, _ n: Int) -> Int { ... }
func reverseBit(_ x: Int, _ n: Int) -> Int { ... }

func print8Bit(_ x: Int) { ... }

// 標準入力から整数を読み込む
// 整数以外であれば例外
func readInt() throws -> Int { ... }

do {
    let bitFunctions = [setBit, unsetBit, reverseBit]

    var digit = 0
    try (0...).lazy
        .map { _ in
            print("モードを入力（0: セット、 1: アンセット、 2: 反転）：")
            return try readInt()
        }
        .prefix { m in (0..2).contains(m) }
        .map { m in
            print("何ビット目かを入力（0～7）：")
            return (m, try readInt())
        }
        .prefix { (m, n) in (0...7).contains(n) }
        .forEach { (m, n) in
            digit = bitFunctions[m](digit, n)
            print8Bit(digit)
        }
}
catch {
    // do nothing
}