来源:力扣(LeetCode)
描述:
给你一个字符串数组 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 <= 103
- 1 <= names[i].length <= 20
- 1 <= heights[i] <= 105
- names[i] 由大小写英文字母组成
- heights 中的所有值互不相同
方法:排序
思路与算法
我们可以将 names[i] 和 heights[i] 绑定为一个二元组,然后对所有的二元组按照 heights 排序。最后取出其中的 names 即为答案。
除了以上方法,我们还可以创建一个索引数组 indices,其中 indices[i] = i。排序完成后,对于所有的 i, j (i < j) 都有 heights[indices[i]] > heights[indices[j]]。然后我们遍历 i 从 0 到 n − 1,将 names[indices[i]] 追加到答案数组中。
代码:
class Solution {
public:
vector<string> sortPeople(vector<string>& names, vector<int>& heights) {
int n = names.size();
vector<int> indices(n);
iota(indices.begin(), indices.end(), 0);
sort(indices.begin(), indices.end(), [&](int x, int y) {
return heights[x] > heights[y];
});
vector<string> res(n);
for (int i = 0; i < n; i++) {
res[i] = names[indices[i]];
}
return res;
}
};
执行用时:24 ms, 在所有 C++ 提交中击败了93.78%的用户
内存消耗:18.4 MB, 在所有 C++ 提交中击败了76.78%的用户
复杂度分析
时间复杂度:O(nlogn),其中 n 是 names 和 heights 的长度。对 indices 数组排序的时间复杂度为 O(nlogn)。
空间复杂度:O(n),其中 n 是 names 和 heights 的长度。排序过程中所需要的栈空间为 O(logn),创建 indices 数组所需要的空间是 O(n),对它们求和后空间复杂度渐进意义下等于 O(n)。
author:LeetCode-Solution