21.合并两个有序链表

题目描述

将两个有序链表合并为一个新的有序链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。

示例:

输入:1->2->4, 1->3->4
输出:1->1->2->3->4->4

思路

1.这道题思想和归并排序的merge函数是相似的,每次分别从两个容器中取出一个元素比较,将较小的放入另一个容器。
2.最后注意处理链表长度不一的情况就可以了。

Java代码实现

      public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        ListNode head = new ListNode(0);
        ListNode p = head;

        while(l1 != null && l2 != null){
            if(l1.val > l2.val){
                p.next = new ListNode(l1.val);
                l1 = l1.next;
            }else{
                p.next = new ListNode(l2.val);
                l2 = l2.next;
            }
            p = p.next;
        }
        //l1还存在元素
        while(l1 != null){
            p.next = new ListNode(l1.val);
            p = p.next;
            l1 = l1.next;
        }
        //l2还存在元素
        while(l2 != null){
            p.next = new ListNode(l2.val);
            p = p.next;
            l2 = l2.next;
        }

        return head.next;
    }

Golang代码实现

func mergeTwoLists(l1 *ListNode, l2 *ListNode) *ListNode {
    res := new(ListNode)
    p := res

    for l1 != nil && l2 != nil {
        if l1.Val < l2.Val{
            res.Next = &ListNode{Val:l1.Val,Next:nil}
            l1 = l1.Next
        }else{
            res.Next = &ListNode{Val:l2.Val,Next:nil}
            l2 = l2.Next
        }
        res = res.Next
    }

    for l1 != nil {
        res.Next = &ListNode{Val:l1.Val,Next:nil}
        l1 = l1.Next
        res = res.Next
    }

    for l2 != nil{
        res.Next = &ListNode{Val:l2.Val,Next:nil}
        l2 = l2.Next
        res = res.Next
    }

    return p.Next
}
全部评论

相关推荐

点赞 评论 收藏
转发
点赞 收藏 评论
分享
牛客网
牛客企业服务