【算法与数据结构】406、LeetCode根据身高重建队列

所有的LeetCode题解索引，可以看这篇文章——【算法和数据结构】LeetCode题解。

一、题目

二、解法

  思路分析：本题难点在于如何对序列进行排序，以及保证序列的顺序（符合题目要求的顺序）。排序有两种一种是对K进行排序，排完之后发现队列仍然混乱，没有意义。按身高排序以后发现队列有序，同时我们按题目要求，K更大的元素方在后面。例如，people = [[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]]， 排序后为[[7,0], [7,1], [6,1], [5,0], [5,2]，[4,4]]，排序使用sort函数的重载。然后使用链表进行插入，链表插入的效率相较于vector容器高很多（参考代码随想录）。关于链表的插入可以看博主的文章【算法与数据结构】707、LeetCode设计链表。最后按题目的要求返回链表（不是链表头，是一个vector<vector>的嵌套vector容器）。
  程序如下：

class Solution {
public:
	static bool cmp(const vector<int>& a, const vector<int>& b) {
		if (a[0] == b[0]) return a[1] < b[1];	// 如果h相同比较k,k比较大的放后面，升序
		return a[0] > b[0];	// 按h排列，降序
	}
	//vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
	//	sort(people.begin(), people.end(), cmp);
	//	vector<vector<int>> que;
	//	for (int i = 0; i < people.size(); i++) {
	//		int position = people[i][1];
	//		que.insert(que.begin() + position, people[i]);
	//	}
	//	return que;
	//}
	vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
		sort(people.begin(), people.end(), cmp);	// 底层用链表实现，效率比vector高很多
		list<vector<int>> que;
		for (int i = 0; i < people.size(); i++) {
			int position = people[i][1];
			list<vector<int>>::iterator it = que.begin();
			while (position--) {	// 寻找插入位置
				it++;
			}
			que.insert(it, people[i]);
		} 
		return vector<vector<int>>(que.begin(), que.end());
	}
};  

复杂度分析：

• 时间复杂度： $O(n^2)$。
• 空间复杂度：$O(n)$。

三、完整代码

# include <iostream>
# include <vector>
# include <list>
# include <algorithm>
using namespace std;

class Solution {
public:
	static bool cmp(const vector<int>& a, const vector<int>& b) {
		if (a[0] == b[0]) return a[1] < b[1];	// 如果h相同比较k,k比较大的放后面，升序
		return a[0] > b[0];	// 按h排列，降序
	}
	//vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
	//	sort(people.begin(), people.end(), cmp);
	//	vector<vector<int>> que;
	//	for (int i = 0; i < people.size(); i++) {
	//		int position = people[i][1];
	//		que.insert(que.begin() + position, people[i]);
	//	}
	//	return que;
	//}
	vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
		sort(people.begin(), people.end(), cmp);	// 底层用链表实现，效率比vector高很多
		list<vector<int>> que;
		for (int i = 0; i < people.size(); i++) {
			int position = people[i][1];
			list<vector<int>>::iterator it = que.begin();
			while (position--) {	// 寻找插入位置
				it++;
			}
			que.insert(it, people[i]);
		} 
		return vector<vector<int>>(que.begin(), que.end());
	}
};  

int main() {
	Solution s1;
	vector<vector<int>> people = { {7, 0}, {4, 4}, {7, 1}, {5, 0}, {6, 1}, {5, 2} };
	vector<vector<int>> result = s1.reconstructQueue(people);
	//vector<vector<int>> result = people; 
	for (vector<vector<int>>::iterator it = result.begin(); it != result.end(); it++) {
		for (vector<int>::iterator jt = (*it).begin(); jt != (*it).end(); jt++) {
			cout << *jt << " ";
		}
		cout << endl;
	}
	system("pause");
	return 0;
}

end