数据流中的中位数

数据流中的中位数

如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流，使用GetMedian()方法获取当前读取数据的中位数。

import java.util.ArrayList;
import java.util.Collections;

public class DataStreamMedian {

	public static void main(String[] args) {
		// TODO Auto-generated method stub

		DataStreamMedianSolution solution = new DataStreamMedianSolution();

		solution.Insert(1);
		solution.Insert(6);
		solution.Insert(13);
		solution.Insert(18);
		solution.Insert(5);
		solution.Insert(6);
		solution.Insert(10);
		solution.Insert(8);
		solution.Insert(9);
		
		Double num = solution.GetMedian();
		System.out.print(num);
		
	}

}

class DataStreamMedianSolution {
	
	public ArrayList<Integer> arrayList = new ArrayList<>();

	public void Insert(Integer num) {
		arrayList.add(num);
	}

	public Double GetMedian() {
		if (arrayList.size() < 1) {
			return 0.0;
		}		
		Collections.sort(arrayList); //升序排序
		if (arrayList.size() % 2 == 1) { //size（）方法用于获取ArrayList中存储的对象的个数
			return Double.valueOf(arrayList.get(arrayList.size() / 2)); //Double.valueOf方法是把数字类型的字符串，转换成Double类型
		} else {
			Double num1 = Double.valueOf(arrayList.get(arrayList.size() / 2)); //java.util.ArrayList.get(int index) 方法会返回元素在此列表中的指定位置
			Double num2 = Double.valueOf(arrayList.get(arrayList.size() / 2 - 1));
			return (num1 + num2) / 2;
		}
	}
	
}