【JAVA】PAT 乙级 1025 反转链表(测试点5超时,牛客网通过)
题目链接
给定一个常数 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
用一个类node来储存每个节点的地址,信息和下一个节点的地址。在main方法里,用一个长度为100000的node数组Node来储存输入的各节点,下标为该节点的地址。长度为n+1的node数组r按链表顺序储存节点。根据题目要求,把链表中的节点的位置替换。
要注意输入的节点可能会是无效节点
JAVA又日常超时,同样还是牛客网上通过,PAT不通过。
读者如果有更好的解决方法欢迎在评论区留言讨论。
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
public class Main {
public static void main(String[] args) throws Exception {
BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
int[] hnk = stii(bf.readLine());
int h = hnk[0], n = hnk[1], k = hnk[2];
node[] Node = new node[100000];
for (int i = 0; i < n; i++) {
String[] adn = bf.readLine().split(" ");
Node[sti(adn[0])] = new node(adn[0], adn[1], adn[2]);
}
node tem = Node[h];
node[] r = new node[n + 1];
int count = 0;
while (tem != null) {
r[count] = tem;
count++;
if (tem.next.equals("-1")) {
break;
}
tem = Node[sti(tem.next)];
}
r[count] = new node("-1", "-1", "-1");
int m = count % k;
for (int i = 0; i < count - m; i += k) {
for (int j = 0; j < k / 2; j++) {
node t = r[j + i];
r[j + i] = r[k + i - 1 - j];
r[k + i - 1 - j] = t;
}
}
for (int i = 0; i < count; i++) {
out.println(r[i].address + " " + r[i].data + " " + r[i + 1].address);
}
out.flush();
}
static class node {
String address;
String data;
String next;
node() {
}
node(String a, String d, String n) {
address = a;
data = d;
next = n;
}
}
static int sti(String a) {
return Integer.parseInt(a);
}
static int[] stii(String a) {
String[] ra = a.split(" ");
int[] r = new int[ra.length];
for (int i = 0; i < r.length; i++) {
r[i] = sti(ra[i]);
}
return r;
}
}