《Cracking the Coding Interview程式設計師面試金典》----實時中位數

現有一些隨機生成的數字要將其依次傳入，請設計一個高效演算法，對於每次傳入一個數字後，算出當前所有傳入數字的中位數。(若傳入了偶數個數字則令中位數為第n/2小的數字，n為已傳入數字個數)。

給定一個int陣列A，為傳入的數字序列，同時給定序列大小n，請返回一個int陣列，代表每次傳入後的中位數。保證n小於等於1000。

[1,2,3,4,5,6],6

返回：[1,1,2,2,3,3]

思路：這道題應該使用兩個優先順序堆，左邊一個大堆，存放小於中位數的值，右邊一個小堆，用於存放大於中位數的值，每次插入都需要進行平衡。

程式碼如下:

#include<iostream>
#include<vector>
#include<map>
#include<queue>
using namespace std;

vector<int> getMiddle(vector<int> A, int n) {
	// write code here
	//插入排序
	vector<int> result;
	result.push_back(A[0]);
	for (int i = 1; i<n; i++)
	{
		int k = i;
		while (k>0)
		{
			if (A[k]<A[k - 1])
				swap(A[k], A[k - 1]);
			else
				break;
			k--;
		}
		result.push_back(A[i / 2]);
	}
	return result;
}
void printVector(vector<int>mat, int n)
{
	for (int j = 0; j < n; j++)
	{
		cout << mat[j] << " ";
	}
	cout << endl;
}

int main()
{
	vector<int> v;
	int m;
	int temp;
	while (cin >> m)
	{
		v.clear();
		for (int j = 0; j < m; j++)
		{
			cin >> temp;
			v.push_back(temp);
		}
		printVector(getMiddle(v, m), m);
	}
	return 0;
}

不懂的可以加我的QQ群：261035036（IT程式設計師面試寶典

群） 歡迎你的到來哦，看了博文給點腳印唄，謝謝啦~~