第一章 常用的数据结构（下）

第一章 常用的数据结构

四、队列

双端队列

leetcode 239. 滑动窗口最大值

给定一个数组 nums，有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。

返回滑动窗口中的最大值。

进阶：
你能在线性时间复杂度内解决此题吗？

示例:

输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
输出: [3,3,5,5,6,7] 
解释: 

  滑动窗口的位置                最大值
---------------               -----
[1  3  -1] -3  5  3  6  7       3
 1 [3  -1  -3] 5  3  6  7       3
 1  3 [-1  -3  5] 3  6  7       5
 1  3  -1 [-3  5  3] 6  7       5
 1  3  -1  -3 [5  3  6] 7       6
 1  3  -1  -3  5 [3  6  7]      7

思路:采用双端队列来表示当前窗口，双端队列保留当前窗口最大那个数的下标，双端队列的头总是当前队列最大的那个数的下标，（注意双端队列一定存放的是下标，如果存放的是数，就无法判断是否超出当前窗口）。数组的每一个数都被压入和弹出双端队列一次，时间复杂度为O（n)。

比较当前数，如果比队尾元素大，则将队尾弹出，如果比队尾元素小，则压入。

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        deque<int> S;
        vector<int> V;
        //如果数组nums长度小于k，直接寻找最大值
        if(nums.size()<=k){
            int max = nums[0];
            for(int i=0;i<nums.size();i++){
                if(nums[i]>max) max = nums[i];
            }
            V.push_back(max);
            return V;
        }
        for(int i=0;i<nums.size();i++){
            if(!S.empty() && i-S.front()==k){
                S.pop_front();
            }
            while( !S.empty() && nums[i]>nums[S.back()]){
                S.pop_back();
            }
            S.push_back(i);
            if(i>=k-1){
                V.push_back(nums[S.front()]);
            }
        }
        return V;
    }
};

五、树

230. 二叉搜索树中第K小的元素

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。

说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

示例 1:

输入: root = [3,1,4,null,2], k = 1
   3
  / \
 1   4
  \
   2
输出: 1

二叉搜索树的性质：左子树<根节点<右子树
我们只要对二叉树进行中序遍历LDR，然后取出第k个元素即可，考察递归和中序遍历的熟悉程度。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
private:
    vector<int> num;
public:
    int kthSmallest(TreeNode* root, int k) {
        LDR(root);
        return num[k-1];
    }
    void LDR(TreeNode* root){
        if(!root) return;
        LDR(root->left);
        num.push_back(root->val);
        LDR(root->right);
    }
};