这题,一开始想复杂了,不需要专门写一个添加数据的方法。
题目给的方法以及参数是这样的:
public ListNode mergeTwoLists(ListNode list1, ListNode list2)
我们可以先声明一个节点head,这个节点的值是-1.这样就免去了判断要把list1和list2中的哪一个节点作为新的合并后的有序链表后的头节点,到最后,我们返回head.next就好了。
然后,我们就不停地迭代就行了。怎么做呢?很简单,我们弄一个循环,当list1和list2都不为空时,这个循环就不停地进行下去,一等到这两个中的某一个为空时,循环就会退出。
在循环内部,我们判断这两个“指针”的值哪个更小,我们把更小的那个给p.p最开始就是head,不停地迭代一直到这个链表的最后。我们把最小的给p,然后更小的指针和p都向前迭代,就行了。
循环退出后,最多只会有一个指针不为空,我们把这个不为空的指针直接给p就行了。
代码:
public ListNode mergeTwoLists(ListNode list1, ListNode list2) { ListNode head = new ListNode(-1); ListNode p = head; while (list1 != null && list2 != null) { if (list1.val < list2.val) { p.next = list1; list1 = list1.next; } else { p.next = list2; list2 = list2.next; } p = p.next; } p.next = list1 == null ? list2 : list1; return head.next; }
现在,我们来看看递归的解法:
public ListNode mergeTwoLists(ListNode list1, ListNode list2) { if (list1 == null) { return list2; } if (list2 == null) { return list1; } if (list1.val < list2.val) { list1.next = mergeTwoLists(list1.next, list2); return list1; } else { list2.next = mergeTwoLists(list1, list2.next); return list2; } }
你会发现这份代码很酷,而且一看就懂了。如果1的值比2的更小,我们就从1开始连接节点,连接的范围呢?1的next,2的全部,因为我们是以1为起始节点的。这里仔细想一下,以1开始,那我接下来要判断的自然就是1的下一个和2的全部,然后我直接return这个就好了。