目录
力扣2418. 按身高排序
解析代码
力扣2418. 按身高排序
2418. 按身高排序
难度 简单
给你一个字符串数组 names
,和一个由 互不相同 的正整数组成的数组 heights
。两个数组的长度均为 n
。
对于每个下标 i
,names[i]
和 heights[i]
表示第 i
个人的名字和身高。
请按身高 降序 顺序返回对应的名字数组 names
。
示例 1:
输入:names = ["Mary","John","Emma"], heights = [180,165,170] 输出:["Mary","Emma","John"] 解释:Mary 最高,接着是 Emma 和 John 。
示例 2:
输入:names = ["Alice","Bob","Bob"], heights = [155,185,150] 输出:["Bob","Alice","Bob"] 解释:第一个 Bob 最高,然后是 Alice 和第二个 Bob 。
提示:
n == names.length == heights.length
1 <= n <= 10^3
1 <= names[i].length <= 20
1 <= heights[i] <= 10^5
names[i]
由大小写英文字母组成heights
中的所有值互不相同
class Solution {
public:
vector<string> sortPeople(vector<string>& names, vector<int>& heights) {
}
};
解析代码
此题并不是贪心的题,只是一个简单的排序,为下一道贪心的题做准备。
可以直接创建一个二元组vector<pair<int, string>>,或者用哈希的解法。下面看另一种解法:
对下标排序解法(通过排序 ''索引'' 的方式):我们不能直接按照 i 位置对应的 heights 来排序,因为排序过程是会移动元素的,但是names 内的元素是不会移动的。
由题意可知, names数组和 heights数组的下标是壹壹对应的,因此可以重新创建出来一个下标数组,将这个下标数组按照 heights[i] 的大小排序。那么,当下标数组排完序之后,里面的顺序就相当于heights这个数组排完序之后的下标。之后通过排序后的下标,依次找到原来的name ,完成对名字的排序。
class Solution {
public:
vector<string> sortPeople(vector<string>& names, vector<int>& heights) {
int sz = heights.size();
vector<int> index(sz);
for(int i = 0; i < sz; ++i)
{
index[i] = i;
}
sort(index.begin(), index.end(),[&](int index1, int index2)
{
return heights[index1] > heights[index2];
});
vector<string> ret;
for(auto& e : index)
{
ret.push_back(names[e]);
}
return ret;
}
};