给定一个常数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是下一结点的地址。
输出描述:
对每个测试用例,顺序输出反转后的链表,其上每个结点占一行,格式与输入相同。
示例1
输入
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
下面是我的代码QAQ
import java.util.HashMap;
import java.util.Scanner;
public class Hw_04_ReverseLink {
public static class Node {
private String addr;
private int data;
private String next;
public Node(String addr, int data, String next) {
this.addr = addr;
this.data = data;
this.next = next;
}
public Node(String addr) {
this.addr = addr;
}
public void setNext(String next) {
this.next = next;
}
}
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
String firstAddr = sc.next();
int N = sc.nextInt();
int K = sc.nextInt();
int i = 0;
HashMap<String, Node> map = new HashMap<String, Node>(); // 地址-node的映射
while (i < N) {
String addr = sc.next();
int data = sc.nextInt();
String next = sc.next();
Node node = new Node(addr, data, next);
map.put(addr, node);
i++;
}
// printNode(map, map.get(firstAddr));
//第一个节点是map.get(firstAddr),某个节点node的下一个节点是map.get(node.next)
//更改某个节点的下一个节点:map.get(node.addr).setNext(node2);
//每K个反转链表
if (N < K) {
printNode(map, map.get(firstAddr));
}
String firstAddr2 = firstAddr;
for (int j = 0; j < N / K; j++) {
Node cur = map.get(firstAddr2);
Node pre = new Node("-1"); // addr == -1表示空地址
Node next = new Node("-1");
Node node = cur; // 保存第一个节点,等饭转完之后链接到下一个反转的开始
i = 0;
while (i < K) {
next = map.get(cur.next);
map.get(cur.addr).setNext(pre.addr);
pre = cur;
cur = next;
i++;
}
if (j == 0) {
firstAddr = pre.addr;
}
// 结束后,cur为下一个反转的开始节点
map.get(node.addr).setNext(cur.addr);
firstAddr2 = cur.addr;
}
printNode(map, map.get(firstAddr));
}
public static void printNode(HashMap<String, Node> map, Node head) {
while (!head.next.equals("-1")) {
System.out.println(head.addr + " " + head.data + " " + head.next);
head = map.get(head.next);
}
System.out.println(head.addr + " " + head.data + " " + head.next);
}
}