欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LeetCode21:合并两个有序链表 迭代解法

程序员文章站 2022-06-17 19:46:52
...

将两个有序链表合并为一个新的有序链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例:
输入:1->2->4, 1->3->4
输出:1->1->2->3->4->4

思路:

  • 设置一个虚拟的头指针head和一个迭代用的pre指针,pre永远指向新链表的最后一个节点
  • 通过不断向后迭代l1,l2,并且比较l1, l2的值的大小,决定pre->next该指向l1还是l2的节点,随后跟进pre指针
  • l1,l2任意一指针走到头,将pre->next指向另一指针

一趟迭代 ↓
LeetCode21:合并两个有序链表 迭代解法

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution
{
public:
    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2)
    {
        ListNode* head = new ListNode(114514);
        ListNode* pre = head;
        while(l1 && l2)
        {
            if(l1->val < l2->val)
            {
                pre->next = l1;
                pre = l1;
                l1 = l1->next;
            }
            else
            {
                pre->next = l2;
                pre = l2;
                l2 = l2->next;
            }
        }
        if(l1)
        {
            pre->next = l1;
        }
        else
        {
            pre->next = l2;
        }
        return head->next;
    }
};