PAT1025 反转链表

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

代码

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
#define MAX 100001
struct Node{
	int address;
	int data;
	int next;
};
Node nodes[MAX];//存放结点 
int main(){
	int first,n,k;
	cin>>first>>n>>k;
	for(int i=0;i<n;i++){//将节点按地址存放入数组 
		Node x;
		cin>>x.address>>x.data>>x.next;
		nodes[x.address]=x; 
	}
	
	vector<Node> v ;
	while(first!=-1){//根据数组地址取得结点 依次放入V中 
		v.push_back(nodes[first]);
		first=nodes[first].next;
	}
	
	vector<Node> :: iterator it = v.begin(); //定义迭代器
	for(int i = 0; i < v.size()/k; i++){
        reverse(it, it + k);	//用reverse函数进行反转
        it = it + k;
    }
	for(int i = 0; i < v.size()-1; i++){ 
	        printf("%05d %d %05d\n", v[i].address, v[i].data, v[i + 1].address); //将除最后一个节点的其他节点格式化打印
	    }
	    printf("%05d %d -1\n", v[v.size()-1].address, v[v.size()-1].data); //打印最后一个节点
	
	
	return 0;
}