1025. 反轉連結串列 (25)

時間限制 300 ms
記憶體限制 65536 kB
程式碼長度限制 8000 B
判題程式 Standard 作者 CHEN, Yue

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

/*

 * 分析：輸入樣例正確連線順序應該是：

/*
00100 1 12309
12309 2 33218
33218 3 00000
00000 4 99999
99999 5 68237
68237 6 -1

輸入樣例中有不在連結串列中的結點的情況。所以用個sum計數

在輸入的時候就處理可以優化演算法
 */
import java.util.ArrayList;

import java.util.List;
import java.util.Scanner;

public class num1025 {

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int firstAddress = in.nextInt();
        int n = in.nextInt();
        int k = in.nextInt();
        Node[] list = new Node[100000];
        for (int i = 0; i < n; i++) {
            Node temp = new Node(in.nextInt(), in.nextInt(), in.nextInt());
            list[temp.address] = temp;//node數組裡面，第地址位[temp.address]個元素是當前元素
        }
        in.close();

        List<Node> array = new ArrayList<>();//list其實是Node陣列，array是List集合。。。。。
        link(list, firstAddress, array);

        reverse(array, k);

        for (int i = 0; i < array.size() - 1; i++) {
            System.out.printf("%05d %d %05d\n", array.get(i).address, array.get(i).data, array.get(i + 1).address);
        }
        int end = array.size() - 1;
        System.out.printf("%05d %d -1", array.get(end).address, array.get(end).data);
    }
    //c++好像直接有這個api
    public static void reverse(List<Node> array, int k) {//反轉
        for (int i = 0; i + k <= array.size(); i += k) {
            for (int m = i + k - 1, n = i; m >= n; m--, n++) {
                Node temp = array.get(n);        //一個簡單的換值操作，兩個迴圈比較複雜
                array.set(n, array.get(m));//set(int index, E element)
                                              //用指定的元素替代此列表中指定位置上的元素。
                array.set(m, temp);
            }
        }
    }

    public static void link(Node[] list, int firstAddress, List<Node> array) {
        while (firstAddress != -1) {
            array.add(list[firstAddress]);
            firstAddress = list[firstAddress].next;//這裡直接就是Node物件的next屬性，不是list的api
        }
    }
    
}
class Node {
    int address;
    int data;
    int next;

    Node(int address, int data, int next) {
        this.address = address;
        this.data = data;
        this.next = next;
    }
}