TreeNodeUpdated.swift

class TreeNode<T>: Comparable, CustomStringConvertible where T: Comparable, T: CustomStringConvertible {
    
    static func <(lhs: TreeNode<T>, rhs: TreeNode<T>) -> Bool {
        return lhs.data < rhs.data
    }
    
    static func ==(lhs: TreeNode<T>, rhs: TreeNode<T>) -> Bool {
        return lhs.data == rhs.data
    }
    
    var description: String {
        return self.data.description
    }
    
    var data: T
    var leftNode: TreeNode?
    var rightNode: TreeNode?
    
    init(data: T,
         leftNode: TreeNode? = nil,
         rightNode: TreeNode? = nil) {
        self.data = data
        self.leftNode = leftNode
        self.rightNode = rightNode
    }
}