LeetCode 315. 计算右侧小于当前元素的个数(C++)

题目描述:

给定一个整数数组 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

 

LeetCode 315. 计算右侧小于当前元素的个数(C++)_第1张图片

下面给出一个更好理解的方法:

这个方法中使用了二叉排序树,把问题转换成了结点一侧的结点个数问题,并在结点中存储。下面是我的代码。

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;
            }
        }
    }
};

LeetCode 315. 计算右侧小于当前元素的个数(C++)_第2张图片

你可能感兴趣的:(LeetCode递归回溯)