[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 (≤105)、以及正整數 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 個節點的地址。
注意:
題目有巨坑,測試點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;
}