【算法】反转队列前K个元素

反转队列前K个元素

想到队列的反转那肯定首先想到的就是堆栈(同理堆栈反转也可以利用队列)

有一个办法我们可以将K个元素入栈,然后可以另一个队列将剩下的元素放入,随后我们首先将栈元素回归原来的队列,最后将临时队列也入队。

//使用一个辅助的队列和堆栈实现 //前k个放入堆栈,后面的放入队列 //取出时堆栈先入后出,达到反转目的 void reversalQueue(queue<int>& que, int k) { stack<int> stk; queue<int> tmpque; int len = que.size(); for (int i = 0; i < len; ++i) { if (i < k) { stk.push(que.front()); que.pop(); } else { tmpque.push(que.front()); que.pop(); } } while (!stk.empty()) { que.push(stk.top()); stk.pop(); } while (!tmpque.empty()) { que.push(tmpque.front()); tmpque.pop(); } } 

显然上面的方法不是很靠谱,我们可以直接将K个元素入堆栈后,反转剩下的元素,就可以达到目的。

//使用大小为k的辅助堆栈 //随后将没有入栈的部分扔到队列前面 void reversalQueue2(queue<int>& que, int k) { //默认k<que.size() stack<int> stk; for (int i = 0; i < k; ++i) { stk.push(que.front()); que.pop(); } while (!stk.empty()) { que.push(stk.top()); stk.pop(); } for (int i = 0; i < que.size() - k; ++i) { que.push(que.front()); que.pop(); } } 
点赞