求二叉树某个节点的深度

描述

1. 给定一个二叉树的根节点，和其中某个节点的值。求该节点的深度
2. 函数定义：int getNodeDeepth(TreeNode* root, int val);
3. TreeNode结构体：

   struct TreeNode {
   	int val;
   	TreeNode *left;
   	TreeNode *right;
   	TreeNode(int v=0,TreeNode *l=nullptr,TreeNode *r=nullptr):
               val(v),left(l),right(r){}
   };
   

例如：

输入：root = [3,5,1,6,2,0,8,null,null,7,4], val = 4
输出：4

分析

求该节点的深度，也就是求该节点所在的层数。可以用层序遍历求解

代码

int getNodeDeepth(TreeNode* root, int val) {
	queue<TreeNode*> que;
	que.push(root);
	int level = 0;						//记录当前是第几层
	while (!que.empty()) {
		int levelSize = que.size();		        //当前层的元素个数
		level++;
		for (int i = 0; i < levelSize; i++) {
			TreeNode* nod = que.front(); que.pop();
			if (nod->val == val) {
				return level;
			}
			if (nod->left) que.push(nod->left);
			if (nod->right) que.push(nod->right);
		}
	}
	return -1;					        //没有找到返回-1
}