编写程序实现将两个带头结点的单链表,合并为一个带头结点的单链表,简单代码
时间: 2024-04-30 10:20:48 浏览: 104
有头结点的单链表表的实现
以下是合并两个带头结点的单链表的简单代码实现:
```python
def merge_lists(head1, head2):
p1 = head1.next
p2 = head2.next
new_head = ListNode(-1)
p = new_head
while p1 and p2:
if p1.val < p2.val:
p.next = p1
p1 = p1.next
else:
p.next = p2
p2 = p2.next
p = p.next
p.next = p1 if p1 else p2
return new_head
```
上述代码中,我们将两个链表的头结点传入函数 `merge_lists()`,并创建一个新的带头结点的链表 `new_head`。然后,我们使用指针 `p1` 和 `p2` 分别指向两个链表的第一个节点。然后,我们创建一个指针 `p`,它指向新链表的头结点。
在循环中,我们比较 `p1` 和 `p2` 的值,将较小值的节点接到新链表的尾部,并将指针 `p1` 或 `p2` 向后移动一位,直到两个指针中至少有一个为空。
最后,我们将剩下的节点接到新链表的尾部,返回新链表的头结点 `new_head`。
阅读全文