题意:求两个有序数组结合之后的中位数
分析:
全部排序:偷懒使用STL,使用sort进行了全部排序,实际写完发现只需要排一半就可以找到中位数。
进阶方法:首先确定要求的中位数是正中间的还是求两个中间数的平均数。可以将两个数组大小加起来除2判定。比较两个数组的第一个数字 ,像接链表似的,每次都将两个链表头中小的那个插入新链表,然后删除它,再接着比较链表的头。注意的是,当其中一个数组读完之后,直接一直读另一个数组就好了。
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
nums1.insert(nums1.begin(),nums2.begin(),nums2.end());
sort(nums1.begin(),nums1.end());
if(nums1.size()%2==1)
return nums1[nums1.size()/2];
else
return (nums1[nums1.size()/2]+nums1[nums1.size()/2-1])/2.0;
}
};
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int flag;
int size1=nums1.size();
int size2=nums2.size();
int n,pre,count=0,i=0,j=0;
double res;
n=(size1+size2)/2+1;
while(count<n){
pre=res;
if(i==size1)
res=nums2[j++];
else if(j==size2)
res=nums1[i++];
else if(nums1[i]<nums2[j])
res=nums1[i++];
else
res=nums2[j++];
count++;
}
if((nums1.size()+nums2.size())%2==1)
return res;
else
return (res+pre)/2.0;
}
};