合并--链表

最新推荐文章于 2024-09-13 22:22:04 发布

HPL__001

最新推荐文章于 2024-09-13 22:22:04 发布

阅读量168

点赞数

分类专栏：刷刷刷题文章标签：链表数据结构

本文链接：https://blog.csdn.net/HPL__001/article/details/129348562

版权

刷刷刷题专栏收录该内容

52 篇文章 0 订阅

订阅专栏

链表定义

struct ListNode {
    int val;
    ListNode *next;
    ListNode() : val(0), next(nullptr) {}
    ListNode(int x) : val(x), next(nullptr) {}
    ListNode(int x, ListNode *next) : val(x), next(next) {}
};

合并两个链表

ListNode* mergeTwoLists(ListNode *list1, ListNode *list2) {
    if ((!list1) || (!list2)) 
    	return list1 ? list1 : list2;
   
    ListNode *head = new ListNode;
    ListNode *temp = head, *l1 = list1, *l2= list2;
	while (l1 && l2) {
        if (l1->val < l2->val) {
            temp->next = l1; 
            l1= l1->next;
        } 
        else {
            temp->next = l2; 
            l2= l2->next;
        }
        temp = temp->next;
    }
    temp->next = (l1? l1: l2);
    
    return head->next;
}

合并n个链表

ListNode* mergeTwoLists(ListNode *list1, ListNode *list2) {
    if ((!list1) || (!list2)) 
    	return list1 ? list1 : list2;
   
    ListNode *head = new ListNode;
    ListNode *temp = head, *l1 = list1, *l2= list2;
	while (l1 && l2) {
        if (l1->val < l2->val) {
            temp->next = l1; 
            l1= l1->next;
        } 
        else {
            temp->next = l2; 
            l2= l2->next;
        }
        temp = temp->next;
    }
    temp->next = (l1? l1: l2);
    
    return head->next;
}

ListNode* mergeKLists(vector<ListNode*>& lists) {
    ListNode *res = nullptr;
    for (int i = 0; i < lists.size(); ++i) {
        res = mergeTwoLists(res, lists[i]);
    }
    return res;
}