23. Merge k Sorted Lists

23. Merge k Sorted Lists


Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

题目大意

合并k个已经排好序的序列。

解题思路

一开始没有考虑空间复杂度,将k个序列用分治法的思路进行合并,而合并两个序列依然是使用分治法,所以整个算法的时间复杂度为 O((logn)2)

ListNode *dividedList(vector<ListNode*> lists, int left, int right) {
  if (left >= right) return lists[left];
  int mid = (left + right) / 2;
  return mergeTwoLists(dividedList(lists, left, mid), dividedList(lists, mid+1, right));
}
ListNode *mergeKLists(vector<ListNode *> &lists) {
  int k = lists.size();
  if (k == 0) return NULL;
  return dividedList(lists, 0, k-1);
}

但是,这种算法没有考虑过空间复杂度,导致无法通过一个检测空间复杂度的测例。
于是,我考虑在合并的序列的时候不使用递归,以减少空间复杂度。

ListNode *mergeKLists(vector<ListNode *> &lists) {
  int k = lists.size();
  if (k == 0) return NULL;
  while (k-1) {
    int mid = (k+1) / 2;
    for (int i = 0; i < k / 2; i++) {
      lists[i] = mergeTwoLists(lists[i], lists[mid+i]);
    }
    k = mid;
  }
  return lists[0];
}

算法复杂度

O((logn)2)

代码实现

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
  ListNode *mergeKLists(vector<ListNode *> &lists) {
    int k = lists.size();
    if (k == 0) return NULL;
    while (k-1) {
      int mid = (k+1) / 2;
      for (int i = 0; i < k / 2; i++) {
        lists[i] = mergeTwoLists(lists[i], lists[i + mid]);
      }
      k = mid;
    }
    return lists[0];
  }
  ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
    if (l1 == NULL) return l2;
    if (l2 == NULL) return l1;
    if (l1->val < l2->val) {
      l1->next = mergeTwoLists(l1->next, l2);
      return l1;
    } else {
      l2->next = mergeTwoLists(l1, l2->next);
      return l2;
    }
  }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值