armstrong.swift

infix operator ** {}

func ** (n: Int, p: Int) -> Int {
    return Int(pow(Double(n), Double(p)))
}


func splitInt(n: Int) -> [Int] {
    if n < 10 {
        return [n]
    }
    return splitInt(n / 10) + [n % 10]
}


func isArmstrongNumber(n: Int) -> Bool {
    let nums = splitInt(n)
    let size = nums.count
    var sum = 0
    
    for i in nums {
        sum += (i ** size)
    }
    
    return sum == n
}

print(isArmstrongNumber(100))
print(isArmstrongNumber(153))