我們在查詢二叉樹上的節點時,會通過遞迴去進行遍歷,如果我們要把這個節點作為函式返回值進行返回的話,遞迴遍歷法就無法滿足

我們可以通過層序遍歷去查詢結點然後進行返回,只需一個輔助佇列就行

中序遍歷的思想很簡單

(1)先將父親結點入隊

(2)如果父親結點的左子不為空,左子入隊

(3)如果父親結點的右子不為空,右子入隊

(4)父親結點出隊

(5)重複上述操作直到佇列為空

我們通過層序遍歷去查詢指定結點,找到後進行返回即可,程式碼如下

Node* layerfindNode(char node) {
		if (root != nullptr) {
			nodeQueue.push(root);
			while (!nodeQueue.empty()) {
				if (nodeQueue.front()->data == node) {
					return nodeQueue.front();
					break;
				}
				else {
					if (nodeQueue.front()->lchild != nullptr) {
						nodeQueue.push(nodeQueue.front()->lchild);
					}
					if (nodeQueue.front()->rchild != nullptr) {
						nodeQueue.push(nodeQueue.front()->rchild);
					}
					nodeQueue.pop();
				}
			}
		}
		return nullptr;
	}