PAT(乙级)1025 反转链表（最后两个测试点的问题）

给定一个常数 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

 

解析

        本题的坑在于最后两个测试点的链表上有游离的结点，这些结点是无效的，不能输出。本题一开始我的做法是正确的，超时后想了很久不得其解。看了看其他博主的做法，其实与我超时的做法是相同的（尽管我没能想到会有游离的结点 -_-），这就让我更迷惑了，开始从代码细节上下手。由于之前做题有被cin，cout卡超时的惨痛经历，而且从其他博主那里学习到了scanf的新用法，最后做了少许修改就AC了。大致说一下我的思考过程吧。

        首先说一下做法。接收所有结点组成链表然后遍历，将遍历过程中的结点地址存储到数组中（这样就排除了游离的结点），然后将数组中的数据按k个一组进行倒置，之后按数组中地址的顺序来输出结点。

        然后是数据结构。这题刚看到就没想用链表，用数组又觉得太占空间了，最后采用了map来储存这种“地址”—“结点”形式的数据。由于不知道scanf可以直接输入带前置0的int型数据，所以地址了采用string型。结点采用结构体。

        超时后我第一反应是有一些特例没判，加了好多特判然而还是超时。看博客明白了最后两个点的坑和做法，由于和我的做法一致，没找到解决办法。开始考虑代码细节问题，最后定位到了cin，cout的问题上。在另一个博主那里知道了scanf可以输入带前置0的int型数据，也就是说用scanf输入00100实质上等于输入了100。然后将所有string改成了int，所有cin cout改成了scanf printf。输出需要加前置0时使用%05d这样的形式。代码如下：

#include<bits/stdc++.h>
using namespace std;
struct node
{
	int data;
	int next;
};
int main()
{
	vector<int> l;	//地址序列 
	map<int,node> s; 
	int first,n,k,add;
	scanf("%d %d %d",&first,&n,&k);
	while(n--)
	{
		node t;
		cin>>add>>t.data>>t.next;
		s[add]=t;
	}
	while(first!=-1)	//遍历链表，将所有链上地址存到地址序列中 
	{
		l.push_back(first);
		first=s[first].next;
	}
	for(int i=0;i<(l.size()-l.size()%k);i+=k)	//将序列中的地址按k个一组进行倒置，末尾不足一组的不处理 
		reverse(l.begin()+i,l.begin()+i+k);
	for(int i=0;i<l.size();i++)
	{
		printf("%05d %d ",l[i],s[l[i]].data);
		if(i==l.size()-1)	//最后一个结点的next应为-1 
			printf("-1");
		else
			printf("%05d\n",l[i+1]);
	}
	return 0;
}