给定一个整数数组 nums,按要求返回一个新数组 counts。数组 counts 有该性质: counts[i]
的值是 nums[i]
右侧小于 nums[i]
的元素的数量。
输入: [5,2,6,1]
输出: [2,1,1,0]
解释:
5 的右侧有 2 个更小的元素 (2 和 1).
2 的右侧仅有 1 个更小的元素 (1).
6 的右侧有 1 个更小的元素 (1).
1 的右侧有 0 个更小的元素.
首先这道题要了解一种分治的思想:
将一个规模为N的问题分解为K个规模较小的子问题,这些子问题相互独立且与原问题性质相同,求出子问题的解时进行合并,就可得到原问题的解。
一般步骤:
1.分解,将要解决的问题划分为若干个规模较小的同类问题;
2.求解,当子问题划分的足够小的时候。用较简单的方法求解;
3.合并:按照原问题的要求,将子问题的解逐层合并,构成原问题的解。
我们可以还想到:
在归并两个有序数列时,当需要将前一个数组元素的指针i插入时,对应的count[i],即为指向后一个数组的指针j的值。
可以看到其实就是归并排序的归并时加了一些步骤。在归并时改变count的值。
下面是我的代码:
class Solution {
public:
vector countSmaller(vector& nums) {
vector> vec;//从新存储原数组以保证顺序
vector count;//保存结果
for(int i=0;i> &sub_vec1,vector>& sub_vec2,vector> &vec,vector& count){
int i=0;
int j=0;
while (i> &vec,vector &count){
if(vec.size()<2){
return ;
}
int mid=vec.size()/2;
vector> sub_vec1;
vector> sub_vec2;
for(int i=0;i
下面给出一个更好理解的方法:
这个方法中使用了二叉排序树,把问题转换成了结点一侧的结点个数问题,并在结点中存储。下面是我的代码。
struct BSTNode{
int val;
int count; //二叉树左子树中的节点的个数
BSTNode *left;
BSTNode *right;
BSTNode(int x):val(x),left(NULL),right(NULL),count(0){}
};
class Solution {
public:
vector countSmaller(vector& nums) {
vector result;//最终逆序数组
vector node_vec;//创建的二叉树查找池
vector count; //从后向前插入过程中的,比当前结点值小的count_small数组
for(int i=nums.size()-1;i>=0;i--){ //按照从后向前的顺序创建二叉查找树结点
node_vec.push_back(new BSTNode(nums[i]));
}
count.push_back(0);第一个节点的count_small=0
for(int i=1;i=0;i--){
delete node_vec[i];
result.push_back(count[i]);
}
return result;
}
void BST_insert(BSTNode *node,BSTNode *insert_node,int &count_small){
if(insert_node->val<=node->val){ //count_small二叉排序树中比insert_node 值小的结点个数
node->count++;
if(node->left){
BST_insert(node->left,insert_node,count_small);
}else{
node->left=insert_node;
}
}else{
count_small+=node->count+1;
if(node->right){
BST_insert(node->right,insert_node,count_small);
}else{
node->right=insert_node;
}
}
}
};