贪心算法-根据身高重建队列

假设有打乱顺序的一群人站成一个队列,数组 people 表示队列中一些人的属性(不一定按顺序)。每个 people[i] = [hi, ki] 表示第 i 个人的身高为 hi ,前面 正好ki 个身高大于或等于 hi 的人。

请你重新构造并返回输入数组 people 所表示的队列。返回的队列应该格式化为数组 queue ,其中 queue[j] = [hj, kj] 是队列中第 j 个人的属性(queue[0] 是排在队列前面的人)。

示例 1:

输入:people = [[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]]
输出:[[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]]
解释:
编号为 0 的人身高为 5 ,没有身高更高或者相同的人排在他前面。
编号为 1 的人身高为 7 ,没有身高更高或者相同的人排在他前面。
编号为 2 的人身高为 5 ,有 2 个身高更高或者相同的人排在他前面,即编号为 0 和 1 的人。
编号为 3 的人身高为 6 ,有 1 个身高更高或者相同的人排在他前面,即编号为 1 的人。
编号为 4 的人身高为 4 ,有 4 个身高更高或者相同的人排在他前面,即编号为 0、1、2、3 的人。
编号为 5 的人身高为 7 ,有 1 个身高更高或者相同的人排在他前面,即编号为 1 的人。
因此 [[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]] 是重新构造后的队列。

示例 2:

输入:people = [[6,0],[5,0],[4,0],[3,2],[2,2],[1,4]]
输出:[[4,0],[5,0],[2,2],[3,2],[1,4],[6,0]]

提示:

  • 1 <= people.length <= 2000
  • 0 <= hi <= 10^6
  • 0 <= ki < people.length
  • 题目数据确保队列可以被重建
分析

本题相当于将一个包含身高和前面比自己高的人数所组成的数组打乱了再给你,让你根据已有的信息来重新排序数组。对于这么一系列杂乱的数据,妄图直接将其还原是不可能的,我们必须找到一个切入点,逐步还原。通常会从最高的或者最矮的开始,因为中间部分没有任何突出的特点(唉!)。

可以设想一下极端情况,假如我们从最高的开始还原people[i] = [hi, ki],对于最高的那位,没有任何人比他更高,因此他的ki大概率为0,假如有人和他一样高,我们也顶多知道这几个高个子之间的顺序,而无从获取更多的信息,更别谈还原了。

假如我们从最矮的开始还原,由于没有人比他高,因此他前面有多少个高的,就代表他前面得给大哥们留下多少个位置。将这条规律引申开来,从矮到高,一个个找准自己的位置(对于已经被占用的位置需要分条件判断是否算高个人数)这样便能完整还原啦。

需要注意的是,在排序过程中,我们需要先按身高从小到大排序,对于身高相同的,需要按前方人数按从小到大排序,因为两个同样高的人不可能站在同一个位置,一定要先把两个人当中更靠前的先还原,因为后一个会受到前一个的影响。

vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
	if (people.size() <= 1) {
		return people;
	}
	//按身高从小到大排序
	sort(people.begin(), people.end(), [](const vector<int>& a, const vector<int>& b) {
		if (a[0] == b[0])
			return a[1] < b[1];
		return a[0] < b[0];
	});
	vector<vector<int>> retQueue(people.size());
	for (int i = 0; i < people.size(); i++) {
		int step = people[i][1];
		for (int j = 0; j < retQueue.size(); j++) {
			if (retQueue[j].size() <= 0) {//位置尚未占用
				if (step == 0) {//可以放入
					retQueue[j] = people[i];
				}
				step--;
			}
			else if (retQueue[j][0] == people[i][0]) {
				step--;
			}
		}
	}
	return retQueue;
}

你可能感兴趣的:(算法,贪心算法,算法,leetcode,c++)