數組與鏈表區別:
二叉排序樹的創建和遍歷 代碼實現
package tree.binarysorttree;
public class BinarySortTreeDemo {
public static void main(String []args){
int [] arr = {7,3,10,12,5,1,9};
BinarySortTree binarySortTree = new BinarySortTree();
//循環添加節點
for (int i = 0; i < arr.length; i++) {
binarySortTree.add(new Node(arr[i]));
}
//中序遍歷
binarySortTree.infixOrder();
}
}
//創建二叉排序樹
class BinarySortTree{
private Node root;
//添加節點的方法
public void add(Node node){
if (root == null){
root = node;
}else {
root.add(node);
}
}
//中序遍歷
public void infixOrder(){
if (root != null){
root.infixOrder();
}else {
System.out.println("二叉排序樹爲空,無法遍歷");
}
}
}
//創建節點
class Node{
int value;
Node left;
Node right;
public Node(int value) {
this.value = value;
}
//添加節點的方法
//遞歸的形式添加節點,滿足二叉排序樹
public void add(Node node){
if(node == null){
return;
}
if (node.value < this.value){
//當前節點的左子節點爲空直接掛上
if (this.left == null){
this.left = node;
}else {//不爲空遞歸向下
this.left.add(node);
}
}else { //添加的節點的值大於等於當前節點的值,右節點爲空,掛到右節點
if (this.right == null){
this.right = node;
}else{//不爲空就掛上
this.right.add(node);
}
}
}
//中序遍歷
public void infixOrder(){
if (this.left != null){
this.left.infixOrder();
}
System.out.println(this);
if (this.right != null){
this.right.infixOrder();
}
}
@Override
public String toString() {
return "Node{" +
"value=" + value +
'}';
}
}
刪除:
編寫查找節點的方法
//查找要刪除的節點
public Node search(int value){
if (root == null){
return null;
}else {
return root.search(value);
}
}
//查找父節點
public Node searchParent(int value){
if (root == null){
return null;
}else{
return root.searchParent(value);
}
}
刪除第三種需要用到的,查找以target.right爲根節點節點的樹的最小值並刪除,然後返回value
//編寫一個方法,查找以傳入節點爲根節點的二叉排序樹中的最小節點的值,並刪除此節點
public int delRightTreeMin(Node node){
Node target = node;
//循環查找左節點,就會找到最小值
while(target.left != null){
target = target.left;
}
//刪除最小節點
delNode(target.value);
return target.value;
}
刪除節點方法
public void delNode(int value){
if (root == null){
return;
}else {
//1.找到要刪除的節點
Node targetNode = search(value);
if (targetNode == null){
return;
}
//如果二叉排序樹只有一個節點,並且刪除的此節點
if (root.left == null && root.right == null){
root = null;
return;
}
//找到父節點
Node parent = searchParent(value);
//如果刪除的節點是葉子節點
if (targetNode.left == null && targetNode.right == null){//第一種,刪除葉子
if (targetNode.value < parent.value){ //比較大小判斷刪除的左葉子還是右葉子
parent.left = null;
}else {
parent.right = null;
}
}else if (targetNode.left != null && targetNode.right != null){//刪除有兩棵子樹的(因爲第二種判斷比較麻煩,所以直接寫在else裏)
//從target的右子樹找到最小節點,
//用一個臨時變量存儲這個最小節點的value
//刪除最小節點
int min = delRightTreeMin(targetNode.right);
//把最小節點的值賦給targetNode
targetNode.value = min;
}else {//第二種,只有一棵子樹
//如果要刪除的節點有左子節點
if (targetNode.left != null){
//如果只有根節點和一個葉子節點,要刪除根節點,那就要判斷根節點的父節點是否爲空
if (parent != null) {
//如果targetNode是parent的左子節點
if (parent.left.value == value) {
parent.left = targetNode.left;
} else {//targetNode是parent的右子節點
parent.right = targetNode.left;
}
}else {
root = targetNode.left;
}
}else {//要刪除的節點有右子節點
if (parent != null) {
if (parent.left.value == value) {
//如果targetNode是parent的左子節點
parent.left = targetNode.right;
} else {//targetNode是parent的右子節點
parent.right = targetNode.right;
}
}else {
root = targetNode.right;
}
}
}
}
}
完整代碼
package tree.binarysorttree;
public class BinarySortTreeDemo {
public static void main(String []args){
int [] arr = {7,3,10,12,5,1,9,2};
BinarySortTree binarySortTree = new BinarySortTree();
//循環添加節點
for (int i = 0; i < arr.length; i++) {
binarySortTree.add(new Node(arr[i]));
}
//中序遍歷
binarySortTree.infixOrder();
//測試刪除
binarySortTree.delNode(2);
binarySortTree.delNode(5);
binarySortTree.delNode(9);
binarySortTree.delNode(12);
binarySortTree.delNode(7);
binarySortTree.delNode(3);
binarySortTree.delNode(10);
System.out.println();
binarySortTree.infixOrder();
}
}
//創建二叉排序樹
class BinarySortTree{
private Node root;
public Node getRoot() {
return root;
}
//查找要刪除的節點
public Node search(int value){
if (root == null){
return null;
}else {
return root.search(value);
}
}
//查找父節點
public Node searchParent(int value){
if (root == null){
return null;
}else{
return root.searchParent(value);
}
}
//編寫一個方法,查找以傳入節點爲根節點的二叉排序樹中的最小節點的值,並刪除此節點
public int delRightTreeMin(Node node){
Node target = node;
//循環查找左節點,就會找到最小值
while(target.left != null){
target = target.left;
}
//刪除最小節點
delNode(target.value);
return target.value;
}
//刪除節點
public void delNode(int value){
if (root == null){
return;
}else {
//1.找到要刪除的節點
Node targetNode = search(value);
if (targetNode == null){
return;
}
//如果二叉排序樹只有一個節點,並且刪除的此節點
if (root.left == null && root.right == null){
root = null;
return;
}
//找到父節點
Node parent = searchParent(value);
//如果刪除的節點是葉子節點
if (targetNode.left == null && targetNode.right == null){//第一種,刪除葉子
if (targetNode.value < parent.value){ //比較大小判斷刪除的左葉子還是右葉子
parent.left = null;
}else {
parent.right = null;
}
}else if (targetNode.left != null && targetNode.right != null){//刪除有兩棵子樹的(因爲第二種判斷比較麻煩,所以直接寫在else裏)
//從target的右子樹找到最小節點,
//用一個臨時變量存儲這個最小節點的value
//刪除最小節點
int min = delRightTreeMin(targetNode.right);
//把最小節點的值賦給targetNode
targetNode.value = min;
}else {//第二種,只有一棵子樹
//如果要刪除的節點有左子節點
if (targetNode.left != null){
//如果只有根節點和一個葉子節點,要刪除根節點,那就要判斷根節點的父節點是否爲空
if (parent != null) {
//如果targetNode是parent的左子節點
if (parent.left.value == value) {
parent.left = targetNode.left;
} else {//targetNode是parent的右子節點
parent.right = targetNode.left;
}
}else {
root = targetNode.left;
}
}else {//要刪除的節點有右子節點
if (parent != null) {
if (parent.left.value == value) {
//如果targetNode是parent的左子節點
parent.left = targetNode.right;
} else {//targetNode是parent的右子節點
parent.right = targetNode.right;
}
}else {
root = targetNode.right;
}
}
}
}
}
//添加節點的方法
public void add(Node node){
if (root == null){
root = node;
}else {
root.add(node);
}
}
//中序遍歷
public void infixOrder(){
if (root != null){
root.infixOrder();
}else {
System.out.println("二叉排序樹爲空,無法遍歷");
}
}
}
//創建節點
class Node{
int value;
Node left;
Node right;
public Node(int value) {
this.value = value;
}
//查找要刪除的節點
public Node search(int value){
if (value == this.value){//找到
return this;
}else if(value < this.value){//應該向左繼續查找
//如果左子節點爲空就不能找了,說明不存在
if (this.left == null) {
return null;
}
return this.left.search(value);
}else {
if (this.right == null){
return null;
}
return this.right.search(value);
}
}
//查找要刪除節點的父節點
public Node searchParent(int value){
if ((this.left != null && this.left.value == value)||
(this.right != null && this.right.value == value)){
return this;
}else {
//如果查找的值小於當前節點的值,並且當前節點的左子節點不爲空
if (value < this.value && this.left != null){
return this.left.searchParent(value);//像左子樹遞歸查找
}else if (value >= this.value && this.right != null){
return this.right.searchParent(value);//像右子樹遞歸查找
}else {
return null; //沒有父節點
}
}
}
//添加節點的方法
//遞歸的形式添加節點,滿足二叉排序樹
public void add(Node node){
if(node == null){
return;
}
if (node.value < this.value){
//當前節點的左子節點爲空直接掛上
if (this.left == null){
this.left = node;
}else {//不爲空遞歸向下
this.left.add(node);
}
}else { //添加的節點的值大於等於當前節點的值,右節點爲空,掛到右節點
if (this.right == null){
this.right = node;
}else{//不爲空就掛上
this.right.add(node);
}
}
}
//中序遍歷
public void infixOrder(){
if (this.left != null){
this.left.infixOrder();
}
System.out.println(this);
if (this.right != null){
this.right.infixOrder();
}
}
@Override
public String toString() {
return "Node{" +
"value=" + value +
'}';
}
}