PAT A1025 反转链表

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 个元素不反转。

#include <iostream>
#include <algorithm>
#include <iomanip>
using namespace std;
const int MAX = 100005;

struct Node {
	int address, data, next;
	int order;
}node[MAX];

bool cmp1(Node a, Node b) {
	return a.order < b.order;
}

bool cmp2(Node a, Node b) {
	return a.address < b.address;
}

int main() {
	int begin, N, K, add, i;
	int cir;
	//cin >> begin >> N >> K;//起始地址 数据项数 步长
	scanf_s("%d%d%d", &begin, &N, &K);
	//预处理
	for (i = 0; i < MAX; i++) {
		node[i].address = i;
		node[i].order = MAX;
	}
	//输入
	for (i = 0; i < N; i++) {
		//cin >> add;
		scanf_s("%d", &add);
		//node[add].address = add;
		//cin >> node[add].data >> node[add].next;
		scanf_s("%d%d", &node[add].data, &node[add].next);
	}
	//预处理
	node[begin].order = 1;
	int rank = 2;
	int p = node[begin].next;
	while (p != -1) {
		node[p].order = rank++;
		p = node[p].next;
	}
	sort(node, node + MAX, cmp1);
	测试
	//for (int i = 0; i < rank-1; i++) {
	//	cout << setw(5) << setfill('0') << node[i].address << " " << node[i].data << " " << setw(5) << setfill('0') << node[i].next << "***"<<node[i].order<<"***"<< endl;
	//}
	//开始反转
	cir = (rank - 1) / K;// N！=有效结点数
	i = 1;
	begin = node[K - 1].address;
	int temp;
	while (i<=cir) {
		int j = i * K - 1;
		if (i < cir) temp = node[(i+1)*K-1].address;
		else temp = node[j].next;
		while (j > (i - 1)*K) {
			//int fz = node[j - 1].address;
			node[j].next = node[j - 1].address;
			j--;
		}
		node[j].next = temp;
		i++;
	}
	//测试
	//for (i = 0; i < rank - 1; i++) {
	//	cout << node[i].address << " " << node[i].data << " " << node[i].next << endl;
	//}
	//输出
	p = begin;
	sort(node, node + MAX, cmp2);
	//测试
	//cout << node[p].address << " " << node[p].data << " " << node[p].next << endl;
	//p = node[begin].next;
	//cout << node[p].address << " " << node[p].data << " " << node[p].next << endl;
	while (node[p].next  != -1) {
		//cout << setw(5) << setfill('0') << node[p].address << " " << node[p].data << " " << setw(5) << setfill('0') << node[p].next<< endl;
		printf("%05d %d %05d\n", node[p].address, node[p].data, node[p].next);
		p = node[p].next;
	}
	//cout << setw(5) << setfill('0') << node[p].address << " " << node[p].data << " " << node[p].next;
	printf("%05d %d %d\n", node[p].address, node[p].data, node[p].next);
	return 0;
}

/*
测试用例
00100 6 00002
00000 4 99999
00100 1 12309
68237 6 -1
33218 3 00000
99999 5 68237
12309 2 33218
*/

代码中//测试仅为了检查每个操作后的结果是否正确

测试点1未通过是因为代码段最后附的测试用例，此类例子有错；
没有考虑有多组需要反转的情况，前几次和最后一次的next赋值情况不同。

测试点5运行超时，把cin，cout改成了scanf和printf。