81. 数据流中位数
数字是不断进入数组的,在每次添加一个新的数进入数组的同时返回当前新数组的中位数。
样例
样例1
输入: [1,2,3,4,5]
输出: [1,1,2,2,3]
样例说明:
[1] 和 [1,2] 的中位数是 1.
[1,2,3] 和 [1,2,3,4] 的中位数是 2.
[1,2,3,4,5] 的中位数是 3.
样例2
输入: [4,5,1,3,2,6,0]
输出: [4,4,4,3,3,3,3]
样例说明:
[4], [4,5] 和 [4,5,1] 的中位数是 4.
[4,5,1,3], [4,5,1,3,2], [4,5,1,3,2,6] 和 [4,5,1,3,2,6,0] 的中位数是 3.
挑战
时间复杂度为O(nlogn)
说明
中位数的定义:
-
这里的中位数不等同于数学定义里的中位数。
-
中位数是排序后数组的中间值,如果有数组中有n个数,则中位数为A[(n-1)/2]A[(n−1)/2]。
-
比如:数组A=[1,2,3]的中位数是2,数组A=[1,19]的中位数是1。
public class Solution {
/**
* @param nums: A list of integers
* @return: the median of numbers
*/
public int[] medianII(int[] nums) {
int[] result = new int[nums.length];
for (int i = 0; i < nums.length; i++) {
result[i] = median(nums, i);
}
return result;
}
private int median(int[] nums, int i) {
for (int j = i; j > 0; j--) {
if (nums[j] < nums[j - 1]) {
int temp = nums[j];
nums[j] = nums[j - 1];
nums[j - 1] = temp;
}
}
return nums[i / 2];
}
}