写在前面:摸鱼将近5个月,终于回来刷题了。
题目:
假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(h, k)表示,其中h是这个人的身高,k是排在这个人前面且身高大于或等于h的人数。 编写一个算法来重建这个队列。
思路:
1.首先考虑从高到低排列身高,在同等身高的情况下,k值越小置前,按照这个顺序重新整理队列
2.然后按照顺序依次取出(h,k),插入到位置res[k]上,这样可以保证前面至少有k个位置可以摆放比他高的,又因为我们已经按照身高(第一属性,从高到低)和K值(第二属性,从小到大)整理了队列,已经按照从高到低的顺序排好了身高,这样我们在执行add/insert操作的时候,就不会出现在同k值情况下,身高高的排在身高低的前面这种情况。而在同等身高情况下,按照我们插入的方法,k值小的一定排在k值大的前面。这样,我们这样每当我们取出一个(h,k)时,就不会影响到事先已经排列好的队伍。
代码:(懒狗直接引用了官方题解自用,以后再来优化……)
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;
}
};