fd

Author: dingjikerbo

solution/src/main/java/com/inuker/solution/CountOfSmallerNumbersAfterSelf.java

@@ -9,10 +9,12 @@
 
 /**
  * 这题理解了好久才搞明白
- * 建立一棵二叉树，从数组末尾开始遍历，这里每个树的节点要记录sum和dup，dup表示到当前为止，该节点重复的次数
- * sum表示该节点左子树中的节点总数，注意如果有的节点有重复也要算上。现在假如遍历到了数组中的第i个数，则其右边的所有数
- * 都已经生成了节点插到树中，此时我们只要统计从树的root开始，遍历到num[i]对应的树节点为止，所有路径上小于num[i]的节点的sum和dup之和即可，注意也要算上num[i]节点本身的sum
- * 因为num[i]本身可能有重复。
+ * 建立一棵二叉树，从数组末尾开始遍历，这里每个树的节点要记录sum和dup
+ * dup表示到当前为止，该节点重复的次数
+ * sum表示该节点左子树中的节点总数，注意如果有的节点有重复也要算上。
+ * 现在假如遍历到了数组中的第i个数，则其右边的所有数都已经生成了节点插到树中，
+ * 此时我们只要统计从树的root开始，遍历到num[i]对应的树节点为止，所有路径上小于num[i]的节点的sum和dup之和即可，
+ * 注意也要算上num[i]节点本身的sum
  */
 public class CountOfSmallerNumbersAfterSelf {
 

test/src/main/java/com/inuker/test/main.java

@@ -31,30 +31,61 @@
 public class main {
 
     public static void main(String[] args) {
-        int[] arr = new int[]{
-                4,3,2,7,8,2,3,1
-        };
-        for (int n : new FindAllNumbersDisappearedInAnArray().findDisappearedNumbers(arr)) {
+        List<Integer> list = new test().countSmaller(new int[] {
+                5, 2, 6, 1
+        });
+        for (Integer n : list) {
             System.out.print(n + " ");
         }
     }
 
-    public boolean wordBreak(String s, List<String> wordDict) {
-        int n = s.length();
 
-        boolean[] dp = new boolean[n + 1];
-        dp[0] = true;
+    static class test {
+        class Node {
+            Node left, right;
+            int val;
 
-        for (int i = 1; i < s.length(); i++) {
-            for (String word: wordDict) {
-                int j = i - word.length();
-                if (j >= 0 && dp[j] && s.substring(j, i).equals(word)) {
-                    dp[i] = true;
-                    break;
-                }
+            /**
+             * 左子树的节点个数
+             */
+            int sum;
+
+            /**
+             * 节点重复数
+             */
+            int dup = 1;
+
+            public Node(int v, int s) {
+                val = v;
+                sum = s;
+            }
+        }
+
+        public List<Integer> countSmaller(int[] nums) {
+            Node root = null;
+            Integer[] res = new Integer[nums.length];
+            for (int i = nums.length - 1; i >= 0; i--) {
+                root = insert(root, nums[i], 0, res, i);
             }
+            return Arrays.asList(res);
         }
 
-        return dp[n];
+        private Node insert(Node root, int n, int pre, Integer[] res, int i) {
+            if (root == null) {
+                root = new Node(n, 0);
+                res[i] = pre;
+            } else if (root.val == n) {
+                root.dup++;
+                res[i] = pre + root.sum;
+            } else if (n < root.val) {
+                root.sum++;
+                root.left = insert(root.left, n, pre, res, i);
+            } else {
+                root.right = insert(root.right, n, pre + root.sum + root.dup, res, i);
+            }
+            return root;
+        }
     }
+
+
 }