解法:排序 + 挨个找位置。
具体思路:将people按照h从小到大,h相同按照k从大到小排列。从下标0开始遍历,一个一个找对应位置(第k+1个空位置,前k个位置用来放比他高的人)。
h从小到大的理由:因为k代表的是前面有多少个大于等于h的个数,所以比它小的先放,并不会影响到它的位置
h相同按照k从大到小排列的理由:h相同k较大的先放不影响较小的放的位置(h相同k较大的被看作是更低的)
class Solution {
public:
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
int n = people.size();
vector<vector<int>> ans(n);
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]);
});
for (auto& each : people) {
int cnt = each[1] + 1; //找第ki + 1个空位置的写法:
for (int i = 0; i < n; ++i) {
if (ans[i].empty()) --cnt;
if (!cnt) {
ans[i] = each;
break;
}
}
}
return ans;
}
};
代码上的难点:从前往后找第k个空位置
int cnt = k;
for (int i = 0; i < n; ++i) {
if (ans[i].empty()) --cnt;
if (!cnt) {
ans[i] = each;
break;
}
}