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


思路和代码均参考了算法笔记一书

思路：先要想办法记住每个结点的顺序(序号从0开始)，按这个顺序把n个结点分成n/k个块。枚举每个块，注意每个块的最后一个结点的next的处理
然后对块分两种情况，判断当前块不是最后一个完整的块(一个完整的块有k个结点)，若不是，就输出下一个块的最后一个结点。
若是最后一个完整块，又分两种情况，判断n % k 是否等于 0 ，如果等于0，则说明该结点是链表中最后一个结点，直接输出-1。
不等于0，则说明这个结点的next是 下一个不完整块的 第一个结点；之后按顺序输出结点就行了。

#include<stdio.h>
#include<algorithm>
using namespace std;

struct Node{
	int address;
	int num;
	int next;
	int order;/*记录每个结点的序号*/
}node[100001];

bool cmp(Node a, Node b)/*将链表的结点按序号的大小排列*/
{
	return a.order < b.order;
}

int main(void)
{
	int first,n,k,address;
	scanf("%d%d%d",&first,&n,&k);
	for(int i = 0; i < 100001; i++)
	{
		node[i].order = 100001;/*无效结点*/
	}
	
	for(int i = 0; i < n; i++)
	{
		scanf("%d",&address);
		scanf("%d%d",&node[address].num,&node[address].next);
		node[address].address = address;
	}
	
	
	int p = first;/*记下第一个结点*/
	int count = 0;/*有效结点的数量*/
	while(p != -1)
	{
		node[p].order = count++;
		p = node[p].next;
	}
	sort(node, node + 100001, cmp);
	
	
	n = count;
	
	for(int i = 0; i < n/k; i++)/*把n个结点分成n/k个块，枚举每个块*/
	{
		for(int j = (i + 1)*k - 1; j > i*k; j--)/*每个块反向输出*/
		{
			printf("%05d %d %05d\n",node[j].address,node[j].num,node[j-1].address);
		}
		
		printf("%05d %d ",node[i*k].address,node[i*k].num);/*每个块的最后一个结点的输出情况*/
		
		if(i < n / k -1)/*当这个块不是最后一个完整块时，就指向下个块的最后一个结点*/
		{
			printf("%05d\n",node[(i + 2)*k - 1].address);
		}
		else/*是最后一块时*/
		{
			if(n % k == 0)/*恰好是最后一个结点*/
			{
				printf("-1\n");
			}
			else/*剩下的凑不成一个块的结点按顺序输出*/
			{
				printf("%05d\n",node[(i + 1)*k].address);
				for(int i = n / k * k; i < n; i++)
				{
					printf("%05d %d ",node[i].address,node[i].num);
					if(i < n - 1)
					{
						printf("%05d\n",node[i + 1].address);
					}
					else
					{
						printf("-1\n");
					}
				}
			}
		}
	}
	return 0;
}