【算法】反转队列前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();
	}
}
©️2020 CSDN 皮肤主题: 大白 设计师: CSDN官方博客 返回首页
实付0元
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值