Question

2418. 按身高排序

English Version

题目描述

给你一个字符串数组 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 中的所有值互不相同

解法

方法一：排序

根据题目描述，我们可以创建一个长度为 $n$ 的下标数组 $idx$，其中 $idx[i]=i$。然后我们对 $idx$ 中的每个下标按照 $heights$ 中对应的身高降序排序，最后遍历排序后的 $idx$ 中的每个下标 $i$，将 $names[i]$ 加入答案数组即可。

我们也可以创建一个长度为 $n$ 的数组 $arr$，数组中每个元素是一个二元组 $(heights[i], i)$，然后我们对 $arr$ 按照身高降序排序。最后遍历排序后的 $arr$ 中的每个元素 $(heights[i], i)$，将 $names[i]$ 加入答案数组即可。

时间复杂度 $O(n \times \log n)$，空间复杂度 $O(n)$。其中 $n$ 是数组 $names$ 和 $heights$ 的长度。

class Solution:
  def sortPeople(self, names: List[str], heights: List[int]) -> List[str]:
    idx = list(range(len(heights)))
    idx.sort(key=lambda i: -heights[i])
    return [names[i] for i in idx]

class Solution {
  public String[] sortPeople(String[] names, int[] heights) {
    int n = names.length;
    Integer[] idx = new Integer[n];
    for (int i = 0; i < n; ++i) {
      idx[i] = i;
    }
    Arrays.sort(idx, (i, j) -> heights[j] - heights[i]);
    String[] ans = new String[n];
    for (int i = 0; i < n; ++i) {
      ans[i] = names[idx[i]];
    }
    return ans;
  }
}

class Solution {
public:
  vector<string> sortPeople(vector<string>& names, vector<int>& heights) {
    int n = names.size();
    vector<int> idx(n);
    iota(idx.begin(), idx.end(), 0);
    sort(idx.begin(), idx.end(), [&](int i, int j) { return heights[j] < heights[i]; });
    vector<string> ans;
    for (int i : idx) {
      ans.push_back(names[i]);
    }
    return ans;
  }
};

func sortPeople(names []string, heights []int) (ans []string) {
  n := len(names)
  idx := make([]int, n)
  for i := range idx {
    idx[i] = i
  }
  sort.Slice(idx, func(i, j int) bool { return heights[idx[j]] < heights[idx[i]] })
  for _, i := range idx {
    ans = append(ans, names[i])
  }
  return
}

function sortPeople(names: string[], heights: number[]): string[] {
  const n = names.length;
  const idx = new Array(n);
  for (let i = 0; i < n; ++i) {
    idx[i] = i;
  }
  idx.sort((i, j) => heights[j] - heights[i]);
  const ans: string[] = [];
  for (const i of idx) {
    ans.push(names[i]);
  }
  return ans;
}

impl Solution {
  pub fn sort_people(names: Vec<String>, heights: Vec<i32>) -> Vec<String> {
    let mut combine: Vec<(String, i32)> = names.into_iter().zip(heights.into_iter()).collect();
    combine.sort_by(|a, b| b.1.cmp(&a.1));
    combine
      .iter()
      .map(|s| s.0.clone())
      .collect()
  }
}

方法二

class Solution:
  def sortPeople(self, names: List[str], heights: List[int]) -> List[str]:
    return [name for _, name in sorted(zip(heights, names), reverse=True)]

class Solution {
  public String[] sortPeople(String[] names, int[] heights) {
    int n = names.length;
    int[][] arr = new int[n][2];
    for (int i = 0; i < n; ++i) {
      arr[i] = new int[] {heights[i], i};
    }
    Arrays.sort(arr, (a, b) -> b[0] - a[0]);
    String[] ans = new String[n];
    for (int i = 0; i < n; ++i) {
      ans[i] = names[arr[i][1]];
    }
    return ans;
  }
}

class Solution {
public:
  vector<string> sortPeople(vector<string>& names, vector<int>& heights) {
    int n = names.size();
    vector<pair<int, int>> arr;
    for (int i = 0; i < n; ++i) {
      arr.emplace_back(-heights[i], i);
    }
    sort(arr.begin(), arr.end());
    vector<string> ans;
    for (int i = 0; i < n; ++i) {
      ans.emplace_back(names[arr[i].second]);
    }
    return ans;
  }
};

func sortPeople(names []string, heights []int) []string {
  n := len(names)
  arr := make([][2]int, n)
  for i, h := range heights {
    arr[i] = [2]int{h, i}
  }
  sort.Slice(arr, func(i, j int) bool { return arr[i][0] > arr[j][0] })
  ans := make([]string, n)
  for i, x := range arr {
    ans[i] = names[x[1]]
  }
  return ans
}

function sortPeople(names: string[], heights: number[]): string[] {
  return names
    .map<[string, number]>((s, i) => [s, heights[i]])
    .sort((a, b) => b[1] - a[1])
    .map(([v]) => v);
}