编写最简单的二叉树

编写最简单的二叉树

二叉树结构

源码

-swift-

//
//  Node.swift
//  swift-TreeStructure
//
//  Created by YouXianMing on 15/10/19.
//  Copyright  2015年 ZiPeiYi. All rights reserved.
//

import UIKit

class Node: NSObject {

    /// 节点名字
    var nodeName  : String?

    /// 左节点
    var leftNode  : Node?

    /// 右节点
    var rightNode : Node?

    /**
    便利构造器方法

    - parameter nodeWithName: 节点名字

    - returns: 节点
    */
    init(withName : String?) {

        super.init()
        nodeName = withName
    }
}

//
//  ViewController.swift
//  swift-TreeStructure
//
//  Created by YouXianMing on 15/10/19.
//  Copyright  2015年 ZiPeiYi. All rights reserved.
//

import UIKit

class ViewController: UIViewController {

    let rootNode : Node = Node(withName : "A")

    override func viewDidLoad() {

        super.viewDidLoad()

        // 插入节点
        insertNode(rootNode, node: Node(withName : "B"))
        insertNode(rootNode, node: Node(withName : "C"))
        insertNode(rootNode, node: Node(withName : "D"))
        insertNode(rootNode, node: Node(withName : "E"))
        insertNode(rootNode, node: Node(withName : "F"))

        // 便利节点
        treeInfomationWith(rootNode)
    }

    /**
    插入节点

    - parameter tree: 根节点
    - parameter node: 被插入节点
    */
    func insertNode(tree : Node, node : Node) {

        if tree.leftNode == nil {

            tree.leftNode = node
            return
        }

        if tree.rightNode == nil {

            tree.rightNode = node
            return
        }

        insertNode(tree.leftNode!, node: node)
    }

    /**
    遍历节点

    - parameter node: 节点
    */
    func treeInfomationWith(node : Node) {

        if node.leftNode != nil {

            treeInfomationWith(node.leftNode!)
        }

        print(node.nodeName)

        if node.rightNode != nil {

            treeInfomationWith(node.rightNode!)
        }
    }
}

-objective-c-

//
//  Node.h
//  TreeStructure
//
//  Created by YouXianMing on 15/10/19.
//  Copyright  2015年 ZiPeiYi. All rights reserved.
//

#import <Foundation/Foundation.h>

@interface Node : NSObject

/**
 *  节点名字
 */
@property (nonatomic, strong) NSString *nodeName;

/**
 *  左节点
 */
@property (nonatomic, strong) Node  *leftNode;

/**
 *  右节点
 */
@property (nonatomic, strong) Node  *rightNode;

/**
 *  便利构造器方法
 *
 *  @param nodeName 节点名字
 *
 *  @return 节点
 */
+ (instancetype)nodeWithName:(NSString *)nodeName;

@end

//
//  Node.m
//  TreeStructure
//
//  Created by YouXianMing on 15/10/19.
//  Copyright  2015年 ZiPeiYi. All rights reserved.
//

#import "Node.h"

@implementation Node

+ (instancetype)nodeWithName:(NSString *)nodeName {

    Node *node     = [[[self class] alloc] init];
    node.nodeName  = nodeName;

    return node;
}

@end

//
//  ViewController.m
//  TreeStructure
//
//  Created by YouXianMing on 15/10/19.
//  Copyright  2015年 ZiPeiYi. All rights reserved.
//

#import "ViewController.h"
#import "Node.h"

@interface ViewController ()

@property (nonatomic, strong) Node *rootNode;

@end

@implementation ViewController

- (void)viewDidLoad {

    [super viewDidLoad];

    // 根节点
    self.rootNode = [Node nodeWithName:@"A"];

    // 插入节点
    [self insertNodeTree:self.rootNode node:[Node nodeWithName:@"B"]];
    [self insertNodeTree:self.rootNode node:[Node nodeWithName:@"C"]];
    [self insertNodeTree:self.rootNode node:[Node nodeWithName:@"D"]];
    [self insertNodeTree:self.rootNode node:[Node nodeWithName:@"E"]];
    [self insertNodeTree:self.rootNode node:[Node nodeWithName:@"F"]];

    // 遍历节点
    [self treeInfomationWithNode:self.rootNode];
}

/**
 *  插入节点
 *
 *  @param tree 根节点
 *  @param node 被插入节点
 */
- (void)insertNodeTree:(Node *)tree node:(Node *)node {

    if (tree.leftNode == nil) {

        tree.leftNode = node;
        return;
    }

    if (tree.rightNode == nil) {

        tree.rightNode = node;
        return;
    }

    [self insertNodeTree:tree.leftNode node:node];
}

/**
 *  遍历节点
 *
 *  @param node 节点
 */
- (void)treeInfomationWithNode:(Node *)node {

    if (node.leftNode) {

        [self treeInfomationWithNode:node.leftNode];
    }

    NSLog(@"%@", node.nodeName);

    if (node.rightNode) {

        [self treeInfomationWithNode:node.rightNode];
    }
}

@end

打印结果

2015-10-19 20:05:24.493 TreeStructure[33002:267671] F

2015-10-19 20:05:24.494 TreeStructure[33002:267671] D

2015-10-19 20:05:24.494 TreeStructure[33002:267671] B

2015-10-19 20:05:24.494 TreeStructure[33002:267671] E

2015-10-19 20:05:24.494 TreeStructure[33002:267671] A

2015-10-19 20:05:24.494 TreeStructure[33002:267671] C