PAT乙级1025

1025. 反转链表 (25)

时间限制

300 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

CHEN, Yue

给定一个常数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⁵)、以及正整数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

#include<stdio.h>
#include<algorithm>
#include<stack>
using namespace std;
struct Node
{
	int data;
	int next;
}node[100000];//定义静态链表节点
struct record
{
	int address;
	//int next;
};//用于反转链表
int main()
{
	for (int i = 0; i < 100000; i++)
	{
		node[i].data = 0;
		node[i].next = -1;
	}//初始化
	int head, N, K;
	scanf("%d%d%d", &head, &N, &K);
	int address, data, next;
	while (N--)
	{
		scanf("%d%d%d", &address, &data, &next);
		node[address].data = data;
		node[address].next = next;//链接各节点
	}
	int count = 0;//计数器
	int addressNow = head;
	int pre;
	stack<record> sta;//用于暂存要被逆转连接的节点
	record temprecord;//保存节点地址,本来以为要保存下一个节点的地址，其实根本不需要，也不想改了就这样吧
	int oldLast;//上一次逆转后指向最后一个节点的指针
	while (addressNow!= -1)
	{
		temprecord.address = addressNow;
		//temprecord.next = node[addressNow].next;
		sta.push(temprecord);
		pre = addressNow;
		addressNow = node[addressNow].next;
		count++;
		if (count%K == 0)
		{	
				if (count == K)
					head = pre;//第一个开始反转的节点成为头了
				if (!sta.empty())
				{
					sta.pop();
				}
				if (pre != head)
				{
					node[oldLast].next = pre;
				}//如果不是第一个那么需要将上一次反转后的最后一个节点的下个地址做修改
				//将样例中的K由4改为2你就知道为什么了
			for (int i = 0; i < K-1; i++)
			{
				node[pre].next = sta.top().address;//逆转链表
				pre = sta.top().address;
				if (!sta.empty())
				{
					sta.pop();
				}
			}
			node[pre].next = addressNow;//修改反转节点的最后一个节点的下一个地址，
			//若后面不会再反转，那么就是当前扫描的节点的地址，注意我是先计数后又把指针
			//往后移一位了
			oldLast = pre;//保存反转节点的最后一个节点的地址
		}
	}
	addressNow = head;
	while (addressNow!= -1)
	{
		if(node[addressNow].next!=-1)
		printf("%05d %d %05d\n", addressNow, node[addressNow].data, node[addressNow].next);
		else
			printf("%05d %d %d\n", addressNow, node[addressNow].data, node[addressNow].next);
		addressNow = node[addressNow].next;
	}
	return 0;
}
/*注意把每次反转后的节点的下一个地址修改正确，
比如说1 2 3 4 5 6，反转个数K为2时，应该是
2 1 4 3 6 5，这时要记得把1跟4连上，3跟6
连上。*/