题目
假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(h, k)表示,其中h是这个人的身高,k是排在这个人前面且身高大于或等于h的人数。 编写一个算法来重建这个队列。
示例
输入:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]
输出:
[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]
思路
- 先排序,二维数组的一项用vec表示,则vec[0]表示这个人的身高,vec[1]表示这个人前面有几个比他高的。排序的规则是,高个子在前面;个子一样高的时候,前面人少的放到前面。
- 遍历排好序的数组,当前的vec能保证,前面人的高度肯定大于等于vec[0]。vec[1]表示前面比vec[0]高的人数,因为vec前的人都比他高,所以直接插入到vec[1]数字代表的位置。等于身高相同的vec,因为排序时是按照前面人少的靠前,所以插入后先对位置不变。
代码
class Solution {
private:
// 要用static修饰,让cmp属于类,使得cmp不会有默认的this指针参数,使得可以使用在sort里。
static bool cmp( vector<int>& a, vector<int>& b ) {
if ( a[0] != b[0] ) return a[0] > b[0];
return a[1] < b[1];
}
public:
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
if ( people.size() == 0 ) return people;
sort( people.begin(), people.end(), cmp );
for ( int i = 1; i < people.size(); ++i ) {
if ( people[i][1] == i ) continue;
vector<int> temp = people[i];
for ( int j = i - 1; j >= temp[1]; --j )
people[j+1] = people[j];
people[temp[1]] = temp;
}
return people;
}
};