1、优先考虑身高较矮的人
根据题意我们可以发现,假如我们想要向数组中插入不同身高的人来获得最终的结果,那么在插入时,身高越矮的人在选择位置时需要进行的改动更少。例如身高最矮的人,他 k i k_i ki的值就是他在整个数列中的位置。因此我们在插入人时,我们可以优先从身高较矮的人进行插入,此时他的插入不会对在他之前的所有人产生影响;当有两个人的身高一致时,我们优先考虑排列在后面的,因为若先插入排列在前的人会对后面的人产生影响。
class Solution {
public:
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
sort(people.begin(), people.end(), [](const vector<int>& u, const vector<int>& v) {
return u[0] < v[0] || (u[0] == v[0] && u[1] > v[1]);
});
int n = people.size();
vector<vector<int>> ans(n);
for (const vector<int>& person: people) {
int spaces = person[1] + 1;
for (int i = 0; i < n; ++i) {
if (ans[i].empty()) {
--spaces;
if (!spaces) {
ans[i] = person;
break;
}
}
}
}
return ans;
}
};
2、优先考虑身高较高的人
我们可以采用与方法一相反的思路,我们先确定身高高的人所处的位置。当有身高比他矮的人插入时,我们只需要直接将其插入队列当中,只需要在他之前刚好有 k i k_i ki个人即可。当有两个人身高相同时,我们优先插入排列靠前的人。
class Solution {
public:
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
sort(people.begin(), people.end(), [](const vector<int>& u, const vector<int>& v) {
return u[0] > v[0] || (u[0] == v[0] && u[1] < v[1]);
});
vector<vector<int>> ans;
for (const vector<int>& person: people) {
ans.insert(ans.begin() + person[1], person);
}
return ans;
}
};