剑指offer第二十九题

题目描述

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。

思路：
这道题目刚开始看到，我就想到了用最大堆来实现，那样的话时间复杂度是O(nlogK)，空间复杂度是K。但是后来想想，对实现起来可不简单呀，所以先用了冒泡排序，时间复杂度是O(n*K)。后来为了复习一下堆排序，我有实现了一下堆的思路。代码都如下：

（1）对排序的算法，自己实现了一个K个节点的堆

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        		len = input.size();
		if (k > len) return result;

		data = new int[len + 1];
		for (int i = 0;i < len;i++)
		{
			data[i + 1] = input[i];
		}
		Heapify(k);
		//对k+1之后的数据，依次和堆中第一个元素比较，如果小于，则和第一个元素比较，然后进行下沉操作
		for (int i = k + 1;i <=input.size();i++)
		{
			if (data[i] < data[1])
			{
				swap(data[i], data[1]);
				shiftdown(1);
			}
		}
		//将堆中的数据按顺序弹出
		stack<int> tmp;
		while (len >= 1)
		{
			tmp.push(data[1]);
			swap(data[1], data[len]);
			len--;
			shiftdown(data[1]);
		}
		result.clear();
		while (!tmp.empty())
		{
			result.push_back(tmp.top());
			tmp.pop();
		}
		delete[] data;
		return result;
	}
public:
	//把输入的数组的K个数排列成堆，堆的大小为k
	void Heapify(int n)
	{
		len = n;
		int j = n / 2;
		while (j >= 1)
		{
			shiftdown(j);
			j--;
		}
	}


private:
	int len;
	//因为最大堆里的数据是从1开始计数的
	int* data;
	vector<int> result;
	//对序号为i的堆内元素进行沉降操作
	void shiftdown(int i)
	{
		while (2*i<=len)
		{
			//左孩子的索引
			int j = 2 * i;
			//找出孩子节点中比较大的数
			if (j+1 <= len&&data[j + 1] > data[j])
			{
				j = j + 1;
			}
			//如果当前节点比孩子节点最大值小，则交换该节点和孩子节点
			if (data[i] < data[j])
			{
				swap(data[i], data[j]);
				i = j;
			}
			else if (data[i]>=data[j])
			{
				break;
			}
		}
	}
	//对序号为i的元素进行上升操作
	void shiftup(int i)
	{
		while (i > 1 && data[i] > data[i / 2])
		{
			swap(data[i], data[i / 2]);
			i = i / 2;
		}
	}
};

超级复杂有木有！！！

（2）冒泡排序实现
冒泡比较简单，就不注释了。

class Solution {
public:
	vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
		vector<int> result;
		if (k > input.size())
		{
			return result;
		}
		for (int i = 0;i < k;i++)
		{
			for (int j = i + 1;j < input.size();j++)
			{
				if (input[j] < input[i])
					swap(input[j], input[i]);
			}
			result.push_back(input[i]);
		}
		return result;
	}
};

后面复习其他排序算法的时候，再添加，未完待续。。。。。。