PAT (Basic Level) Practise （中文） 1025. 反转链表 (25)

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(<= 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

/*

 * 分析：输入样例正确连接顺序应该是：

/*

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()
4000
- 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;

    }

}