1. 程式人生 > 其它 >[PAT乙級]1025 反轉連結串列 (思路+精簡程式碼)

[PAT乙級]1025 反轉連結串列 (思路+精簡程式碼)

技術標籤:PAT乙級(C語言)

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

思路:雖然連結串列是資料結構內容,但本題完全可以用陣列來進行相應操作。
先將 N 個節點按輸入順序存入L[ ]中,地址直接對應下標。隨後利用首地址add[0]遍歷連結串列,按連結串列順序將各節點地址存到add[ ]中,則 add[i] 將對應第 i+1 個節點的地址。

此後,從第k個節點 (對應add[k-1]) 開始進行迴圈操作,使其next變為 add[k-2],即上一節點的地址,以此類推。當一輪迴圈進行了 k-1 次修改後,會遇到一個邊界節點,它的上一個節點要麼不存在,要麼已經修改過,此時必須看是否能夠進行下一輪修改。詳細實現見程式碼第11至13行。

注意:
題目有巨坑,測試點5和6中輸入的節點可能是無效節點,如果沒有按連結串列順序遍歷連結串列就會遇到錯誤,迴圈控制條件為 N 也會導致錯誤。因此必須用 valid 來統計有效節點數,控制迴圈。由此,題目的 K ≤ N 隱含的意思還有,若 valid < K ≤ N,無需反轉。
測試點3對應K = 1的情況,此時也無需反轉。

#include <stdio.h>
int L[100000][2] = { 0 }, add[100001] = { 0 }, n, k, valid = 0;
int main() {
	scanf("%d %d %d", &add[0], &n, &k);	//add[0]儲存首地址
	for (int i = 0, temp; i < n; i++) {
		scanf("%d", &temp);
		scanf("%d %d", &L[temp][0], &L[temp][1]);	//地址直接作為下標,L[][0]為data,L[][1]為next
	}
	for (int i = 0; add[i] != -1; i++, valid++) add[i + 1] = L[add[i]][1];	//valid統計有效節點數,因為部分節點可能是無效的
	if (k > 1 && k <= valid) {	//k為1時不反轉,k大於valid也不能反轉
		for (int i = k - 1; i < valid; i += k)
			for (int j = i, cnt = 0; cnt < k; j--, cnt++)
				cnt != k - 1 ? (L[add[j]][1] = add[j - 1]) : (i + k < valid ? L[add[j]][1] = add[i + k] : L[add[j]][1] = add[i + 1]);
	}													//若i + k < valid,則還可進行下輪迴圈
	for (int i = 0, j = (k <= valid ? add[k - 1] : add[0]); i < valid; i++, j = L[j][1])
		L[j][1] == -1 ? printf("%05d %d -1\n", j, L[j][0]) : printf("%05d %d %05d\n", j, L[j][0], L[j][1]);
		//反轉過的話,首地址變為add[k-1];注意-1的輸出,不要按5位輸出
	return 0;
}