数据结构 - 树

树

数据结构 - 树。

// 树实现
// ES5
var Tree = function() {
var root = null;

// 辅助类
var Node = function(value) {
this.value = value;
this.left = null;
this.right = null;
};

// 插入节点
var insertNode = function(node, newNode) {
// 节点比较
if (newNode.value > node.value) {
// 往右走
if (node.right == null) {
node.right = newNode;
} else {
insertNode(node.right, newNode);
}
} else if (newNode.value < node.value) {
// 往左走
if (node.left == null) {
node.left = newNode;
} else {
insertNode(node.left, newNode);
}
}
};
this.insert = function(value) {
// 新的节点
var newNode = new Node(value);

// 检查是否为空树
if (root == null) {
root = newNode;
} else {
insertNode(root, newNode);
}
};

// 遍历节点
var traverse = function(node, callback) {
// 检查节点是否为空
if (node != null) {
// callback(node.value); // 8,2,3,9
traverse(node.left, callback);
// callback(node.value); // 2,3,8,9
traverse(node.right, callback);
// callback(node.value);  // 3,2,9,8
}
};
this.traverse = function(callback) {
traverse(root, callback);
};

// 移除节点
var findMinNode = function(node) {
if (node == null) return null;
while(node && node.left) {
return node.left;
}
return node;
};
var removeNode = function(node, value) {
if (node == null) return null;
if (value > node.value) {
// 向右查找
node.right = removeNode(node.right, value);
return node;
} else if (value < node.value) {
// 向左查找
node.left = removeNode(node.left, value);
return node;
} else {
// 无子节点
if (node.left == null && node.right == null) {
node = null;
return node;
}

// 只有一个子节点
if (node.left == null && node.right) {
return node.right;
}
if (node.left && node.right == null) {
return node.left;
}

// 有两个子节点
var aux = findMinNode(node);  // 获取最小节点
node.value = aux.value;
node.right = removeNode(node.right, aux.value);
return node;
}
};
this.remove = function(value) {
root = removeNode(root, value);
};

// 获取根节点
this.getRoot = function() {
return root;
};
};

var t = new Tree();
t.insert(11);
t.insert(8);
t.insert(4);
t.insert(9);
t.insert(3);
t.insert(5);

t.remove(4);

var result = function(value) {
console.log(`value - ${value}`);
};
t.traverse(result);