From 28deddd5d78064274b76df56126c9a72163ad361 Mon Sep 17 00:00:00 2001
From: JayH2018 <42029911+JayH2018@users.noreply.github.com>
Date: Tue, 28 Aug 2018 14:36:56 +0800
Subject: [PATCH] source coding is a little complex

---
 DataStructures/Trees/LevelOrderTraversal.java | 14 +++++---------
 1 file changed, 5 insertions(+), 9 deletions(-)

diff --git a/DataStructures/Trees/LevelOrderTraversal.java b/DataStructures/Trees/LevelOrderTraversal.java
index 8cb304f18..e20753638 100644
--- a/DataStructures/Trees/LevelOrderTraversal.java
+++ b/DataStructures/Trees/LevelOrderTraversal.java
@@ -37,14 +37,10 @@ public class LevelOrderTraversal
            return 0;
         else
         {
-            /* compute  height of each subtree */
-            int lheight = height(root.left);
-            int rheight = height(root.right);
-             
-            /* use the larger one */
-            if (lheight > rheight)
-                return(lheight+1);
-            else return(rheight+1); 
+            /**
+             * return the larger one;
+             */
+            return Math.max(height(root.left),height(root.right)) + 1;
         }
     }
  
@@ -75,4 +71,4 @@ public class LevelOrderTraversal
        System.out.println("Level order traversal of binary tree is ");
        tree.printLevelOrder();
     }
-}
\ No newline at end of file
+}