Leetcode - Count of Smaller Numbers After Self

My code:

import java.util.Arrays;
import java.util.List;

public class Solution {
    class Node {
        int val = 0;
        int sum = 0;
        int dup = 0;
        Node left;
        Node right;
        Node(int val, int sum) {
            this.val = val;
            this.sum = sum;
            this.dup = 1;
        }
    }
    
    
    public List<Integer> countSmaller(int[] nums) {
        Node root = null;
        Integer[] arr = new Integer[nums.length];
        for (int i = nums.length - 1; i >= 0; i--) {
            root = helper(root, nums[i], i, arr, 0);
        }
        
        return Arrays.asList(arr);
    }
    
    private Node helper(Node root, int num, int i, Integer[] arr, int preSum) {
        if (root == null) {
            root = new Node(num, 0);
            arr[i] = preSum;
        }
        else if (root.val == num) {
            root.dup++;
            arr[i] = preSum + root.sum;
        }
        else if (root.val < num) {
            root.right = helper(root.right, num, i, arr, preSum + root.sum + root.dup);
        }
        else {
            root.sum += 1;
            root.left = helper(root.left, num, i, arr, preSum);
        }
        return root;
    }
    
}

reference:
https://discuss.leetcode.com/topic/31405/9ms-short-java-bst-solution-get-answer-when-building-bst/2

这道题目的这个解法,还是很难理解的。花了挺长一段时间去理解。
做难题真的花时间,今天的效率也真够低的。
他通过建立一个 类似于BST的树,来解决这个问题。
sum 存的是,当前结点所表示的value,和当前访问处的value,这段范围内,小于当前结点value的个数。
比较绕。
preSum 表示的是,当前结点表示的value,在该value右侧,比他小的个数

比如:
[3, 2, 1, 2, 6, 1]
1(0, 1) sum = 0, dup = 1
=>
1(0, 1)
\ preSum = 0 + 0 + 1
6(0, 1)

=>
1(0, 1)
\ preSum = 0 + 0 + 1
6(1, 1) dup++
/ preSum = 1
2(0, 1)

=>
1(0, 2) dup++

6(1, 1)
/
2(0, 1)

=>

1(0, 2)

6(1, 1)
/
2(0, 2) dup++
preSum = preSum + root.sum

=>

1(0, 2)

6(1, 1)
/
2(0, 2)

3(0, 1)

还是不太好理解的。

Solution2:
My code:

public class Solution {
    public List<Integer> countSmaller(int[] nums) {
        List<Integer> list = new ArrayList<Integer>();
        Integer[] ret = new Integer[nums.length];
        for (int i = nums.length - 1; i >= 0; i--) {
            int index = findIndex(list, nums[i]);
            ret[i] = index;
            list.add(index, nums[i]);
        }
        
        return Arrays.asList(ret);
    }
    
    private int findIndex(List<Integer> ret, int target) {
        if (ret.size() == 0) {
            return 0;
        }
        int begin = 0;
        int end = ret.size() - 1;
        
        while (begin <= end) {
            int mid = begin + (end - begin) / 2;
            if (ret.get(mid) > target) {
                end = mid - 1;
            }
            else if (ret.get(mid) < target) {
                begin = mid + 1;
            }
            else {
                while (mid - 1 >= 0 && ret.get(mid - 1).equals(ret.get(mid))) {
                    mid--;
                }
                return mid;
            }
        }
        return begin;
    }
}

reference:
https://discuss.leetcode.com/topic/31173/my-simple-ac-java-binary-search-code/2

这个时间复杂度很大。
注意, Integer 比较,用 .equals() !!!!!

还有一种 Merge sort的写法,以后再理解吧。

Anyway, Good luck, Richardo! — 10/08/2016

My code:

public class Solution {
    int[] count;
    int[] index;
    public List<Integer> countSmaller(int[] nums) {
        List<Integer> ret = new ArrayList<Integer>();
        if (nums == null || nums.length == 0) {
            return ret;
        }
        
        int len = nums.length;
        count = new int[len];
        index = new int[len];
        for (int i = 0; i < len; i++) {
            index[i] = i;
        }
        
        mergesort(nums, 0, len - 1);
        
        for (int i = 0; i < count.length; i++) {
            ret.add(count[i]);
        }
        
        return ret;
    }
    
    private void mergesort(int[] nums, int start, int end) {
        if (start >= end) {
            return;
        }
        int mid = start + (end - start) / 2;
        mergesort(nums, start, mid);
        mergesort(nums, mid + 1, end);
        merge(nums, start, mid, end);
    }
    
    private void merge(int[] nums, int start, int mid, int end) {
        int[] arr = new int[end - start + 1];
        for (int i = start; i <= end; i++) {
            arr[i - start] = nums[i];
        }
        int[] temp = new int[end - start + 1];
        for (int i = start; i <= end; i++) {
            temp[i - start] = index[i];
        }
        
        int p = start;
        int p1 = start;
        int p2 = mid + 1;
        int cnt = 0;
        while (p1 <= mid && p2 <= end) {
            if (arr[p1 - start] > arr[p2 - start]) {
                cnt++;
                index[p] = temp[p2 - start];
                nums[p] = arr[p2 - start];
                p++;
                p2++;
            }
            else {
                index[p] = temp[p1 - start];
                nums[p] = arr[p1- start];
                count[index[p]] += cnt;
                p++;
                p1++;
            }
        }
        
        while (p2 <= end) {
            index[p] = temp[p2 - start];
            nums[p] = arr[p2 - start];
            p++;
            p2++;
        }
        
        while (p1 <= mid) {
            index[p] = temp[p1 - start];
            nums[p] = arr[p1 - start];
            count[index[p]] += cnt;
            p++;
            p1++;
        }
    }
}

这是一种新的办法,并且可以保证时间复杂度是 O(nlogn)
上面的BST方法,在最坏情况下复杂度是 O(n^2)

这个 merge sort 方法其实挺难理解的。问了同学才略懂。
当我们merge 到一层时,假设有左右两个部分,left and right
left is sorted and right is sorted
然后我们 merge 的时候,
right部分的所有元素, 他们的 count of smaller numbers 一定是 0!
我们要关注的是左部分。 left part
如果 left[p1] < right[p2]
按照以往的merge sort经验,我们有一个arrray来放merge过后的新数组。

nums[p] = left[p1]
p++;
p1++;

if left[p1] > right[p2]
nums[p] = right[p2];
p++;
p2++;

这里要注意了,我们需要记录的,是右侧比左侧小的个数。
所以我们有一个变量 cnt
当 left[p1] > right[p2]
我们需要把right[p2] 移动到 left[p1] 的左侧,记住,这个时候,
对于 left[p1, mid],他们的 count of smaller numbers 至少有了1,即,这个 right[p2]
同时,因为merge sort, 原有的顺序都被打乱了。我们需要记录下,新merge过的数组,他们的元素,在原数组中的index
这样,当我们发现右边比左边小时,cnt++
当我们发现左边比右边小时,就可以更新 count 了
对应的index 就是 index[p]

差不多理解下把。

Anyway, Good luck, Richardo! — 10/09/2016

    原文作者:Richardo92
    原文地址: https://www.jianshu.com/p/47aff0f63c62
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞