右侧更小数

第一次博客就当是记笔记吧

本人小白，今天学习了一道关于二叉树的算法，复习一下。

题目

求给定任意数组对于每个nums[i]返回其右侧比它更小的数的个数counts[i]。

解决思路

最容易想到两重循环暴力遍历，通过遍历整个数组，比较每个数与其之后的数的大小，返回更小个数。由于是算法题更注重本身的优化。

在学习了网上大佬们的题解分析以及通过自身总结后，推荐二叉树。

1. 二叉树特性：左边节点值小于等于当前节点，右边节点大于当前节点，该特性优化了算法，使得复杂度降低，避免超时。
2. 在二叉树结构基础上，对于nums[]数组采用倒序插入，即在示例中对于[5,2,6,1]来说，第一个插入的节点的值为1，依次从右往左插入。这样的好处在于对每一个即将要插入的数，它的右侧的所有数都已经在二叉树中，在插入的同时，已经计算出右侧比它更小数的个数。

具体实现

1.构建二叉树

（C++代码编写）

struct BTNode{    
    int val;    
    int leftSub;//左子树结点数
    BTNode* left;//左子树
    BTNode* right;//右子树    
    BTNode (int x) : val(x), leftSub(0), left(nullptr),  right(nullptr){
};

1. 这里的nullptr替代NULL（NULL在处理时可能被认为0而出错）
2. 定义变量leftSub记录左子树的结点个数

2.定义插入树结点函数

BTNode* Insert(BTNode* root, BTNode* insertNode, int &Count){
    if (!root){
        return insertNode;
    }
    else if (insertNode->val <= root->val) {
        root->leftSub++;
        root->left = Insert(root->left, insertNode, Count);
    }
    else{
        Count += root->leftSub + 1;
        root->right = Insert(root->right, insertNode, Count);
    }
    return root;
}

• root为当前结点，insertNode为插入结点，Count记录右侧比它小的数的个数
• root为空，则创建根结点；若插入值小于等于当前结点值，它将插入到当前结点的左子树，root->leftSub++当前结点的左子树结点个数+1；若插入值大于当前结点值，它将插入到当前结点的右子树，那么计算插入结点值的右侧有多少个比它更小的数，显然，它的个数=当前结点左子树结点个数+1即Count+=root->leftSub+1。

主函数设计

vector<int> countSmaller(vector<int>& nums) {
    vector<int> counts(nums.size());
    if (nums.empty())
        return counts;
    BTNode* root = nullptr;
    for (int i = nums.size() - 1; i >= 0; i--) {
        int count = 0;
        root = Insert(root, new BTNode(nums.at(i)), count);
        counts.at(i) = count;
    }
    return counts;
}

• 给出一个for循环遍历（从后往前）依次插入，Insert插入函数见上，最后返回counts[]。

源代码

struct BTNode{
    int val;
    int leftSub;
    BTNode* left;
    BTNode* right;
    BTNode (int x) : val(x), leftSub(0), 
            left(nullptr),  right(nullptr) {}
};
class Solution {
public:
    vector<int> countSmaller(vector<int>& nums) {
        vector<int> counts(nums.size());
        if (nums.empty())
            return counts;
        BTNode* root = nullptr;
        for (int i = nums.size() - 1; i >= 0; i--) {
            int count = 0;
            root = Insert(root, new BTNode(nums.at(i)), count);
            counts.at(i) = count;
        }
        return counts;
    }
private:
    BTNode* Insert(BTNode* root, BTNode* insertNode, int &Count){
        if (!root){
            return insertNode;
        }
        else if (insertNode->val <= root->val) {
            root->leftSub++;
            root->left = Insert(root->left, insertNode, Count);
        }
        else{
            Count += root->leftSub + 1;
            root->right = Insert(root->right, insertNode, Count);
        }
        return root;
    }
};

总结/感想

本题用了二叉树的模型以及后插法非常精妙，本题还引申出一些排序问题和树结构应用问题。刚开始上码路，有点方。有不对的地方还请指教。。。

感谢您的阅读！