21. 合并两个有序链表
将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例
自己初步想法
:
public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
ListNode root = new ListNode(0);
ListNode listNode = root;
while(l1 != null || l2 != null){
if(l1 == null){
listNode.next = l2;
l2 = l2.next;
}else if(l2 == null){
listNode.next = l1;
l1 = l1.next;
}else {
if(l1.val > l2.val){
listNode.next = l2;
l2 = l2.next;
}else {
listNode.next = l1;
l1 = l1.next;
}
}
listNode = listNode.next;
}
return root.next;
}
## 改进
public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
ListNode dummyHead = new ListNode(0);
ListNode cur = dummyHead;
while (l1 != null && l2 != null) {
if (l1.val < l2.val) {
cur.next = l1;
cur = cur.next;
l1 = l1.next;
} else {
cur.next = l2;
cur = cur.next;
l2 = l2.next;
}
}
if (l1 == null) {
cur.next = l2;
} else {
cur.next = l1;
}
return dummyHead.next;
}
减少了循环中的判断进行时间上的优化。
递归写法
public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
if (l1 == null) return l2;
if (l2 == null) return l1;
ListNode res = l1.val < l2.val ? l1 : l2;
res.next = mergeTwoLists(res.next, l1.val >= l2.val ? l1 : l2);
return res;
}
代码格式好看的