Sort the People — LeetCode 2418 Python Solution

The problem

You are given an array of strings names, and an array heights that consists of distinct positive integers. Both arrays are of length n.

Example

Input

names = ["Mary","John","Emma"], heights = [180,165,170]

Output

["Mary","Emma","John"]

Explanation

Mary is the tallest, followed by Emma and John.

Python solution

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]

Complexity

Pattern: Sorting

Spend O(n log n) once to buy an ordering that makes the rest of the problem trivial. LeetCode 2418. Sort the People is filed here because LeetCode tags it Sorting, which is the vocabulary this hub collects.

The sorting guide has the Python template for the pattern and the 401 LeetCode problems that use it.

Related problems