算法日常・每日刷题--贪心＜11＞

2418. 按身高排序 - 力扣（LeetCode）

题目描述

给你两个数组：names 人名数组，heights 身高数组。 names[i] 对应 heights[i]，代表第 i 个人的名字和身高。 请按照身高从高到低的顺序，返回排序后的人名数组。

核心难点：两个数组并行关联，不能单独排序其中一个数组，否则名字和身高对应关系会错乱。

思路讲解（下标索引排序法）

整体思想

不去直接排序names和heights，而是保存原始下标，对下标数组进行排序。

• index数组存储的是原始数组的下标：0,1,2,3...n-1
• 自定义 sort 的比较器：按下标对应的身高降序排序
• 排序完成后，index里面的下标顺序，就是身高从高到低的顺序
• 最后遍历排好序的 index，取出对应的名字，存入结果数组

原始index = [0,1,2]

• index 0=0 → Mary，身高 180
• index 1=1 → John，身高 165
• index 2=2 → Emma，身高 170

排序 index，按照 heights 降序： 比较后得到 index = [0,2,1]

• index=0 → Mary(180)
• index=2 → Emma(170)
• index=1 → John(165)

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