【LeetCode】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 <= 103
• 1 <= names[i].length <= 20
• 1 <= heights[i] <= 105
• names[i] 由大小写英文字母组成
• heights 中的所有值互不相同

方法一：排序

思路与算法：创建一个索引数组index，根据heights[i]对index[i]排序，最后遍历存储入字符串数组。

class Solution {
public:
    vector<string> sortPeople(vector<string>& names, vector<int>& heights) {
        int n = names.size();
        vector<int> index(n);
        //index初始化
        iota(index.begin(), index.end(), 0);
        //排序索引
        sort(index.begin(), index.end(), [&](int a, int b) {
            return heights[a] > heights[b];
        });
        //存储入字符串数组
        vector<string> s(n);
        for (int i = 0; i < n; i++) {
            s[i] = names[index[i]];
        }
        return s;
    }
};

class Solution {
    public String[] sortPeople(String[] names, int[] heights) {
        int n = names.length;
        //这里注意要用Integer创建索引数组，java的许多api不支持int，其中就包含Arrays.sort
        Integer[] index = new Integer[n];
        for (int i = 0; i < n; i++) {
            index[i] = i;
        }
        Arrays.sort(index, (a, b) -> heights[b] - heights[a]);
        String[] s = new String[n];
        for (int i = 0; i < n; i++) {
            s[i] = names[index[i]];
        }
        return s;
    }
}

方法二：哈希表

思路与算法：姓名和身高存入哈希表中，TreeMap可实现Integer身高排序，之后获取键值遍历存储姓名至字符串数组即可即可。 

class Solution {
    public String[] sortPeople(String[] names, int[] heights) {
        //创建TreeMap存储键值 这里可直接进行排序
        Map<Integer,String> mp = new TreeMap<>();
        int n = names.length;
        for(int i = 0;i<n;i++){//数据存入哈希表
            mp.put(heights[i],names[i]);
        }
        //获取键值
        Set<Integer> keySet = mp.keySet(); 
        //获取value并遍历存入字符串数组
        String s[] = new String[n];
        for(Integer i: keySet){
            s[--n] = mp.get(i);
        }
        return s;
    }
}