Skip to main content

Check for continuous binary tree in scala

Scala program for Check for continuous binary tree. Here problem description and explanation.

/* 
  Scala program for
  Check if binary tree is continuous tree
  Using recursion
*/
// Binary Tree Node
class TreeNode(var data: Int,
	var left: TreeNode,
		var right: TreeNode)
{
	def this(data: Int)
	{
		// Set node value
		this(data, null, null);
	}
}
class BinaryTree(var root: TreeNode)
{
	def this()
	{
		this(null);
	}
	def absValue(num: Int): Int = {
		if (num < 0)
		{
			return -num;
		}
		return num;
	}
	// Check tree is continuous or not
	def isContinuous(node: TreeNode): Boolean = {
		if (node != null)
		{
			if ((node.left != null && 
                 absValue(node.data - node.left.data) != 1) ||
                (node.right != null && 
                 absValue(node.data - node.right.data) != 1))
			{
				// Case
				// When fail continuous tree rule
				return false;
			}
			if (isContinuous(node.left) && 
                isContinuous(node.right))
			{
				return true;
			}
			// When node value is not satisfied
			//  continuous tree properties
			return false;
		}
		return true;
	}
}
object Main
{
	def main(args: Array[String]): Unit = {
		// Create new tree
		var tree: BinaryTree = new BinaryTree();
		/*  
		   Binary Tree
		   ------------
		       5
		     /   \
		    4     4
		   /     /  \
		  3     5    3
		   \
		    2
		        
		*/
		// Add tree node
		tree.root = new TreeNode(5);
		tree.root.left = new TreeNode(4);
		tree.root.right = new TreeNode(4);
		tree.root.left.left = new TreeNode(3);
		tree.root.right.right = new TreeNode(3);
		tree.root.right.left = new TreeNode(5);
		tree.root.left.left.right = new TreeNode(2);
		if (tree.isContinuous(tree.root) == true)
		{
			println("Continuous Tree ");
		}
		else
		{
			println("Not Continuous Tree ");
		}
		// Case 2
		/* 
		       5
		     /   \
		    4     4
		   /     /  \
		  3     5    3
		   \
		    1  <--- change value
		        
		*/
		tree.root.left.left.right.data = 1;
		if (tree.isContinuous(tree.root) == true)
		{
			println("Continuous Tree ");
		}
		else
		{
			println("Not Continuous Tree ");
		}
	}
}

Output

Continuous Tree
Not Continuous Tree




Comment

Please share your knowledge to improve code and content standard. Also submit your doubts, and test case. We improve by your feedback. We will try to resolve your query as soon as possible.

New Comment