PAT (Basic Level) Practice 1025 反转链表 (25分) (链表的反转 使用栈 / reverse函数)

1.题目

给定一个常数 K 以及一个单链表 L，请编写程序将 L 中每 K 个结点反转。例如：给定 L 为 1→2→3→4→5→6，K 为 3，则输出应该为 3→2→1→6→5→4；如果 K 为 4，则输出应该为 4→3→2→1→5→6，即最后不到 K 个元素不反转。

输入格式：

每个输入包含 1 个测试用例。每个测试用例第 1 行给出第 1 个结点的地址、结点总个数正整数 N (≤10​5​​)、以及正整数 K (≤N)，即要求反转的子链结点的个数。结点的地址是 5 位非负整数，NULL 地址用 −1 表示。

接下来有 N 行，每行格式为：

Address Data Next

其中 Address 是结点地址，Data 是该结点保存的整数数据，Next 是下一结点的地址。

输出格式：

对每个测试用例，顺序输出反转后的链表，其上每个结点占一行，格式与输入相同。

输入样例：

00100 6 4
00000 4 99999
00100 1 12309
68237 6 -1
33218 3 00000
99999 5 68237
12309 2 33218

输出样例：

00000 4 33218
33218 3 12309
12309 2 00100
00100 1 99999
99999 5 68237
68237 6 -1

2.题目分析

1.首先可能存在无效节点，与其他节点无法连到一起，这种节点不能算，所以首先以next为线索存放数组，之后再将其按链表顺序存放在另外一个数组，同时计算有效节点数。

2.注意reverse函数的使用

3.代码

1.第一版较复杂

#include<iostream>
#include<stack>
#include<cstdio>

using namespace std;
 struct node
{
	int address;
	int data;
	int next;
}sorts[100011],list[100011];
int main()
{
	int start, n, k;
	cin >> start >> n >> k;
	int total = 0;
	int index = 0;
	for (int i = 0; i < n; i++)
	{
		cin >> index;
		cin >> list[index].data;
		cin>> list[index].next;
		list[index].address = index;
	}
	int count = 0;
	while (start!=-1)//找出合理的个数，可能有的连不上 对应测试点6
	{
		total++;
		sorts[count++] = list[start];
		start = list[start].next;
	}

	stack<node>run;
	int time = total / k;
	for (int s = 0; s < time;s++)
	{
		for (int i = s*k; i < s*k+k; i++)
			run.push(sorts[i]);
		int temp =run.top().next;
		while (1)
		{
			printf("%05d %d ", run.top().address, run.top().data);
			run.pop();
			if (!run.empty())
				printf("%05d
", run.top().address);
			else
			{
				if (temp >= 0)//注意还有0 也就是链表后面还有东西
				{
					if ((s + 2)*k <= total)//如果下一段个数还能支持反转，找到下一段的开始（例如1 2 3 4 k=2;输出2 1 后要在1后面输出4的地址，所以找4 ）
					{
						printf("%05d
", sorts[(s + 2)*k-1].address);
					}
					else//如果下一段个数不能支持反转，找到下一段的开始，输出地址后就break
					{
						printf("%05d
", sorts[(s + 1)*k].address);
						break;
					}
				}
			else
				printf("%d
", temp);//-1,表示到达链表末尾，正好total整除K
				break;
			}
		}
	}
	for (int i = time*k; i <time*k+total%k; i++)//未整除，输出剩下的
	{
		if(sorts[i].next>0)
		printf("%05d %d %05d
", sorts[i].address, sorts[i].data, sorts[i].next);
		else
			printf("%05d %d %d
", sorts[i].address, sorts[i].data, sorts[i].next);
	}

}

第二版

#include<iostream>
#include<algorithm>
using namespace std;
#define max 100011
struct node
{
	int address;
	int data;
	int next;
} sorts[max], list[max];
int main()
{
	int start, n, k;
	cin >> start >> n >> k;
	int temp;
	for (int i = 0; i < n; i++)
	{
		cin >> temp;
		cin >> list[temp].data;
		cin >> list[temp].next;
		list[temp].address = temp;
	}
	int total = 0;
	int count = 0;
	while (start != -1)
	{
		total++;
		sorts[count++] = list[start];
		start = list[start].next;
	}
	for (int i = 0; i < total / k; i++)
		reverse(sorts + i*k, sorts + i*k + k);
	int i = 0;
	for ( i = 0; i < total - 1; i++)
		printf("%05d %d %05d
", sorts[i].address, sorts[i].data, sorts[i + 1].address);
	printf("%05d %d -1
", sorts[i].address, sorts[i].data, sorts[i + 1].address);
}