第一次博客就当是记笔记吧
本人小白,今天学习了一道关于二叉树的算法,复习一下。
题目
求给定任意数组对于每个nums[i]返回其右侧比它更小的数的个数counts[i]。
解决思路
最容易想到两重循环暴力遍历,通过遍历整个数组,比较每个数与其之后的数的大小,返回更小个数。由于是算法题更注重本身的优化。
在学习了网上大佬们的题解分析以及通过自身总结后,推荐二叉树。
- 二叉树特性:左边节点值小于等于当前节点,右边节点大于当前节点,该特性优化了算法,使得复杂度降低,避免超时。
- 在二叉树结构基础上,对于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){
};
- 这里的nullptr替代NULL(NULL在处理时可能被认为0而出错)
- 定义变量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;
}
};
总结/感想
本题用了二叉树的模型以及后插法非常精妙,本题还引申出一些排序问题和树结构应用问题。刚开始上码路,有点方。有不对的地方还请指教。。。
感谢您的阅读!