🥳每日一练-排序二叉树中求解节点层次-JS简易版

这篇文章分享一个简单的内容,在排序二叉树中求一个节点所在的层次。

就像这张图,6 节点的层次是 1,3 节点的层次是 4,4 节点的层次是 5

问题很清晰,也很简单,直接上代码吧

准备数据

先准备一个排序二叉树的数据:

javascript 复制代码
const treeData = [6, 5, 2, 3, 7, 12, 9, 4, 8];

这个数据是按照上面图片的数据来的,下面就会用代码将这个数组,转成一个排序二叉树:

javascript 复制代码
class Tree {
	constructor() {
		this.root = null;
	}
	insert(value) {
		this.root = this._insert(this.root, value);
	}
	_insert(node, value) {
		if (!node) return { value, left: null, right: null };
		if (value < node.value) node.left = this._insert(node.left, value);
		else node.right = this._insert(node.right, value);
		return node;
	}
}

构建二叉树:

javascript 复制代码
const tree = new Tree();
treeData.forEach((item) => {
	tree.insert(item);
});
console.log(tree);

这是二叉树的 json 结构:

javascript 复制代码
{
  value: 6,
  left: {
    value: 5,
    left: {
      value: 2,
      left: null,
      right: {
        value: 3,
        left: null,
        right: {
          value: 4,
          left: null,
          right: null,
        },
      },
    },
    right: null,
  },
  right: {
    value: 7,
    left: null,
    right: {
      value: 12,
      left: {
        value: 9,
        left: {
          value: 8,
          left: null,
          right: null,
        },
        right: null,
      },
      right: null,
    },
  },
};

结构和图片的一摸一样,可以互相对照着看。

找到节点的层次

javascript 复制代码
class Tree{
  // ...
  	getLayer(value) {
		return this._getLayer2(this.root, value, 0);
	}

	_getLayer(node, value, layer) {
		if (!node) return -1;
		if (node.value == value) return layer + 1;
		if (value < node.value) return this._getLayer(node.left, value, layer + 1);
		return this._getLayer(node.right, value, layer + 1);
	}
  // ...
}

在 Tree 对象中添加两个方法,一个是供外界调用的getLayer,一个是供内部调用的_getLayer
_getLayer接收三个参数:

  1. node,当前比较的节点
  2. value,要寻找节点的值
  3. layer 上一个节点的层次

代码的逻辑:

  • 如果当前 node 的 value 和寻找的 value 相等,说明已经找到了。就反回当前的层次
  • 如果不相等,就继续往下层寻找,根据 node 和要查找的值的大小,决定是要往左子树寻找,还是右子树
  • 如果当前 node 为空,说明已经找到了叶子还没有找到,那么二叉排序树中没有寻找的节点,就反悔-1

代码很简单,我们来测试一下:

javascript 复制代码
console.log(tree.getLayer(6)); // 1
console.log(tree.getLayer(3)); // 4
console.log(tree.getLayer(4)); // 5

以上是递归的写法,还有非递归的写法:

javascript 复制代码
class Tree{
  // ...
  _getLayer2(node, value) {
		let layer = 1;
		if (node.value == value) return layer;
		while (node) {
			if (value < node.value) {
				node = node.left;
				layer++;
			} else if (value > node.value) {
				node = node.right;
				layer++;
			} else {
				return layer;
			}
		}
		return -1;
	}
}

非递归代码的逻辑也是类似的,就不解释了

完整代码

js 复制代码
class Tree {
	constructor() {
		this.root = null;
	}
	insert(value) {
		this.root = this._insert(this.root, value);
	}
	_insert(node, value) {
		if (!node) return { value, left: null, right: null };
		if (value < node.value) node.left = this._insert(node.left, value);
		else node.right = this._insert(node.right, value);
		return node;
	}

	getLayer(value) {
		return this._getLayer2(this.root, value, 0);
	}

	_getLayer(node, value, layer) {
		if (!node) return -1;
		if (node.value == value) return layer + 1;
		if (value < node.value) return this._getLayer(node.left, value, layer + 1);
		return this._getLayer(node.right, value, layer + 1);
	}

	_getLayer2(node, value) {
		let layer = 1;
		if (node.value == value) return layer;
		while (node) {
			if (value < node.value) {
				node = node.left;
				layer++;
			} else if (value > node.value) {
				node = node.right;
				layer++;
			} else {
				return layer;
			}
		}
		return -1;
	}
}

const treeData = [6, 5, 2, 3, 7, 12, 9, 4, 8];
const tree = new Tree();
treeData.forEach((item) => {
	tree.insert(item);
});

console.log(tree.getLayer(4)); //5

总结

这篇文章很轻松,分享了一道简单的题目: 设计一个算法,求解二叉树的层次。文中也给出了相应的 JS 代码实现,读者可以自行 copy 代码到本地跑一跑。下篇文章再见👋

你觉得这篇文章怎么样?喜欢就点赞+关注吧

相关推荐
小小小新人1212310 分钟前
C语言 ATM (4)
c语言·开发语言·算法
你的冰西瓜38 分钟前
C++排序算法全解析(加强版)
c++·算法·排序算法
我命由我123451 小时前
前端开发问题:SyntaxError: “undefined“ is not valid JSON
开发语言·前端·javascript·vue.js·json·ecmascript·js
Jokerator1 小时前
深入解析JavaScript获取元素宽度的多种方式
javascript·css
এ᭄画画的北北1 小时前
力扣-31.下一个排列
算法·leetcode
海天胜景1 小时前
vue3 当前页面方法暴露
前端·javascript·vue.js
GISer_Jing1 小时前
前端面试常考题目详解
前端·javascript
绝无仅有2 小时前
企微审批对接错误与解决方案
后端·算法·架构
中微子2 小时前
JavaScript 防抖与节流:从原理到实践的完整指南
前端·javascript