C++ - PAT - 1025. 反转链表 (25)

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

思路：

可以用2个数组来存储    1个数组下标为当前数字的地址，用来记录数据，另一个记录下一个地址。由于地址，数据，下一地址要整体考虑，所以用结构体Struct来存储这三个数据更方便一些。就不用开那么多数组。用三个vector容器，1个用来存储输入的数据。1个将数据按照顺序存放，最后一个存放反转后的数据。反转后的数据有了，就可以更新每个数据的下一个结点的地址。最后遍历输出即可。需要注意的是，最后一个数据的地址要输出-1，而不是-00001。

#include<cstdio>
#include<vector>

using namespace std;

struct node{
	int add,data,next;
};

vector<node> orig(100000);	//用来存储输入的数据 (因为输入的时候要直接用orig[n]，所以要先开足够大的副本)
vector<node> list;		//用来存储排列好的数据(未反转)
vector<node> ans;		//用来存储翻转后的顺序 

int first,n,k;

int main(){
	
	scanf("%d%d%d",&first,&n,&k);	
	
	node temp;
	
	//数据输入的顺序 
	for(int i=0 ;i<n ;i++){  
		scanf("%d%d%d",&temp.add,&temp.data,&temp.next);
		orig[temp.add] = temp;
	}
	
	//更新数据的顺序 
	int x = first;
	while(x!=-1){
		list.push_back(orig[x]);
		x = orig[x].next;
	}
	

	//数据的个数(n和n2可能会不同) 
	int n2 = list.size();

	//更新反转后的顺序 
	int end = k-1;
	
	while(end<n2){
		for(int i=end ;i>end-k ;i--){
			ans.push_back(list[i]);
		}
		end += k;
	}
	for(int i=end-k+1 ;i<n2 ;i++){
		ans.push_back(list[i]);
	}
	
	for(int i=0 ;i<n2-1 ;i++){
		ans[i].next = ans[i+1].add;	//更新反转后的数据的next数据 
		printf("%05d %d %05d\n",ans[i].add,ans[i].data,ans[i].next );
	}
	printf("%05d %d -1\n",ans[n2-1].add,ans[n2-1].data);
	
	return 0;
}