【Leetcode每日一题】面试题 03.05. 栈排序

题目

思路分析

如果这道题没有空间复杂度O(1)的限制，那么用两个栈倒腾来倒腾去的方式是最简单的，这里简述一下方法，代码可以参考这道题的评论区，全是这种解法：
初始化两个栈，一个主栈，一个辅助栈。某个数入主栈前，将主栈栈顶大于该数的值依次弹出到辅助栈，再将该数入栈，最后再将辅助栈中的值弹回主栈。以下图为例（字有点丑，将就看看）：

当题目加入空间复杂度O(1)这个限制的时候，其实就只能用堆排序了。题目中描述的“使最小元素位于栈顶”其实也很明显是小顶堆的特征。面试时候也经常听到一个问题：怎么找出n个数里的k个最大值？解法也是用堆排序。这两道题下面都会讲一讲。
堆排序是一种利用了完全二叉树性质的排序方法。完全二叉树简单来说就是除了最后一层，其他每一层都是满的，最后一层的节点都在左边连续排布。最小堆的特征描述很简单：每个节点的值都小于其左右子树的节点值。它的实现也很方便，虽然是用二叉树的表现形式，但是大部分场景下只需要用数组就可以实现，实在是手残福音。下面用刚才的例子解释下如何实现各个操作：

1. push：将当前元素插入到完全二叉树最后一个节点，并与父节点进行比较，若当前节点小于父节点，则交换当前节点与父节点的位置，并递归地向上比较，直到当前节点大于父节点或当前节点为根节点为止。以下为示例：
   
2. peek：根据最小堆的性质，构建的完全二叉树的根节点一定是这棵树的最小值（因为根一定比左子节点和右子节点小）。因此这个操作只需要返回根节点就行。
3. pop：这个操作比较复杂，第一步是将根节点和最后一个叶子节点交换并删除最后一个节点。此时我们已经将最小值弹出，但是交换后的树不一定满足最小堆的性质，需要进行维护。因此，第二部是从根节点开始，将父节点与左右节点中的最小值进行比较，如果父节点大于左右节点中的最小值，将这两个节点进行交换，并递归地再将该节点与交换后的左右节点进行比较，直到该节点为叶子节点或该节点比左右节点都小为止。以下为示例：
   
   上面说到，堆排序虽然是用完全二叉树的性质呈现，但是实际上代码实现的时候一般是用数组，这是因为根据完全二叉树的性质，我们可以从父节点的下标计算出子节点的下标，也能从子节点的下标计算出父节点的下标，而不需要用到指针（当然，如果使用双向指针建树也是可以的，但是我用双向指针总是出错，唉，我是废物55555555555）：
   
   最后，面试题中的求n个数的k个最大值怎么用堆排序呢？首先，将前k个值建成最小堆（不是最大堆哦，容易搞错），然后将后面的n-k个数与堆顶（也就是根节点）比较，如果该值大于堆顶，将堆顶替换为该值，并再更新一次当前的最小堆。该方法时间复杂度为O(n)，空间复杂度O(k)，综合看应该是最优秀的解法。

代码

class SortedStack {
public:
    vector<int>stack;
    SortedStack() {
    }
    void swap(int &a, int &b) {
        int tmp = a;
        a = b;
        b = tmp;
    }
    void push(int val) {
        stack.push_back(val);
        int i = stack.size() - 1;
        while (stack[i] < stack[(i - 1) / 2]) {
            swap(stack[i], stack[(i - 1) / 2]);
            i = (i - 1) / 2;
        }
    }
    
    void pop() {
        if (stack.size() == 0) {
            return;
        }
        swap(stack[0], stack[stack.size() - 1]);
        stack.pop_back();
        int i = 0;
        
        while ((stack.size()) && ((i * 2 + 1) < stack.size())) {
            int next = (i * 2 + 1);
            if ((i * 2 + 2) < stack.size()) {
                next = stack[i * 2 + 1] < stack[i * 2 + 2] ? i * 2 + 1: i * 2 + 2;
            }
            if (stack[i] > stack[next]) {
                swap(stack[i], stack[next]);
                i = next;
            } else {
                break;
            }
        }
        return;
    }
    
    int peek() {
        if (stack.size() == 0) {
            return -1;
        }
        return stack[0];
    }
    
    bool isEmpty() {
        return stack.size() == 0;
    }
};

/**
 * Your SortedStack object will be instantiated and called as such:
 * SortedStack* obj = new SortedStack();
 * obj->push(val);
 * obj->pop();
 * int param_3 = obj->peek();
 * bool param_4 = obj->isEmpty();
 */