题目描述
给你一个字符串数组 names
,和一个由 互不相同 的正整数组成的数组 heights
。两个数组的长度均为 n
。
对于每个下标 i
,names[i]
和 heights[i]
表示第 i
个人的名字和身高。
请按身高 降序 顺序返回对应的名字数组 names
。
样例
输入:names = ["Mary","John","Emma"], heights = [180,165,170]
输出:["Mary","Emma","John"]
解释:Mary 最高,接着是 Emma 和 John。
输入: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
中的所有值互不相同。
算法
(模拟) $O(n \log n)$
- 直接按照题目描述排序,可以排序下标。
时间复杂度
- 排序的时间复杂度为 $O(n \log n)$。
空间复杂度
- 需要 $O(n)$ 的额外空间存储下标数组,答案和排序递归的系统栈。
C++ 代码
class Solution {
public:
vector<string> sortPeople(vector<string>& names, vector<int>& heights) {
const int n = names.size();
vector<int> rk(n);
for (int i = 0; i < n; i++)
rk[i] = i;
sort(rk.begin(), rk.end(), [&](int i, int j) {
return heights[i] > heights[j];
});
vector<string> res(n);
for (int i = 0; i < n; i++)
res[i] = names[rk[i]];
return res;
}
};