1. 程式人生 > >PAT Basic 1025. 反轉連結串列 (25)(C語言實現)

PAT Basic 1025. 反轉連結串列 (25)(C語言實現)

, CSDN內容暫時不更新(將來有計劃更新), 請前往連結檢視最新內容. 歡迎star 我的repo

題目

給定一個常數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。

  • 讀取每一個節點,儲存到一個結構或int[3]陣列;
  • 將節點“連結”成為連結串列,在陣列中排列,找到指向-1的節點說明連結串列已經結束;
  • 反轉連結串列。每K個一組,交換第i個和倒數第i個;
  • 列印列表。反轉時沒有(也不必)修改每個節點的next,輸出時應該相應輸出下一個節點的address,最後一個節點的next輸出-1。

程式碼

#include <stdio.h>
#include <stdlib.h>

#define SWAPNODE(A, B) {Node temp = A; A = B; B = temp;}

typedef struct node *Node;

struct node{
    int addr;
    int data;
    int next;
};

int main()
{
    int A, N, K;
    Node *nodes, *p;

    /* read */
    scanf("%d %d %d", &A, &N, &K);
    nodes = (Node*)malloc(N * sizeof(Node));
    for(int i = 0; i < N; i++) 
    {
        nodes[i] = (Node)malloc(sizeof(struct node));
        scanf("%d %d %d", &nodes[i]->addr, &nodes[i]->data, &nodes[i]->next);
    }

    /* link the list */
    for(int i = 0; i < N; i++)
    {
        for(int j = i; j < N; j++)
        {
            if(nodes[j]->addr == (i ? nodes[i - 1]->next : A))
            {
                SWAPNODE(nodes[i], nodes[j]);
                break;
            }
        }
        if(nodes[i]->next == -1)   /* there could be useless nodes */
            N = i + 1;
    }

    /* reverse the list */
    for(int i = 0; i < N / K; i++)
    {
        p = nodes + i * K;
        for(int j = 0; j < K / 2; j++)
        {
            SWAPNODE(p[j], p[K - j - 1]);
        }
    }

    /* print the list */
    for(int i = 0; i < N - 1; i++)
    {
        printf("%05d %d %05d\n", nodes[i]->addr, nodes[i]->data, nodes[i + 1]->addr);
    }
    printf("%05d %d -1\n", nodes[N - 1]->addr, nodes[N - 1]->data);

    return 0;
}