比较简单的一道寻找中位数的题,只要不断取两个队列中头较小的一个,直到取到计算中位数所需的一位或两位即可。
public double FindMedianSortedArrays(int[] nums1, int[] nums2)
{
int len = nums1.Length + nums2.Length;
int mid;
bool isEven;
if(len % 2 == 0)//判断中间位置与奇偶数
{
isEven = true;
mid = len / 2 - 1;
}
else
{
isEven = false;
mid = len / 2;
}
int i = 0, j = 0;
double r = 0;
for (int k = 0; k <= mid; k++)//不断取队列头较小的一个直到中间位置
{
r = FindNext(ref i, ref j, nums1, nums2);
}
if(isEven)//若是偶数则计算中位数还需一位
{
return (r + FindNext(ref i, ref j, nums1, nums2)) / 2;
}
else
{
return r;
}
}
double FindNext(ref int i, ref int j, int[] nums1, int[] nums2)
{
if(nums1.Length == 0)//如果一个是空的,就不断取另一个的队列头
{
return nums2[j++];
}
if(nums2.Length == 0)
{
return nums1[i++];
}
if (i < nums1.Length && j < nums2.Length)//取较小的一个
{
if(nums1[i] < nums2[j])
{
return nums1[i++];
}
else
{
return nums2[j++];
}
}
else if(i < nums1.Length)//如果一个取完了,就取另一个
{
return nums1[i++];
}
else
{
return nums2[j++];
}
}
哈哈写了好多if啊,暴露了菜鸡本质了。。。