1025 反转链表 （25 分）——链表

1025 反转链表 （25 分）

给定一个常数 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
作者: CHEN, Yue
单位: 浙江大学
时间限制: 400 ms
内存限制: 64 MB
代码长度限制: 16 KB

public class Main {
static int[] data; // data[n]表示地址为n的节点的数据
static int[] next; // next[data]表示数据为data的下一节点的地址
static int[] list; // 将节点数据按顺序存储在list里面
static BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
public static void main(String[] args) throws IOException {
	String[] tmp = in.readLine().split(" ");
	int N = Integer.valueOf(tmp[1]);
	int K = Integer.valueOf(tmp[2]);
	data = new int[100000];
	next = new int[100000];
	list = new int[100000];
	int nextAddress = Integer.valueOf(tmp[0]);
	for (int i = 0; i < N; i++) {
		tmp = in.readLine().split(" ");
		int addr = Integer.valueOf(tmp[0]);
		int value = Integer.valueOf(tmp[1]);
		int nex = Integer.valueOf(tmp[2]);
		data[addr] = value;
		next[addr] = nex;
	}
	int n = 0; // 记录链表中节点的个数，因为不确定所有节点都在链表中
	while (nextAddress != -1) {
		list[n++] = nextAddress;
		nextAddress = next[nextAddress];
	}
	int ff = 0; // 记录遍历到的位置，方便处理最后几个不足K个的情况
	for (int i = K-1; i < n; i = i + K) {
		for (int j = 0; j <= K-1; j++) {
			if (j != K-1) {
				System.out.println(String.format("%05d", list[i - j]) + " " + data[list[i - j]] + " " + String.format("%05d", list[i - j - 1]));
			} else {
				if((n%K)==0&&(i-j)==n-K) {
					System.out.println(String.format("%05d", list[i - j]) + " " + data[list[i - j]] + " -1");
				}else if (i + K <= n) {
					System.out.println(String.format("%05d", list[i - j]) + " " + data[list[i - j]] + " " +String.format("%05d", list[i + K]) );
				} else {
					System.out.println(String.format("%05d", list[i - j]) + " " + data[list[i - j]] + " " + String.format("%05d", list[i + 1]));
				}
			}
		}
		ff = i;

	}
	for (int i = ff + 1; i < n; i++) {
		if (i == n - 1) {
			System.out.println(String.format("%05d", list[i]) + " " + data[list[i]] + " -1");
		} else {
			System.out.println(String.format("%05d", list[i]) + " " + data[list[i]] + " " + String.format("%05d", list[i + 1]));
		}
	}
}
// 思路不对，没有按照节点顺序存储，所以反转的时候有点棘手。。。
// public static void main(String[] args) throws IOException {
// String[] tmp=in.readLine().split(" ");
// int N=Integer.valueOf(tmp[1]);
// int K=Integer.valueOf(tmp[2]);
// String nextAddress=tmp[0];
// HashMap<String, Node> map=new HashMap<>(); //用map存储节点的信息，map的key就是节点的地址
// for(int i=0;i<N;i++) {
// tmp=in.readLine().split(" ");
// Node node=new Node(tmp[1],tmp[2]);
// map.put(tmp[0], node);
// }
// int n=0; //记录子链表的序号
// String[] datas=new String[K]; //存储子链表的数据
// String[] addrs=new String[K]; //存储子链表的地址
// while(!nextAddress.equals("-1")) {
// n=n%K;
// Node no=map.get(nextAddress);
// datas[n]=no.Data;
// addrs[n]=nextAddress;
// n++;
// if(n==K) {
// for(int i=datas.length-1;i>=0;i--) {
// System.out.println(addrs[i]+" "+datas[i]+addrs[i-1]);
// }
// }
// }
// if(n!=K) {
// for(int i=0;i<n-1;i++) {
// System.out.println();
// }
// }
// }
// static class Node{
// String Data;
// String Next;
// public Node(String data, String next) {
// super();
// Data = data;
// Next = next;
// }
// }
}