[LeetCode][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 个更小的元素

---

思路：
（思路以及代码都是参考链接1的）
使用二叉排序树，把问题转换成了结点一侧的结点个数问题，并在结点中存储。
代码：

class Solution {
public:
    struct BSTNode{
    int val;
    int count;     //二叉树左子树中的节点的个数
    BSTNode *left;
    BSTNode *right;
    BSTNode(int x):val(x),left(NULL),right(NULL),count(0){}
    };

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

    
    vector<int> countSmaller(vector<int>& nums) {
        vector<int >  result;//最终逆序数组
        vector<BSTNode *>  node_vec;//创建的二叉树查找池
        vector<int> 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<node_vec.size();i++){
            int count_small=0;
            BST_insert(node_vec[0],node_vec[i],count_small);
            count.push_back(count_small);
        }
        for(int i=node_vec.size()-1;i>=0;i--){
            delete  node_vec[i];
            result.push_back(count[i]);
        }
        return result;

    }
};

结果：

参考链接：
[1] 16406070101：LeetCode 315. 计算右侧小于当前元素的个数（C++）