问题描述
有n辆自行车依次来到停车棚,除了第一辆自行车外,每辆自行车都会恰好停放在已经在停车棚里的某辆自行车的左边或右边。(e.g.停车棚里已经有3辆自行车,从左到右编号为:3,5,1。现在编号为2的第4辆自行车要停在5号自行车的左边,所以现在停车棚里的自行车编号是:3,2,5,1)。给定n辆自行车的停放情况,按顺序输出最后停车棚里的自行车编号。
输入格式
第一行一个整数n。
第二行一个整数x。表示第一辆自行车的编号。
以下n-1行,每行3个整数x,y,z。
z=0时,表示编号为x的自行车恰停放在编号为y的自行车的左边
z=1时,表示编号为x的自行车恰停放在编号为y的自行车的右边
输出格式
从左到右输出停车棚里的自行车编号
样例输入
4
3
1 3 1
2 1 0
5 2 1
样例输出
3 2 5 1
数据规模和约定
n<=100000
自行车编号为不超过100000的正整数。
算法分析
其实这道题主要考察的功能为 双链表的增加节点与打印节点。
1、捋一捋x、y、z之间的关系,定义节点
这道题需要用到双链表的数据结构,那么涉及到双链表我们就需要知道在这道题目中节点的左节点和右节点都是谁。每次每行的三个输入变量x,y,z。
x为待停放的自行车的编号,y自行车是x自行车想要的邻居,z决定x是在y的左边还是右边。节点我们用一个类来实现,并定义了三种构造方法。
public static class Node{
public int no;//节点编号
public Node left;
public Node right;
public Node(){};
public Node(Node left,int no,Node right){
this.left=left;
this.right=right;
this.no=no;
}
public Node(int no){
this.no=no;
this.left=new Node();
this.right=new Node();
}
}
2、存放节点的数据结构
由于每次停放自行车之前需要确定y的位置,如果我们单纯地用一个列表来存储这些节点,最后提交时会超时。所以采用哈希表来存放我们节点。
其中key为自行车的编号,value为整个节点。
这道题中,每次定义节点需要
HashMap<Integer,Node> map=new HashMap<Integer,Node>();
3、添加节点
在每次添加节点时会引起添加之后的两个相邻节点的部分节点变化,所以每次增加之后都要对节点的左右节点进行更新。
Node bikeY=map.get(y);
if(z==0)
{
Node add=new Node(bikeY.left,x,bikeY);
bikeY.left.right=add;
bikeY.left=add;
//每次添加完后检查该节点左节点是否存在,若不存在说明该节点为起始节点
if(add.left.no==0)
first=add;
map.put(x,add);
}
else{
Node add=new Node(bikeY,x,bikeY.right);
bikeY.right.left=add;
bikeY.right=add;
//每次添加完后检查该节点左节点是否存在,若不存在说明该节点为起始节点
if(add.left.no==0)
first=add;
map.put(x,add);
}
4、打印节点信息
在添加节点的过程中我们对于第一个节点的位置做了记录,题目要求从左到右输出,那么找到第一个节点后,获取它的right节点,如此循环打印即可。
最后,完整代码如下:
import java.util.*;
public class 自行车停放 {
public static class Node{
public int no;
public Node left;
public Node right;
public Node(){};
public Node(Node left,int no,Node right){
this.left=left;
this.right=right;
this.no=no;
}
public Node(int no){
this.no=no;
this.left=new Node();
this.right=new Node();
}
}
public static void main(String[] args){
Scanner sc=new Scanner(System.in);
HashMap<Integer,Node> map=new HashMap<Integer,Node>();
int n=sc.nextInt();//要停放的自行车数量
int m=sc.nextInt();//第一辆车的编号
Node first=new Node(m);
map.put(m,first);
for (int i=0;i<n-1;i++)
{
int x=sc.nextInt();//待放入车棚的自行车编号
int y=sc.nextInt();//停放的自行车(参照物)
int z=sc.nextInt();//0为左,1为右
Node bikeY=map.get(y);//获取参照物节点
if(z==0){
Node add=new Node(bikeY.left,x,bikeY);
bikeY.left.right=add;
bikeY.left=add;
if(add.left.no==0)
first=add;
map.put(x,add);
}
else{
Node add=new Node(bikeY,x,bikeY.right);
bikeY.right.left=add;
bikeY.right=add;
if(add.left.no==0)
first=add;
map.put(x,add);
}
}
Node out=first;
while (true) {
System.out.print(out.no+" ");
if(out.right.no!=0)
out=out.right;
else
break;
}
}
}