题目
Merge two sorted linked lists and return it as a new list. The new list should be made by splicing together the nodes of the first two lists.
Example:
Input: 1->2->4, 1->3->4
Output: 1->1->2->3->4->4
我的尝试
既然说了有序,那么肯定要利用有序的特性。我第一反应是利用遍历。
两个链表,两个指针解决,方法如下,在第一遍时,定义了一个默认node,但是没有设置head指针指向默认node,这样更改resnode的指针,最后没有办法返回最终结果的头指针了。所以需要设置head,head其实是临时指针,指向当前最小的元素。最终返回的是resnode的next节点,也就是排序后的头指针。
另外最后记得处理不同长度的情况。最开始以为自己写不出来,没有想到竟然写出来了。给自己?!!!
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
ListNode listNodeRes=new ListNode(0);
// 我缺少了头指针,链表都有头指针,这样res指针就一直到了最后,不是最终结果到头指针了
ListNode head=listNodeRes;
while (l1!=null&&l2!=null){
if (l1.val<l2.val){
head.next=l1;
head=head.next;
l1=l1.next;
}
else {
head.next=l2;
head=head.next;
l2=l2.next;
}
}
if (l1!=null){
head.next=l1;
}
if (l2!=null){
head.next=l2;
}
return listNodeRes.next;
}
}
另外,之所以感觉这样写不对,是因为始终感觉没有全部比较,于是测试一下如果两个链表不是有序的情况下,发现代码结果不正确,所以这种比较都是基于有序的情况下,下面是测试结果:
我这样比较是因为,如果l1的某个元素小于l2的某个元素,那么肯定小于其后的元素,所以可以免去比较,但是如果是无序的,就不行了。
优化解法
有个伙伴用递归解决,代码非常简洁,参考学习一下:
public ListNode mergeTwoLists(ListNode l1, ListNode l2){
if(l1 == null) return l2;
if(l2 == null) return l1;
if(l1.val < l2.val){
l1.next = mergeTwoLists(l1.next, l2);
return l1;
} else{
l2.next = mergeTwoLists(l1, l2.next);
return l2;
}
}