给你一个字符串数组 names
,和一个由 互不相同 的正整数组成的数组 heights
。两个数组的长度均为 n
。
对于每个下标 i
,names[i]
和 heights[i]
表示第 i
个人的名字和身高。
请按身高 降序 顺序返回对应的名字数组 names
。
思路:
方法一是利用优先级队列,自己规定排序策略,然后以此将每个人的名字和身高插入,最后遍历队列就是结果。
方法二是利用map直接排序,map默认是会按照pair的第一个数从小到大排序,而题目要求从大到小排序,因此对于map遍历的结果要倒转。
class Solution {
public:
struct cmp
{
bool operator()(pair<int,string>&p1,pair<int,string>&p2)
{
return p1.first<p2.first;
}
};
vector<string> sortPeople(vector<string>& names, vector<int>& heights) {
// map<int,string>hash;
// for(int i=0;i<names.size();i++)
// {
// hash[heights[i]]=names[i];
// }
// vector<string>ret;
// for(auto e:hash)
// {
// ret.push_back(e.second);
// }
// reverse(ret.begin(),ret.end());
priority_queue<pair<int,string>,vector<pair<int,string>>,cmp>dp;
for(int i=0;i<names.size();i++)
{
dp.push({heights[i],names[i]});
}
vector<string>ret;
for(int i=0;i<names.size();i++)
{
auto k=dp.top();
dp.pop();
ret.push_back(k.second);
}
return ret;
}
};