let program = [3,8,1005,8,339,1106,0,11,0,0,0,104,1,104,0,3,8,1002,8,-1,10,101,1,10,10,4,10,1008,8,0,10,4,10,1002,8,1,29,2,1108,11,10,1,1,20,10,2,107,6,10,3,8,102,-1,8,10,101,1,10,10,4,10,108,0,8,10,4,10,101,0,8,62,1006,0,29,1006,0,12,1,1101,5,10,1,2,20,10,3,8,102,-1,8,10,1001,10,1,10,4,10,1008,8,0,10,4,10,1001,8,0,99,1006,0,30,3,8,1002,8,-1,10,1001,10,1,10,4,10,1008,8,0,10,4,10,1001,8,0,124,1006,0,60,3,8,1002,8,-1,10,1001,10,1,10,4,10,1008,8,1,10,4,10,101,0,8,149,2,1007,2,10,1,1105,10,10,3,8,1002,8,-1,10,101,1,10,10,4,10,108,0,8,10,4,10,101,0,8,178,1,1108,15,10,1,1101,5,10,1,109,8,10,1006,0,20,3,8,102,-1,8,10,1001,10,1,10,4,10,108,1,8,10,4,10,101,0,8,215,1006,0,61,1006,0,16,2,1105,15,10,1006,0,50,3,8,1002,8,-1,10,1001,10,1,10,4,10,108,1,8,10,4,10,101,0,8,250,1,1003,10,10,1,9,19,10,2,1004,6,10,2,1106,2,10,3,8,1002,8,-1,10,1001,10,1,10,4,10,1008,8,1,10,4,10,101,0,8,289,1,1103,13,10,2,105,17,10,3,8,1002,8,-1,10,1001,10,1,10,4,10,108,1,8,10,4,10,1002,8,1,318,101,1,9,9,1007,9,1086,10,1005,10,15,99,109,661,104,0,104,1,21101,0,825599304340,1,21101,356,0,0,1106,0,460,21101,0,937108545948,1,21102,1,367,0,1106,0,460,3,10,104,0,104,1,3,10,104,0,104,0,3,10,104,0,104,1,3,10,104,0,104,1,3,10,104,0,104,0,3,10,104,0,104,1,21102,1,21628980315,1,21101,0,414,0,1105,1,460,21101,0,3316673539,1,21101,425,0,0,1106,0,460,3,10,104,0,104,0,3,10,104,0,104,0,21102,988753428840,1,1,21102,1,448,0,1106,0,460,21102,825544569700,1,1,21102,459,1,0,1106,0,460,99,109,2,21202,-1,1,1,21102,1,40,2,21102,491,1,3,21102,481,1,0,1105,1,524,109,-2,2106,0,0,0,1,0,0,1,109,2,3,10,204,-1,1001,486,487,502,4,0,1001,486,1,486,108,4,486,10,1006,10,518,1101,0,0,486,109,-2,2105,1,0,0,109,4,2102,1,-1,523,1207,-3,0,10,1006,10,541,21102,0,1,-3,21201,-3,0,1,22102,1,-2,2,21102,1,1,3,21102,560,1,0,1106,0,565,109,-4,2105,1,0,109,5,1207,-3,1,10,1006,10,588,2207,-4,-2,10,1006,10,588,22101,0,-4,-4,1105,1,656,21202,-4,1,1,21201,-3,-1,2,21202,-2,2,3,21102,1,607,0,1106,0,565,22102,1,1,-4,21101,0,1,-1,2207,-4,-2,10,1006,10,626,21101,0,0,-1,22202,-2,-1,-2,2107,0,-3,10,1006,10,648,21202,-1,1,1,21101,0,648,0,105,1,523,21202,-2,-1,-2,22201,-4,-2,-4,109,-5,2105,1,0]

class Robot: IO {
    static let size = 1000
    static let colorMask = 1 << 0
    static let paintedMask = 1 << 1

    var position = (x:Robot.size / 2, y: Robot.size / 2)
    var direction = 0

    var colors = Array(repeating: 0, count: Robot.size * Robot.size)
    var paintedCells = 0
    var nextIsPaint = true

    var minX = Robot.size
    var maxX = 0
    var minY = Robot.size
    var maxY = 0

    func move(turn: Int) {
        direction = (4 + direction + (turn == 0 ? -1 : 1)) % 4
        switch direction {
        case 0: position.y -= 1
        case 1: position.x += 1
        case 2: position.y += 1
        case 3: position.x -= 1
        default: fatalError()
        }
    }

    func input() -> Int {
        return colors[position] & Robot.colorMask
    }

    func output(_ value: Int) {
        precondition(value == 0 || value == 1)
        if nextIsPaint {
            if colors[position] & Robot.paintedMask == 0 {
                paintedCells += 1
            }
            colors[position] = value | Robot.paintedMask

            minX = min(position.x, minX)
            maxX = max(position.x, maxX)
            minY = min(position.y, minY)
            maxY = max(position.y, maxY)
        } else {
            move(turn: value)
        }
        nextIsPaint.toggle()
    }
}

extension Array {
    subscript(pos: (Int, Int)) -> Element {
        get {
            let (x, y) = pos
            precondition(0..<Robot.size ~= x)
            precondition(0..<Robot.size ~= y)
            return self[x + y * Robot.size]
        }
        set(newValue) {
            let (x, y) = pos
            precondition(0..<Robot.size ~= x)
            precondition(0..<Robot.size ~= y)
            self[x + y * Robot.size] = newValue
        }
    }
}

let robot = Robot()
robot.colors[robot.position] = 1

IntCode(program: program, io: robot).run()
print(robot.paintedCells)

for y in robot.minY ... robot.maxY {
    let lineStart = y * Robot.size
    print(String(robot.colors[lineStart + robot.minX ... lineStart + robot.maxX].map { $0 & Robot.colorMask == 0 ? " " : "*" }))
}