本题源自leetcode 406
----------------------------------------------------------------------------
思路:1 先按身高排序,如果身高相等再按第二个参数排序。
2 根据顺序遍历数组,按第二个元素的值插入结果数组、。
代码:
vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
auto comp = [](const pair<int,int>& p1, const pair<int,int>& p2){ //lamba表达式
return (p1.first > p2.first ||(p1.first == p2.first && p1.second < p2.second));
};
vector<pair<int,int>> res;
sort(people.begin(),people.end(),comp);
for(auto& p : people){
res.insert(res.begin() + p.second, p);
}
return res;
}
struct comp{ //函数子对象
bool operator()(const pair<int,int>& p1, const pair<int,int>& p2){
return (p1.first > p2.first ||(p1.first == p2.first && p1.second < p2.second));
}
};
vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
/*
auto comp = [](const pair<int,int>& p1, const pair<int,int>& p2){
return (p1.first > p2.first ||(p1.first == p2.first && p1.second < p2.second));
};
*/
vector<pair<int,int>> res;
sort(people.begin(),people.end(),comp());
for(auto& p : people){
res.insert(res.begin() + p.second, p);
}
return res;
}