add 1022

fishercoder1534 · fishercoder1534 · commit fc3302f74b84 · 2019-04-07T09:19:41.000-07:00
diff --git a/README.md b/README.md
@@ -27,6 +27,7 @@ Your ideas/fixes/algorithms are more than welcome!
 
 |  #  |      Title     |   Solutions   | Time          | Space         | Video  | Difficulty  | Tag                   
 |-----|----------------|---------------|---------------|---------------|--------|-------------|-------------
+|1022|[Sum of Root To Leaf Binary Numbers](https://leetcode.com/problems/sum-of-root-to-leaf-binary-numbers/)|[Solution](../master/src/main/java/com/fishercoder/solutions/_1022.java) | O(n) | O(n) | |Easy|
 |1021|[Remove Outermost Parentheses](https://leetcode.com/problems/remove-outermost-parentheses/)|[Solution](../master/src/main/java/com/fishercoder/solutions/_1021.java) | O(n) | O(n) | |Easy|
 |1018|[Binary Prefix Divisible By 5](https://leetcode.com/problems/binary-prefix-divisible-by-5/)|[Solution](../master/src/main/java/com/fishercoder/solutions/_1018.java) | O(n) | O(1) | |Easy|
 |1013|[Pairs of Songs With Total Durations Divisible by 60](https://leetcode.com/problems/pairs-of-songs-with-total-durations-divisible-by-60/)|[Solution](../master/src/main/java/com/fishercoder/solutions/_1013.java) | O(n) | O(1) | |Easy|
diff --git a/src/main/java/com/fishercoder/solutions/_1022.java b/src/main/java/com/fishercoder/solutions/_1022.java
@@ -0,0 +1,68 @@
+package com.fishercoder.solutions;
+
+import com.fishercoder.common.classes.TreeNode;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * Source: https://leetcode.com/problems/sum-of-root-to-leaf-binary-numbers/
+ *
+ * 1022. Sum of Root To Leaf Binary Numbers
+ *
+ * Given a binary tree, each node has value 0 or 1.  Each root-to-leaf path represents a binary number starting with the most significant bit.  For example, if the path is 0 -> 1 -> 1 -> 0 -> 1, then this could represent 01101 in binary, which is 13.
+ *
+ * For all leaves in the tree, consider the numbers represented by the path from the root to that leaf.
+ *
+ * Return the sum of these numbers.
+ *
+ * Example 1:
+ *
+ *        1
+ *       / \
+ *      0   1
+ *     / \ / \
+ *    0  1 0 1
+ *
+ * Input: [1,0,1,0,1,0,1]
+ * Output: 22
+ * Explanation: (100) + (101) + (110) + (111) = 4 + 5 + 6 + 7 = 22
+ *
+ * Note:
+ *
+ * The number of nodes in the tree is between 1 and 1000.
+ * node.val is 0 or 1.
+ * The answer will not exceed 2^31 - 1.
+ * */
+public class _1022 {
+    public static class Solution1 {
+        public int sumRootToLeaf(TreeNode root) {
+            List<List<Integer>> paths = new ArrayList<>();
+            dfs(root, paths, new ArrayList<>());
+            int sum = 0;
+            for (List<Integer> list : paths) {
+                int num = 0;
+                for (int i : list) {
+                    num = (num << 1) + i;
+                }
+                sum += num;
+            }
+            return sum;
+        }
+
+        private void dfs(TreeNode root, List<List<Integer>> paths, List<Integer> path) {
+            path.add(root.val);
+            if (root.left != null) {
+                dfs(root.left, paths, path);
+                path.remove(path.size() - 1);
+            }
+            if (root.right != null) {
+                dfs(root.right, paths, path);
+                path.remove(path.size() - 1);
+            }
+            if (root.left == null && root.right == null) {
+                paths.add(new ArrayList<>(path));
+            }
+        }
+    }
+}
diff --git a/src/test/java/com/fishercoder/_1022Test.java b/src/test/java/com/fishercoder/_1022Test.java
@@ -0,0 +1,29 @@
+package com.fishercoder;
+
+import com.fishercoder.common.classes.TreeNode;
+import com.fishercoder.common.utils.TreeUtils;
+import com.fishercoder.solutions._1022;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+import java.util.Arrays;
+
+import static junit.framework.Assert.assertEquals;
+
+public class _1022Test {
+    private static _1022.Solution1 solution1;
+    private static TreeNode root;
+
+    @BeforeClass
+    public static void setup() {
+        solution1 = new _1022.Solution1();
+    }
+
+    @Test
+    public void test1() {
+        root = TreeUtils.constructBinaryTree(Arrays.asList(1, 0, 1, 0, 1, 0, 1));
+        TreeUtils.printBinaryTree(root);
+        assertEquals(22, solution1.sumRootToLeaf(root));
+    }
+
+}
