PAT甲级真题1074 反转链表(简单解法）

给定一个常数 KK 和一个单链表 LL，请你在单链表上每 KK 个元素做一次反转，并输出反转完成后的链表。

如果链表最后一部分不足 KK 个元素，则最后一部分不翻转。

例如，假设 LL 为 1→2→3→4→5→61→2→3→4→5→6，如果 K=3K=3，则你应该输出 3→2→1→6→5→43→2→1→6→5→4；如果 K=4K=4，则你应该输出 4→3→2→1→5→64→3→2→1→5→6。

补充
1、本题中可能包含不在链表中的节点，这些节点无需考虑。

输入格式
第一行包含头节点地址，总节点数量 NN 以及常数 KK。

节点地址用一个 55 位非负整数表示（可能有前导 00），NULL 用 −1−1 表示。

接下来 NN 行，每行描述一个节点的信息，格式如下：

Address Data Next
其中 Address 是节点地址，Data 是一个整数，Next 是下一个节点的地址。

输出格式
将重新排好序的链表，从头节点点开始，依次输出每个节点的信息，格式与输入相同。

数据范围
1≤N≤1051≤N≤105,
1≤K≤N1≤K≤N
输入样例：
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<bits/stdc++.h>
using namespace std;
const int N=100010;
int e[N],ne[N];
int main(){
	int h,n,k;
	cin>>h>>n>>k;
	while(n--){
		int address,data,next;
		cin>>address>>data>>next;
		e[address]=data;
		ne[address]=next;
	}
	vector<int>s;//存地址
	for(int i=h;i!=-1;i=ne[i]){
	    s.push_back(i);
	}
	for(int i=0;i+k-1<s.size();i+=k){ //每次反转k个，不足k个退出
		reverse(s.begin()+i,s.begin()+i+k);
	}
    for(int i=0;i<s.size();i++){
		printf("%05d %d ",s[i],e[s[i]]);
		if(i==s.size()-1)puts("-1");
		else printf("%05d\n",s[i+1]);
	}
	return 0;
}