问题描述:
给定两个链表(前提是链表有序),将两个链表中的值由小到大进行合并
算法思路:
①首先给定两个单链表(有序),我们将第一个单链表的头用headA表示,第二个单链表的头用headB表示,其次还要定义一个虚拟节点node来表示新链表的头(数据域可给为-1),定义一个tmp来进行移动(tmp刚开始和node相等)。之后将一个一个经过比较的值往tmp后依此串就可以,再打印新连接的链表就是排好序的
②注意循环条件:headA和headB都不为空。首先比较headA和headB的值,将小的一个节点给tmp.next,然后自己向后走一位,方便后一个再进行比较,tmp再自己向后走一步。
③两个链表有可能长度不同,所以循环跳出来还需要进行一下判断。一个链表走完,另一个链表还没有完,直接将没有走完的节点依此串在tmp的next域即可。
参考代码:
class ListNode{
public int data;
public ListNode next;
public ListNode(int data){
this.data=data;
this.next=null;
}
} //节点类
class MyList{
public ListNode head;
public MyList(){
this.head=null;
}
//尾插建立链表
public void addLast(int data){
ListNode node=new ListNode(data);
if(this.head==null){
this.head=node;
}else{
ListNode cur=this.head;
while(cur.next!=null){
cur=cur.next;
}
cur.next=node;
}
}
//打印链表
public void display(ListNode newhead){
ListNode cur=newhead;
while(cur!=null){
System.out.print(cur.data+" ");
cur=cur.next;
}
}
} //链表类
public class Test {
public static ListNode mergeTwoLists(ListNode headA,ListNode headB) {
ListNode node=new ListNode(-1);
ListNode tmp=node; //定义一个虚拟节点,之后依此连接由小到大尾插进来的节点
while(headA!=null && headB!=null){
if(headA.data<headB.data){ //比较数据的大小
tmp.next=headA;
headA=headA.next;
tmp=tmp.next;
}else{
tmp.next=headB;
headB=headB.next;
tmp=tmp.next;
}
}
if(headA!=null){ //此前头结点为headB的链表遍历完
tmp.next=headA;
}
if(headB!=null){ //此前头结点为headA的链表遍历完
tmp.next=headB; //直接将headB剩下的串入
}
return node.next;
}
public static void main(String[] args) {
MyList my1 = new MyList();
my1.addLast(1);
my1.addLast(6);
my1.addLast(8);
my1.addLast(26);
my1.addLast(90);
MyList my2 = new MyList();
my2.addLast(2);
my2.addLast(10);
my2.addLast(35);
my2.addLast(100);
ListNode node=mergeTwoLists(my1.head,my2.head);
my2.display(node);
}
}
//打印结果
1 2 6 8 10 26 35 90 100
如图所示:
新链表头节点为node