题目
代码(7.25 首刷看解析)
做这个题的时候,要先确定一个维度,再确定另一个维度。如果两个维度一起考虑一定会顾此失彼。
这里先确定身高h,再确定k。
class Solution {
public:
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
auto cmp = [&](vector<int>& a, vector<int>& b){
if(a[0] == b[0]) return a[1] < b[1];
return a[0] > b[0];
};
sort(people.begin(), people.end(), cmp);
vector<vector<int>> res;
for(auto& p : people) {
int position = p[1];
res.insert(res.begin() + position, p);
}
return res;
}
};
但使用vector是非常费时的,C++中vector(可以理解是一个动态数组,底层是普通数组实现的)如果插入元素大于预先普通数组大小,vector底部会有一个扩容的操作,即申请两倍于原先普通数组的大小,然后把数据拷贝到另一个更大的数组上。
所以使用vector(动态数组)来insert,是费时的,插入再拷贝的话,单纯一个插入的操作就是O(n2)了,甚至可能拷贝好几次,就不止O(n2)了。
改用list
,注意,iterator是只能进行++操作的。
class Solution {
public:
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
auto cmp = [&](vector<int>& a, vector<int>& b){
if(a[0] == b[0]) return a[1] < b[1];
return a[0] > b[0];
};
sort(people.begin(), people.end(), cmp);
list<vector<int>> res;
for(auto& p : people) {
int position = p[1];
auto it = res.begin();;
while(position--) it++;
res.insert(it, p);
}
return vector<vector<int>>(res.begin(), res.end());
}
};
代码(10.1 二刷看解析)
class Solution {
public:
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
sort(people.begin(), people.end(), [](auto& a, auto& b) {
if(a[0] == b[0])
return a[1] < b[1];
return a[0] > b[0];
});
// vector<vector<int>> res;
list<vector<int>> res;
for(auto& p : people) {
int k = p[1];
auto it = res.begin();
while(k--) {
it++;
}
res.insert(it, p);
}
return vector<vector<int>>(res.begin(), res.end());
}
};