LeetCode315. Count of Smaller Numbers After Self

论坛 期权论坛 脚本     
匿名技术用户   2020-12-27 00:28   26   0

You are given an integer array nums and you have to return a new counts array. The counts array has the property where counts[i] is the number of smaller elements to the right of nums[i].

Example:

Input: [5,2,6,1]
Output: [2,1,1,0] 
Explanation:
To the right of 5 there are 2 smaller elements (2 and 1).
To the right of 2 there is only 1 smaller element (1).
To the right of 6 there is 1 smaller element (1).
To the right of 1 there is 0 smaller element.

这道题我们主要参考了博文:here。我们从后往前遍历数据nums,并将每一个元素插入一个升序列表,这样该元素所在的位置的值就刚好是其右边小于该元素的元素个数。

class Solution {
public:
    vector<int> countSmaller(vector<int>& nums) {
  vector<int> record, res(nums.size());
  for (int i = nums.size() - 1; i >= 0; i--) {
   int d = distance(record.begin(), lower_bound(record.begin(), record.end(), nums[i]));
   res[i] = d;
   record.insert(record.begin() + d, nums[i]);
  }
  return res;
    }
};

分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:7942463
帖子:1588486
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP