PAT 1025反转链表 JAVA实现(25分)
最近在刷pat乙级的题目,用的是java。尽管尝试了很多次但是这题也只拿了21分,希望有大神能指导一下。也希望能给大家带来一点帮助。(以下是题目)
给定一个常数 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
import java.lang.reflect.Array;
import java.util.ArrayList;
import java.util.Arrays;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
//1025
public class Main{
public static void main(String args[]) throws IOException {
InputStreamReader ir=new InputStreamReader(System.in);
BufferedReader rd=new BufferedReader(ir);
String one[]=rd.readLine().split(" ");
String first=one[0];
int N=Integer.parseInt(one[1]);
int K=Integer.parseInt(one[2]);
Node node[]=new Node[N];
int i=0;
while(N>0) {
String ne[]=rd.readLine().split(" ");
node[i]=new Node(ne[0],ne[1],ne[2]);
N--;i++;
}
rd.close();
ArrayList<Node> list=new ArrayList<Node>();
while(!first.equals("-1")) {
for(int j=0;j<node.length;j++) {
if(node[j].Address.equals(first)) {
first=node[j].Next;
list.add(node[j]);
break;
}
}
}
if(K==1){//K为1的情况
for(int j=0;j<list.size();j++) {
System.out.println(list.get(j).Address+" "+list.get(j).Data+" "+list.get(j).Next);
}
}
else{//K大于1,反转链表
ArrayList<Node> newNode=sort(list,K,node.length);
for(int j=0;j<newNode.size();j++) {
System.out.println(newNode.get(j).Address+" "+newNode.get(j).Data+" "+newNode.get(j).Next);
}
}
}
public static ArrayList<Node> sort(ArrayList<Node>node,int K,int N) {
ArrayList<ArrayList<Node>> Noder=new ArrayList<ArrayList<Node>>();
ArrayList<Node> node3=new ArrayList<Node>();
int i=N-N%K;
int num=0;
while(i>1) {
ArrayList<Node> node2=new ArrayList<Node>();
int l=i;
if(N-l<K){
for(;i>l-K;i--) {
Node n=node.get(i-1);
if(i>l-K+1) {
n.Next=node.get(i-2).Address;}
else {
if(l==N){
n.Next="-1";}
else{
n.Next=node.get(i+K-1).Address;
}
}
node2.add(n);
}
}else{
for(;i>l-K;i--) {
Node n=node.get(i-1);
if(i>l-K+1) {
n.Next=node.get(i-2).Address;}
else {
if(i+K-1<N){
n.Next=Noder.get(num++).get(0).Address;}
else{
n.Next="-1";
}
}
node2.add(n);
}
}
Noder.add(node2);
}
for(int o=Noder.size()-1;o>=0;o--){
for(int q=0;q<Noder.get(o).size();q++){
node3.add(Noder.get(o).get(q));
}
}
for(int k=N-N%K;k<N;k++) {
Node n=node.get(k);
node3.add(n);
}
return node3;
}
}
class Node{
String Address;//结点地址
String Data;//数据
String Next;//下个结点的地址
Node(String Address,String Data,String Next){
this.Address=Address;
this.Data=Data;
this.Next=Next;
}
}