遞歸判斷兩個二叉樹是否相同以及對葉子節點計數

// 判斷兩棵二叉樹是否完全相同
bool isSameTree(TreeNode* p, TreeNode* q) {
	if (p == NULL && q == NULL)
		return true;
	if (p == NULL || q == NULL)
		return false;
	if (p->val != q->val)
		return false;
	return isSameTree(p->left, q->left) && isSameTree(p->right, q->right);
}

// 如果要判斷二叉樹是否對稱 把該函數嵌進去就行
class Solution {
public:
	bool isSymmetrical(TreeNode* l1, TreeNode* l2)
	{
		if (l1 == NULL && l2 == NULL)
			return true;
		if (l1 == NULL || l2 == NULL)
			return false;
		if (l1->val != l2->val)
			return false;
		return isSymmetrical(l1->left, l2->right) && isSymmetrical(l1->right, l2->left);

	}
	bool isSymmetrical(TreeNode* pRoot)
	{
		return isSymmetrical(pRoot, pRoot);
	}
};

 

// 計算二叉樹葉子節點個數（遞歸方法）
int Num_of_Leaves(TreeNode* p) {
	if (p == NULL)
		return 0;
	if (p->left == NULL && p->right == NULL)
		return 1;

	return (Num_of_Leaves(p->left) + Num_of_Leaves(p->right));
}

都是很容易讀懂，很容易寫出來的遞歸，下面是全部代碼，用於測試

#include <iostream>
using namespace std;
#define D(x) cout<<#x<<"="<<x<<endl;

typedef struct TreeNode {
	char val;
	TreeNode *left;
	TreeNode *right;
} BiNode, *BiTree;

// 先序建立二叉樹 （輸入時，按先序次序輸入二叉樹中結點的值,以 # 字符表示空樹）
BiTree createBiTree()
{
	BiTree T;
	char c;
	cin >> c;
	if (c == '#')
		T = NULL;
	else {
		T = new BiNode;  // 或 T = (BiTree)malloc(sizeof(BiNode));
		T->val = c;
		T->left = createBiTree();
		T->right = createBiTree();
	}
	return T;
}


// 前序遍歷
void PreOrderTraverse(BiTree T) {
	if (T) {
		cout << T->val << endl;
		PreOrderTraverse(T->left);
		PreOrderTraverse(T->right);
	}
}

// 二叉樹葉子節點個數（遞歸方法）
int Num_of_Leaves(TreeNode* p) {
	if (p == NULL)
		return 0;
	if (p->left == NULL && p->right == NULL)
		return 1;

	return (Num_of_Leaves(p->left) + Num_of_Leaves(p->right));
}

// 判斷兩棵二叉樹是否完全相同
bool isSameTree(TreeNode* p, TreeNode* q) {
	if (p == NULL && q == NULL)
		return true;
	if (p == NULL || q == NULL)
		return false;
	if (p->val != q->val)
		return false;
	return isSameTree(p->left, q->left) && isSameTree(p->right, q->right);
}

int main() {

	BiTree T1 = createBiTree(); // 建立
	//123###7#9##
	//12##5##
	//12###
	PreOrderTraverse(T1); // 輸出
	BiTree T2 = createBiTree(); // 建立
	PreOrderTraverse(T2); // 輸出

	int num1 = Num_of_Leaves(T1);
	int num2 = Num_of_Leaves(T2);
	bool flag = isSameTree(T1, T2);
	D(flag)
	D(num1)D(num2)

	return 0;
}

對於樹的輸入，舉個栗子

在 http://mshang.ca/syntree/ 中輸入：[1 [2 [3 [#] [#]] [#] ] [7 [#] [9 [#] [#] ]]]

也就是在代碼編譯完成後輸入 123###7#9## 就會得到上面的樹