C程序-PAT-1025 反轉鏈表

給定一個常數 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​5​​)、以及正整數 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

思路分析：定義一個結構體存儲數據，以下標爲地址存儲數據，然後將數據連續存儲，方便翻轉，翻轉之後指向下一個的地址不是正確指向，直接指向下一個結構體地址即可，輸出時單獨處理最後一個節點。

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

const int N=100000;

struct node
{
	int address;
	int data;
	int next;
}a[N],*b[N];

int main( ) 
{
	int i,j,n,k,head,address,next,data;
	
	scanf("%d %d %d", &head, &n, &k);
	for (i = 0; i < n; i++)
	{
		scanf("%d %d %d", &address, &data, &next);
		//以地址做爲下標存儲數據 
		a[address].address = address;
		a[address].data = data;
		a[address].next = next;
	}
	
	for (i = head,j=0; i != -1; i = a[i].next)//將數據連續存好 ，方便存儲 
		b[j++] = &a[i];
	for (i = 0; i <= j-k; i += k) //翻轉，每 K 個結點反轉
		reverse(b + i, b + i + k);//翻轉函數 
		
	for (i=0;i<j;i++)
	{
		if (i < j - 1)
			printf("%05d %d %05d\n", b[i]->address, b[i]->data, b[i+1]->address);
		else
			printf("%05d %d -1\n", b[i]->address, b[i]->data);//最後一個 
	}
	
	return 0;
}