315. Count of Smaller Numbers After Self 【Hard】 分治?

题目:给一个数组nums,求每一个nums元素的右手边比它小的元素的个数

思路:用一个数组pos给nums排序——把nums从右边的元素开始,一个个有序地插入到pos中,该元素的位置就是该元素在nums中右手边比它小的元素的个数。

看博客才会的。

class Solution {
public:
    vector countSmaller(vector& nums) {
    	int len = nums.size();
        vector t, pos(len);

    	if (len == 0) return t;
        int l, r, m;

        t.push_back(nums[len - 1]);
        pos[len - 1] = 0;
        
        for (int i = len - 2; i >= 0; --i) {
        	l = 0;
        	r = t.size() - 1;
        	while (l <= r) {
        		m = (l + r) / 2;
        		if (nums[i] > t[m]) l = m + 1;
        		else r = m - 1;
        	}

        	t.insert(t.begin() + l, nums[i]);
        	pos[i] = l;
        }

        return pos;
    }
};


你可能感兴趣的:(算法分析课程作业)