1025 反转链表

给定一个常数 K 以及一个单链表 L，请编写程序将 L 中每 K 个结点反转。例如：给定 L 为 $1\to 2\to 3\to 4\to 5\to 6$，K 为 3，则输出应该为 $3\to 2\to 1\to 6\to 5\to 4$；如果 K 为 4，则输出应该为 $4\to 3\to 2\to 1\to 5\to 6$，即最后不到 K 个元素不反转。

输入格式：

每个输入包含 1 个测试用例。每个测试用例第 1 行给出第 1 个结点的地址、结点总个数正整数 $N(\le 105)$、以及正整数 $K(\le 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 <stdio.h>

struct Node {
	int address;
	int value;
	int next;
};

Node arr[100005];
Node sequence[100005];

int main()
{
	int Position;
	int N, K;
	int tmp;

	scanf("%d %d %d", &Position, &N, &K);

	for (int i = 0; i < N; i++) {
		scanf("%d", &tmp);
		scanf("%d %d",&arr[tmp].value, &arr[tmp].next);
	}

	int count = 0;
	for (int i = 0;; i++) {
		sequence[i].address = Position;
		sequence[i].value = arr[Position].value;
		sequence[i].next = Position = arr[Position].next;
		count++;
		if (Position == -1)
			break;
	}
	N = count;
	int cycle = N / K;
	// 分情况考虑
	// 一个都不反转
	if (cycle == 0 || K == 1) {
		for (int i = 0; i < N; i++) {
			printf("%.5d ", sequence[i].address);
			printf("%d ", sequence[i].value);
			if (sequence[i].next != -1)
				printf("%.5d\n", sequence[i].next);
			else
				printf("%d\n", sequence[i].next);
		}
	}

	// 刚好全部反转
	else if (N == K) {
		for (int i = N - 1; i > 0; i--) {
			printf("%.5d ", sequence[i].address);
			printf("%d ", sequence[i].value);
			printf("%.5d\n", sequence[i - 1].address);
		}

		printf("%.5d ", sequence[0].address);
		printf("%d ", sequence[0].value);
		printf("-1\n");
	}

	// 一般情况，部分反转
	else {
		for (int j = 1; j <= cycle; j++) {
			for (int k = K * j - 1; k >= K * (j - 1); k--) {
				printf("%.5d ", sequence[k].address);
				printf("%d ", sequence[k].value);
				if (k > K * (j - 1))
					printf("%.5d\n", sequence[k - 1].address);
				else {
					if (j < cycle)
						printf("%.5d\n", sequence[K*(j + 1) - 1].address);
					else
						if (N%K == 0)
							printf("-1\n");
						else
							printf("%.5d\n", sequence[K*j].address);
				}
			}
		}

		for (int i = K * cycle; i < N; i++) {
			printf("%.5d ", sequence[i].address);
			printf("%d ", sequence[i].value);
			if (sequence[i].next != -1)
				printf("%.5d\n", sequence[i].next);
			else
				printf("%d\n", sequence[i].next);
		}
	}
}