PAT 反转链表

反转链表

题目描述：
给定一个常数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 <stdio.h>

struct node
{
	int add;
	int data;
	int next;
}arr[100000],brr[100000],crr[100000];

int main()
{
	int i,j,A = 0,N = 0,k = 0,index = 0;
	scanf("%d%d%d",&A,&N,&k);
	for(i = 0;i<N;i++)
	{
		scanf("%d",&index);
		scanf("%d%d",&arr[index].data,&arr[index].next);
	}
	int cnt = 0;
	for(i = A;i!=-1;i=arr[i].next)
	{
		brr[cnt].add = i;
		brr[cnt].data = arr[i].data;
		brr[cnt].next = arr[i].next;
		cnt++;
	}
	int x = cnt/k,y = cnt%k;
	int pos = 0;
	for(i = 1;i<=x;i++)
	{
		for(j = i*k-1;j>=(i-1)*k;j--)
		{
			crr[pos].add = brr[j].add;
			crr[pos].data = brr[j].data;
			pos++;
		}
	}
	if(y!=0)
	{
		for(i = k*x;i<cnt;i++)
		{
			crr[pos].add = brr[i].add;
			crr[pos].data = brr[i].data;
			pos++;
		}
	}
	for(i = 0;i<pos-1;i++)
	{
		printf("%05d %d %05d\n",crr[i].add,crr[i].data,crr[i+1].add);
	}
	printf("%05d %d -1\n",crr[pos-1].add,crr[pos-1].data);
	return 0;
}