思路:
首先利用h[i]和k[i]将原数组排序,那么怎么利用呢?可以以h[i]为主,k[i]为辅,即按照身高从高往低的顺序排,如果两者身高相同,则按照前面比他高的人数k[i]排,越少越靠前。
然后再重新定义一个二维数组,用以存放最终结果。遍历排好序的数组,将他们以k[i]进行对应位置的插入,即k[i]决定了他们在新数组中的位置。因为个子高的插入先后顺序对个子低的会有影响,而各自低的插入后对个子高的没有影响,所以我们首先按照由高到矮的顺序排序,然后再依次插入到新数组中。
代码实现(C++):
class Solution {
public:
static bool cmp(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>> reconstructQueue(vector<vector<int>>& people) {
sort(people.begin(), people.end(), cmp);
vector<vector<int>> queue;
int len = people.size();
for(int i = 0; i < len; i++){
int pos = people[i][1];
queue.insert(queue.begin() + pos, people[i]);
}
return queue;
}
};
总结
本题用到了vector动态生成二维数组,比较方便使用,但是更消耗内存。用到了sort函数进行排序,他定义在#include < algorithm > 的c++标准库中需要对其传入三个参数
- 第一个:要排序的数组的起始位置
- 第二个:结束的位置(是最后一个数据的后一个数据的地址)
- 第三个:排序规则,可有可无,如果不写,默认从小到大。