refactor 623

fishercoder1534 · fishercoder1534 · commit 066f23267557 · 2019-08-03T07:03:54.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_623.java b/src/main/java/com/fishercoder/solutions/_623.java
@@ -70,31 +70,34 @@
  */
 public class _623 {
 
-    public TreeNode addOneRow(TreeNode root, int v, int d) {
-        if (d == 1) {
-            TreeNode newRoot = new TreeNode(v);
-            newRoot.left = root;
-            return newRoot;
-        } else {
-            dfs(root, v, d);
-            return root;
+    public static class Solution1 {
+
+        public TreeNode addOneRow(TreeNode root, int v, int d) {
+            if (d == 1) {
+                TreeNode newRoot = new TreeNode(v);
+                newRoot.left = root;
+                return newRoot;
+            } else {
+                dfs(root, v, d);
+                return root;
+            }
         }
-    }
 
-    private void dfs(TreeNode root, int v, int d) {
-        if (root == null) {
-            return;
-        }
-        if (d == 2) {
-            TreeNode newLeft = new TreeNode(v);
-            TreeNode newRight = new TreeNode(v);
-            newLeft.left = root.left;
-            newRight.right = root.right;
-            root.left = newLeft;
-            root.right = newRight;
-        } else {
-            dfs(root.left, v, d - 1);
-            dfs(root.right, v, d - 1);
+        private void dfs(TreeNode root, int v, int d) {
+            if (root == null) {
+                return;
+            }
+            if (d == 2) {
+                TreeNode newLeft = new TreeNode(v);
+                TreeNode newRight = new TreeNode(v);
+                newLeft.left = root.left;
+                newRight.right = root.right;
+                root.left = newLeft;
+                root.right = newRight;
+            } else {
+                dfs(root.left, v, d - 1);
+                dfs(root.right, v, d - 1);
+            }
         }
     }
 
