剑指offer面试题63:二叉搜索树的第k个节点 Java实现

题目:二叉搜索树的第k个节点
         给定一颗二叉搜索树,请找出其中的第k大的节点。例如,下图中的二叉搜索树中,按节点数值大小顺序第三个节点的值 是4.
  剑指offer面试题63:二叉搜索树的第k个节点 Java实现_第1张图片
算法分析:
如果按照中序遍历的顺序遍历一棵二叉搜索树,遍历序列的数值是递增排序的。上图中的二叉搜索树的中序遍历序列为{2,3,4,5,6,7,8},因此,只需要用中序遍历算法遍历一棵二叉搜索树,就很容易找出它的第k大结点。

算法源程序:
/**************************************************************      
* Copyright (c) 2016, 
* All rights reserved.                   
* 版 本 号:v1.0                   
* 题目描述:二叉搜索树的第k个节点
*		         给定一颗二叉搜索树,请找出其中的第k大的节点。例如,下图中的二叉搜索树中,按节点数值大小顺序第三个节点的值 是4.
* 输入描述:无
* 程序输出:第3大的节点是:
*			4
* 问题分析: 无
* 算法描述:如果按照中序遍历的顺序遍历一棵二叉搜索树,遍历序列的数值是递增排序的。
* 			上图中的二叉搜索树的中序遍历序列为{2,3,4,5,6,7,8},因此,只需要用中序遍历算法遍历一棵二叉搜索树,就很容易找出它的第k大结点。

* 完成日期:2016-10-17
***************************************************************/

package org.marsguo.offerproject63;

class TreeNode{
	int val;
	public TreeNode left = null;
	public TreeNode right = null;
	public TreeNode(int val){
		this.val = val;
	}
	public String toString(){
		return val + "";
	}
}

class SolutionMethod1{
	public TreeNode inorderTraversal(TreeNode root,int k){
		if(root == null || k == 0){
			return null;
		}
		int[] temp = {k};
		return inorderTraversalCore(root,temp);
	}
	
	public TreeNode inorderTraversalCore(TreeNode root,int[] k){
		TreeNode target = null;
		
		if(root.left != null )
			target = inorderTraversalCore(root.left,k);
		
		if(target == null){
			if(k[0] == 1)
				target = root;
			k[0]--;
		}
		if(target == null && root.right != null)
			target = inorderTraversalCore(root.right, k);
		
		return target;
	}
}

public class KthNode {
	public static void main(String[] args){
		SolutionMethod1 solution1 = new SolutionMethod1();
		
		TreeNode n1 = new TreeNode(5);
		TreeNode n2 = new TreeNode(3);
		TreeNode n3 = new TreeNode(7);
		TreeNode n4 = new TreeNode(2);
		TreeNode n5 = new TreeNode(4);
		TreeNode n6 = new TreeNode(6);
		TreeNode n7 = new TreeNode(8);
//		TreeNode n8 = new TreeNode(8);
//		TreeNode n9 = new TreeNode(9);
		
		n1.left = n2;
		n1.right = n3;
		
		n2.left = n4;
		n2.right = n5;
		
		n3.left = n6;
		n3.right = n7;
		
//		n4.left = n8;
//		n4.right = n9;
		System.out.println("第" + 3 + "大的节点是:");
		System.out.println(solution1.inorderTraversal(n1, 3));
	}
}

程序运行结果:
剑指offer面试题63:二叉搜索树的第k个节点 Java实现_第2张图片


你可能感兴趣的:(剑指Offer,算法,面试题)