1025. 反转链表 (25)

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <vector>
#include <map>
#include <stack>
#include <queue>
#include <algorithm>
using namespace std;


int main()
{
	freopen("d://1.txt", "r", stdin);
	freopen("d://2.txt", "w", stdout);
	int start, n, k;
	int sort[100001];
	while (cin >> start >> n >> k)
	{
		map<int, int> mon;
		for (int i = 0; i < n; i++)
		{
			int addr, value, next;
			cin >> addr >> value >> next;
			sort[addr] = next;
			mon.insert(pair<int, int>(addr, value));
		}
		int p = start;
		stack<int> s, q;//之前的想法中,每满足K个,就输出栈内的元素和下一个元素。但是不对,最后剩余不足K时,需要正序输出。
		//其输出的第一个元素和最后一个栈满输出的最后一个不同。考虑不先输出,先进入队列。然后一次性输出。
		queue<int> t;
		if (p == -1)
		{
			printf("-1\n");
			continue;
		}
		while (p != -1)
		{
			s.push(p);
			if (s.size() == k)
			{
				while (!s.empty())
				{
					t.push(s.top());
					s.pop();
				}
			}
			p = sort[p];
		}
		while (!s.empty())
		{
			q.push(s.top());
			s.pop();
		}
		while (!q.empty())
		{
			t.push(q.top());
			q.pop();
		}
		while (t.size() != 1)
		{
			printf("%05d %d ", t.front(), mon[t.front()]);
			t.pop();
			printf("%05d\n", t.front());
		}
		printf("%05d %d %d\n", t.front(), mon[t.front()], -1);
	}

	return 0;
}

点赞