Xuất Node con và lá trong cây nhị phân tìm kiếm

// xuất các node có 2 con
void Node_Co_2_Con(TREE t)
{
	if (t != NULL)
	{
		// xử lí
		if (t->pLeft != NULL && t->pRight != NULL) // con trái và con phải có tồn tại phần tử
		{
			cout << t->data << "  ";
		}
		Node_Co_2_Con(t->pLeft); // duyệt sang cây con trái của node hiện tại
		Node_Co_2_Con(t->pRight); // duyệt sang cây con phải của node hiện tại
	}
}

// xuất các node có 1 con
void Node_Co_1_Con(TREE t)
{
	if (t != NULL)
	{
		// xử lí
		if ((t->pLeft != NULL && t->pRight == NULL) || (t->pLeft == NULL && t->pRight != NULL))
		{
			cout << t->data << "  ";
		}
		Node_Co_1_Con(t->pLeft); // duyệt sang cây con trái của node hiện tại
		Node_Co_1_Con(t->pRight); // duyệt sang cây con phải của node hiện tại
	}
}

// xuất các node lá
void Node_La(TREE t)
{
	if (t != NULL)
	{
		// xử lí
		if (t->pLeft == NULL && t->pRight == NULL)
		{
			cout << t->data << "  ";
		}
		Node_La(t->pLeft); // duyệt sang cây con trái của node hiện tại
		Node_La(t->pRight); // duyệt sang cây con phải của node hiện tại
	}
}

#include<iostream>
using namespace std;
// khai báo cấu trúc 1 cái node trong cây nhị phân tìm kiếm
struct node
{
	int data; // dữ liệu chứa trong 1 cái node
	struct node *pLeft; // con trỏ liên kết với cái node bên trái <=> cây con trái
	struct node *pRight; // con trỏ liên kết với cái node bên phải <=> cây con phải
};
typedef struct node NODE;
typedef NODE* TREE;

// hàm khởi tạo cây nhị phân tìm kiếm
void KhoiTaoCay(TREE &t)
{
	t = NULL;
}

// hàm thêm 1 cái phần tử vào cây
void ThemNodeVaoCay(TREE &t, int x)
{
	// nếu cây rỗng
	if (t == NULL)
	{
		NODE *p = new NODE;
		p->data = x;
		p->pLeft = NULL;
		p->pRight = NULL;
		t = p;
	}
	else
	{
		// nếu phần tử thêm vào mà nhỏ hơn nút gốc thì sẽ duyệt qua bên trái
		if (x < t->data)
		{
			ThemNodeVaoCay(t->pLeft, x);
		}
		else if (x > t->data)
		{
			ThemNodeVaoCay(t->pRight, x);
		}
	}
}

// hàm xuất phần tử trong cây nhị phân tìm kiếm
void NLR(TREE t)
{
	if (t != NULL)
	{
		// xử lí
		cout << t->data << "  ";
		NLR(t->pLeft);
		NLR(t->pRight);
	}
}

// xuất các node có 2 con
void Node_Co_2_Con(TREE t)
{
	if (t != NULL)
	{
		// xử lí
		if (t->pLeft != NULL && t->pRight != NULL) // con trái và con phải có tồn tại phần tử
		{
			cout << t->data << "  ";
		}
		Node_Co_2_Con(t->pLeft); // duyệt sang cây con trái của node hiện tại
		Node_Co_2_Con(t->pRight); // duyệt sang cây con phải của node hiện tại
	}
}
// xuất các node có 1 con
void Node_Co_1_Con(TREE t)
{
	if (t != NULL)
	{
		// xử lí
		if ((t->pLeft != NULL && t->pRight == NULL) || (t->pLeft == NULL && t->pRight != NULL))
		{
			cout << t->data << "  ";
		}
		Node_Co_1_Con(t->pLeft); // duyệt sang cây con trái của node hiện tại
		Node_Co_1_Con(t->pRight); // duyệt sang cây con phải của node hiện tại
	}
}

// xuất các node lá
void Node_La(TREE t)
{
	if (t != NULL)
	{
		// xử lí
		if (t->pLeft == NULL && t->pRight == NULL)
		{
			cout << t->data << "  ";
		}
		Node_La(t->pLeft); // duyệt sang cây con trái của node hiện tại
		Node_La(t->pRight); // duyệt sang cây con phải của node hiện tại
	}
}

// hàm nhập cây nhị phân tìm kiếm
void Menu(TREE &t)
{
	int luachon;
	while(true)
	{
		cout << "\n\n\t\t ============ MENU ============";
		cout << "\n1. Nhập phần tử cho cây";
		cout << "\n2. Duyệt cây";
		cout << "\n3. Node có hai con";
		cout << "\n4. Node có một con";
		cout << "\n5. Node lá";
		cout << "\n0. Thoát";
		cout << "\n\n\t\t =============  END  =============";

		cout << "\nNhập lựa chọn của bạn: ";
		cin >> luachon;

		if (luachon == 1)
		{
			int x;
			cout << "\nNhập giá trị: ";
			cin >> x;
			ThemNodeVaoCay(t, x);
		}
		else if (luachon == 2)
		{
			cout << "\nCây nhị phân tìm kiếm\n";
			NLR(t);
		}
		
		else if (luachon == 3)
		{
			cout << "\nNode có hai con: ";
			Node_Co_2_Con(t);
		}
		else if (luachon == 4)
		{
			cout << "\nNode có một con: ";
			Node_Co_1_Con(t);
		}
		else if (luachon == 5)
		{
			cout << "\nNode lá: ";
			Node_La(t);
		}
		else
		{
			break;
		}
	}
}


int main()
{
	TREE t;
	KhoiTaoCay(t);
	Menu(t);

	cout<<"\n---------------------------------\n";
  cout<<"Chương trình này được đăng tại Freetuts.net";
}