leetcode315:计算右侧小于当前元素的个数:三种方法的比较

求解:右侧小于当前元素的个数三种方法
第一种:暴力解法,超时,算法复杂度为O(n^2)
第二种:使用二分查找,但是算法复杂度还是O(n^2)
第三种:可以使用bittree:树状数组进行求解,算法复杂度为O(nlgn)

package divide_and_conquer;

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

public class leetcode_315 {
	
	/**
	 * 
	 *315 计算右侧小于当前数字的个数
	 * 
	 */

	public static void main(String[] args) {
		
		
		
		int[] array = {5,2,6,1};
		List<Integer> list = new ArrayList<>();
		list = countSmaller3(array);
		for(int i : list) {
			System.out.println(i);
		}
		
	}
	
	/**
	 * 第一种最朴素的思路:最笨的方法,算法复杂度为O(n^2)
	 * 
	 * 
	 */
    public static  List<Integer> countSmaller(int[] nums) {

    	List<Integer> list = new ArrayList<Integer>();
    	
    	for(int i = 0 ; i< nums.length;i++) {
    		int number = 0;
    		for(int j = i+1;j<nums.length;j++) {
    			if(nums[i] > nums[j]) {
    				++number;
    			}
    		}
    		list.add(number);
    	}
    	
    	
		return list;
    }
    
	/**
	 * 第二种方法:可以对nums数组进行从后往前进行遍历,然后放入到arrayList中排好顺序,
	 *           当便利nums[i]的时候,可以使用二分查找得到插入的位置
	 *           
	 *           二分查找需要O(lgN),但是插入的过程复杂度为O(n)
	 *  算法复杂度:O(N^2)
	 * 
	 */
    public static List<Integer> countSmaller2(int[] nums) {

    	ArrayList<Integer> list = new ArrayList<Integer>();
    	
    	int len = nums.length;
    	Integer[] result = new Integer[len];
    	for(int i = len-1;i>=0;i--) {
    		//将每个数插入到list中//使用二分查找
    		int start = 0; int end = list.size();
    		
		    while(start<end) {
		    	int middle = start+(end-start)/2;
		    	//判断中间的数
		    	if(list.get(middle) < nums[i]) {//严格小于的话,只能在后面部分,并且不包含middle
		    		start = middle+1;
		    	}else {
		    		end = middle;
		    	}
		    }
		    result[i] = start;
		    list.add(start,nums[i]);
    	}
		return Arrays.asList(result);	
    }
    /**
     * 
     * 第三种方法:使用bitree:树状数组
     * 
     * 算法复杂度:便利一次nums为O(n)
     *			求和和更新的复杂度都为O(lgn)
     *所以整体的算法复杂度为O(nlgn)           
     *
     * 
     * 
     * 
     */
    public static  List<Integer> countSmaller3(int[] nums) {
        if (nums.length == 0) {
            return new ArrayList<>();
        }
        int min = Integer.MAX_VALUE; // nums数组最小值
        for (int value : nums) {
            if (value < min) {
                min = value;
            }
        }
        for (int i = 0; i < nums.length; i++) {
            nums[i] = nums[i] - min + 1;
        }
        
        int max = Integer.MIN_VALUE;
        for (int value : nums) {
            if (value > max) {
                max = value;
            }
        }
        
        int[] BITree = new int[max + 1];
        BITree[0] = 0;
        int[] countArr = new int[nums.length];
        for (int i = nums.length - 1; i >= 0; i--) {
            int count = getSum(nums[i] - 1, BITree);
            countArr[i] = count;
            update(nums[i], BITree);
        }
        List<Integer> result = new ArrayList<>();
        for (int value : countArr) {
            result.add(value);
        }
        return result;
    }
    
    public static int getSum(int value, int[] BITree) { // 获得a[i]从1,value的和
        int sum = 0;
        while (value > 0) {
            sum += BITree[value];
            value -= (value & -value);
        }
        return sum;
    }

    public static void update(int value, int[] BITree) {
        while (value <= BITree.length - 1) {
            BITree[value] += 1;
            value += (value & -value);
        }
    }
}


你可能感兴趣的:(leetcode)