所有的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^2) O(n2)。
- 空间复杂度: O ( n ) O(n) 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