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

最新推荐文章于 2023-04-07 20:10:15 发布

原创最新推荐文章于 2023-04-07 20:10:15 发布 · 2.9k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #C++

算法专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种使用堆栈辅助的队列反转方法，通过将队列的前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();
	}
}