4. Median of Two Sorted Arrays
There are two sorted arrays nums1 and nums2 of size m and n respectively.
Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).
You may assume nums1 and nums2 cannot be both empty.
Example 1:
nums1 = [1, 3]
nums2 = [2]
The median is 2.0
Example 2:
nums1 = [1, 2]
nums2 = [3, 4]
The median is (2 + 3)/2 = 2.5
解题思路
大致思路是,分别求两个序列的中位数,比较——若相等,则该中位数即为合并后序列的中位数;若不相等,可以去掉较短序列的一半(若短序列的中位数较大,去掉更大的部分,因为这一部分合并后必然在中位数的右边,反之亦然),而较长序列去掉对应的相同长度的一部分。
如此,问题规模越来越小,而终止条件为:
若较短序列的长度只有1或者2,直接插入较长序列,然后求中位数即可。
程序代码
void insertX(vector <int> & nums1, int a, int b, int value){
if (b == (int)nums1.size() - 1) nums1.push_back(1);
int j = b;
while (j >= a && nums1[j] > value) { nums1[j + 1] = nums1[j]; j--; }
nums1[j + 1] = value;
}
double get_m(vector <int> & nums2, int a, int b){
if ((b - a) % 2 == 0) return nums2[(a + b) / 2];
return (nums2[(a + b) / 2] + nums2[(a + b) / 2 + 1]) * 1.0 / 2;
}
double fun(vector<int>& nums1, vector<int>& nums2, int a, int b, int c, int d){
//two elements
if (b - a == 1){
insertX(nums2, c, d, nums1[a]);
insertX(nums2, c, d + 1, nums1[a + 1]);
return get_m(nums2, c, d + 2);
}
//one element
if (b == a){
insertX(nums2, c, d, nums1[a]);
return get_m(nums2, c, d + 1);
}
double N1 = get_m(nums1, a, b), N2 = get_m(nums2, c, d);
if (N1 == N2) return N2;
else {
int Y = (b - a) / 2;
if (N1 > N2) return fun(nums1, nums2, a, b - Y, c + Y, d);
else return fun(nums1, nums2, a + Y, b, c, d - Y);
}
}
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
if (nums1.size() == 0 ) return get_m(nums2, 0, nums2.size() - 1);
if (nums2.size() == 0 ) return get_m(nums1, 0, nums1.size() - 1);
if (nums1.size() <= nums2.size())return fun(nums1, nums2, 0, (int)nums1.size() - 1, 0, (int)nums2.size() - 1);
return fun(nums2, nums1, 0, (int)nums2.size() - 1, 0, (int)nums1.size() - 1);
}
};
23. Merge k Sorted Lists
Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.
Example:
Input:
[
1->4->5,
1->3->4,
2->6
]
Output: 1->1->2->3->4->4->5->6
解题思路
法1:直接两两合并,效率低但是可以通过。
法2:每次取出最小的一个,使用尾插法连接即可。具体地,使用优先队列(小项堆)存储n个节点,每次取队头(即最小值),并往队列中插入该节点的下一个节点,遍历直到队列为空。
程序代码
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
//O(mlogn)
struct compare{
bool operator()(const ListNode* A, const ListNode * B){ return A->val > B->val; }
};
ListNode* mergeKLists(vector<ListNode*>& lists){
ListNode * head = new ListNode(0);
ListNode * r = head;
priority_queue<ListNode*, vector<ListNode*>, compare> pq;
for(int i = 0; i < lists.size(); i++) if (lists[i]) pq.push(lists[i]);
while(!pq.empty()){
ListNode * curMin = pq.top();
ListNode * nextNode = curMin->next;
pq.pop();
if (nextNode) pq.push(nextNode);
r->next = curMin;
r = curMin;
}
return head->next;
}
};