如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流,使用GetMedian()方法获取当前读取数据的中位数。
class Solution {
public:
vector<int> result;
void Insert(int num)
{
result.push_back(num);
}
double GetMedian()
{
double mid;
sort(result.begin(),result.end());
int len = result.size();
if(len <= 0)
return 0;
if(len % 2 == 0)
{
mid = ((double)result[len/2-1] + (double)result[len/2]) / 2;
}
if(len % 2 == 1)
{
mid = (double)result[len/2];
}
return mid;
}
};
思路:其实第一次编译的时候,结果是一个数组,我觉得很奇怪,结果不是应该是只有一个数吗?最后发现这道题其实是一个数组,数组的第一个数找一次中位数,然后前两位数找一次中位数,前3位数找一次,以此类推。
注意事项:这道题的GetMedian()函数中,他的返回类型是double,过程中需要做类型转换,转为double型。