Search the Nth element of Level M in a binary tree

// binTreeMN.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <iostream>
#include <cstdio>
using namespace std;

struct Node{
	int val;
	Node *lChild;
	Node *rChild;

	Node(int v)
	{
		val = v;
		lChild = NULL;
		rChild = NULL;
	}
};

void init_Tree(Node *&root, int *arr, int begin, int end)
{
	if(begin > end)
		return;
	int mid = begin + (end-begin)/2;

	if(root == NULL)
	{
		root = new Node(arr[mid]);
	}

	cout<<arr[mid]<<" ";
	init_Tree(root->lChild, arr, begin, mid-1);
	init_Tree(root->rChild, arr, mid+1, end);
}

void travel(Node *root)  
{  
    if(root != NULL)  
    {                          
		travel(root->lChild);  
        cout<<root->val<<" ";
        travel(root->rChild);  
    }  
}

bool print_MN(Node *root, int m, int n, int &count)
{
	if(!root || m<0 || n<0)
		return false;

	if(m == 0)
	{
		if(count == n)
		{
			printf("val = %d", root->val);
			return true;
		}

		count++;
		return 0;
	}

	return print_MN(root->lChild, m-1, n, count) || print_MN(root->rChild, m-1, n, count);
}

/*
	    5
	   / \
	  2   8
	 / \ / \
	1  3 6  9
	    \ \  \
	     4 6  0
*/
int main(int argc, char* argv[])
{
	int test[] = {1,2,3,4,5,6,7,8,9,0};
	int m = 2;
	int n = 3;
	int count = 0;
	Node *root = NULL;
	init_Tree(root, test, 0, 9);
	cout<<endl<<"------In-Order travel------"<<endl;
	travel(root);
	cout<<endl<<"------("<<m<<","<<n<<")------"<<endl;
	print_MN(root, m, n, count);

	//printf("Hello World!\n");
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值