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⁵)、以及正整数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

这道题提交了n次才终于通过，此题第一个点在于将给定的节点按顺序连成一个有序链表，这里为了节省时间所以采用以空间换

时间的方法，然后是根据k值翻转链表，这里边界条件很多，稍不小心就会错。最后一个大坑就是题里没说所有节点都有效，所

要在连接链表时就重新统计有效节点总数。

尝试使用了两种不同的翻转方式实现

#include <iostream>
using namespace std;
struct node{
	int naddress;
	int ndata;
	int nnext;
}nodes[100010];
int main(){
	int data[100010];
	int next[100010];
	int first;
	int N;
	int K;
	scanf("%d%d%d",&first,&N,&K);
	int ad,da,ne;
	for(int i=0;i<N;i++){
		scanf("%d%d%d",&ad,&da,&ne);
		data[ad]=da;
		next[ad]=ne;
	}
	int p=0;
	while(first!=-1){
		nodes[p].naddress=first;
		nodes[p].ndata=data[first];
		nodes[p].nnext=next[first];
		p++;
		first=next[first];
	}
	N=p;//重新计数有效节点数目 
	p=0;
	while(p+K<=N){
		p=p+K;
		int temp=p-1;
		for(int i=0;i<K-1;i++){
			printf("%05d %d %05d\n",nodes[temp].naddress,nodes[temp].ndata,nodes[temp-1].naddress);
			temp--;
		}
		//处理每一段的最后一个节点 
		if(p+K<=N){
			nodes[temp].nnext=nodes[p+K-1].naddress;
			printf("%05d %d %05d\n",nodes[temp].naddress,nodes[temp].ndata,nodes[temp].nnext);
		}
		else if(p==N){
			nodes[temp].nnext=-1;
			printf("%05d %d %d\n",nodes[temp].naddress,nodes[temp].ndata,nodes[temp].nnext);
		}
		else{
			nodes[temp].nnext=nodes[p].naddress;
			printf("%05d %d %05d\n",nodes[temp].naddress,nodes[temp].ndata,nodes[temp].nnext);
		}
	}
	//处理不够反转的节点 
	if(p!=N){
		while(p<N-1){
			printf("%05d %d %05d\n",nodes[p].naddress,nodes[p].ndata,nodes[p].nnext);
			p++;
		}
		printf("%05d %d %d\n",nodes[p].naddress,nodes[p].ndata,nodes[p].nnext);
	}
	return 0;
}

通过学习别人的方法，下面的翻转部分采用栈和队列来实现

#include <iostream>
#include <stack> 
#include <queue>
using namespace std;
struct node{
	int naddress;
	int ndata;
	int nnext;
}nodes[100010];
int main(){
	int data[100010];
	int next[100010];
	int first;
	int N;
	int K;
	scanf("%d%d%d",&first,&N,&K);
	int ad,da,ne;
	for(int i=0;i<N;i++){
		scanf("%d%d%d",&ad,&da,&ne);
		data[ad]=da;
		next[ad]=ne;
	}
	int p=0;
	while(first!=-1){
		nodes[p].naddress=first;
		nodes[p].ndata=data[first];
		nodes[p].nnext=next[first];
		p++;
		first=next[first];
	}
	N=p;//重新计数有效节点数目 
	stack<node>s;
	queue<node>q;
	p=0;
	int count=0;
	while(count<N/K){
		while(p<K){
			s.push(nodes[p+count*K]);
			p++;
		} 
		while(p>0){
			q.push(s.top());
			s.pop();
			p--;
		}
		count++;
	}
	int temp=N-N%K;
	while(temp<N){
		q.push(nodes[temp]);
		temp++;
	} 
	p=0;
	while(p<N-1){
		printf("%05d %d ",q.front().naddress,q.front().ndata); 
		q.pop();
		printf("%05d\n",q.front().naddress);
		p++;
	}
	q.front().nnext=-1;
	printf("%05d %d %d\n",q.front().naddress,q.front().ndata,q.front().nnext);
	return 0;
}