2023.8.2
这题一开始有点让人懵逼的是有两个维度,一个是身高,还一个是前面人高于自己的人数。这种题一般需要先固定一个维度,再去确定另外一个维度,不要想着兼顾。
经过纸上模拟,我的思路是先通过身高进行从大到小排序,如果身高相同则k值小的站前面。排完序之后再将第二个维度 k 值 当作索引重建新的二维数组。代码细节如下:
class Solution {
public:
static bool cmp(vector<int>& a , vector<int>& b)
{
if(a[0] == b[0]) return a[1] < b[1];
return a[0] > b[0];
}
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
sort(people.begin(),people.end(),cmp);
vector<vector<int>> ans;
for(int i=0; i<people.size(); i++)
{
ans.insert(ans.begin()+people[i][1],people[i]);
}
return ans;
}
};
优化:
考虑到vector的插入效率比较低下,可以使用链表list来进行插入,时间效率会高很多:
class Solution {
public:
static bool cmp(vector<int>& a , vector<int>& b)
{
if(a[0] == b[0]) return a[1] < b[1];
return a[0] > b[0];
}
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
sort(people.begin(),people.end(),cmp);
list<vector<int>> ans;
for(int i=0; i<people.size(); i++)
{
int index = people[i][1];
auto it = ans.begin();
while(index--)
{
it++;
}
ans.insert(it,people[i]);
}
return vector<vector<int>>(ans.begin(),ans.end());
}
};
2024.2.6
Java二刷,代码如下:
class Solution {
public int[][] reconstructQueue(int[][] people) {
//将数组按第一个元素降序,按第一个元素升序。
Arrays.sort(people,new Comparator<int[]>(){
@Override
public int compare(int[] p1 , int[] p2){
if(p1[0] == p2[0]){
return p1[1]-p2[1];
}
return p2[0] - p1[0];
}
});
List<int[]> ans = new ArrayList<>();
for(int i=0; i<people.length; i++){
if(ans.size() > people[i][1]){
ans.add(people[i][1] , people[i]);
}
else{
ans.add(people[i]);
}
}
return ans.toArray(new int[ans.size()][]);
}
}