1025 反转链表 (25分)java解析

给定一个常数 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

NOTE:

1.用数组a存放值，数组next存放该地址的下一个地址，两个数组下标都是当前结点地址

2.从首地址开始连接链表，怎么连链表呢，只要找结点地址，用数组b存放地址，下标从0开始，该节点的下一个地址就是b[i+1]。

3.连上链表后（将链表顺序存放后），每k个结点反转，用ArrayList数组存储结点地址，按反转顺序存储

4.总之，这里只对结点的地址进行操作

5.最后输出，结点地址，根据地址下标输出该节点值，它的下一个结点地址就是ArrayList数组的下一个值。

 

代码：java会有一点超时

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int index = sc.nextInt();
        int n = sc.nextInt();
        int[] a = new int[100001];
        int[] next = new int[100001];
        int[] b = new int[100001];//数组b存放更新后的地址值
        int k = sc.nextInt();
        for(int i = 0; i<n; i++){
            int address = sc.nextInt();
            a[address] = sc.nextInt();
            next[address] = sc.nextInt();
        }
        int cnt = 0;//输入的结点数不一定都是有效的，所以还要重新计数
        while(index != -1){//输入的结点是打乱的，从首地址开始连接结点
            b[cnt++] = index;
            index = next[index];
        }
        List<Integer> list = new ArrayList<>();
        int p = k-1;int ans =  0;int q = 0;
        while(p >= 0 && ans <cnt){
            list.add(b[p]);ans++;
            if(p > q)p--;
            else if(p == q){
                if(ans + k <= cnt){p=ans+k-1;q += k;}
                else break;
            }
        }
        for(int i = ans; i<cnt; i++)list.add(b[i]);
        for(int i = 0; i<cnt-1; i++)
            System.out.printf("%05d %d %05d\n",list.get(i),a[list.get(i)],list.get(i+1));
        System.out.printf("%05d %d -1\n",list.get(cnt-1),a[list.get(cnt-1)]);

    }
}