第四题:(终于自己写了道)数组比较问题
(最快的是16ms,但是我提交一模一样的代码却是32ms 我的是16s 嘻嘻)
1.边界情况:
其中一个数组为空或两个都为空。
2.思路:
题目给的两个数组已经排好序了,所以只需要定义两个变量分别对应两个数组的相对头部位置,每次比较得到第i大的数(i是两个数组组合起来的数组的相对位置),当i到达组合数组中间位置(即两个数组长度除2,当然要考虑奇偶),然后就得到了中位数。(奇偶判断在代码里就不描述了)
算法代码
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int n = nums1.size(),m = nums2.size();
double mid1 = 0, mid2 = 0;
if (n == 0){
if ((n + m) % 2 == 0){
mid1 = nums2[nums2.size() / 2 - 1];
mid2 = nums2[nums2.size() / 2];
}
else{
mid1 = nums2[nums2.size() / 2];
mid2 = mid1;
}
return (mid1 + mid2) / 2;
}
else if (m == 0){
if ((n + m) % 2 == 0){
mid1 = nums1[nums1.size() / 2 - 1];
mid2 = nums1[nums1.size() / 2 ];
}
else{
mid1 = nums1[nums1.size() / 2];
mid2 = mid1;
}
return (mid1 + mid2) / 2;
}
int bite = 0,i = 0,j = 0;
while (bite != (n + m) / 2 + 1){
mid1 = mid2;
if (i == n){
mid2 = nums2[j];
j++;
}
else if (j == m){
mid2 = nums1[i];
i++;
}
else{
if (nums1[i] > nums2[j]){
mid2 = nums2[j];
j++;
}
else{
mid2 = nums1[i];
i++;
}
}
bite++;
}
if ((n + m) % 2 == 0)
return (mid1 + mid2) / 2;
else
return mid2;
}
-------wang
-------568551096@qq.com