分散的有高度的人

Question

有一排分散的人，我们需要恢复秩序。

我们知道：

• 他们每个人有多高
• 在他们面前较高的人数。

此信息包含在一个集合中

Person {
    int height;
    int tallerAheadCount;
}

我尝试了多种方式对其进行排序，但没有成功。

我设法弄清楚的是，最短的人的 tallerAheadCount 应该与原始索引匹配，但这在具有排序高度的 for 循环中不起作用。

按tallerAheadCount 排序，然后按height 排序给了我们一个相对接近的答案，但tallerAheadCount 越高，它似乎越不正确。我想不出将较短的人合并到较低的tallerAheadCount 排序行的规则。

你会怎么做？

Answer 1

没什么可继续的，但我想这样的事情可能就是你所追求的。

这可能不是最有效的实现，并且不确定它是否涵盖所有情况（例如重复），但这是一个开始：

import random

# dummy data [(height, taller_ahead_count), ...]
original_line = [
    (10, 0), (12, 0), (3, 2), (8, 2), (9, 2), (5, 4), (1, 6), (4, 5), (2, 7)]

# "scatter" the people
scattered_line = original_line.copy()
random.shuffle(scattered_line)

# restore the original line order based on the taller_ahead_count
descending_height = sorted(scattered_line, key=lambda x: x[0], reverse=True)
restored_line = []
for height, taller_ahead_count in descending_height:
    taller_count = 0
    j = 0
    while taller_count < taller_ahead_count and j < len(restored_line):
        if restored_line[j] > height:
            taller_count += 1
        j += 1
    restored_line.insert(j, height)

# verify result
assert [height for height, __ in original_line] == restored_line

基本思路如下：

我们按照高度递减的顺序迭代人，即从最高到最低。这样，在每次迭代中，我们可以确定所有比当前人高的人都已经在restored line 中。然后我们可以计算身高较高的人的数量，以找到应该插入当前人的位置。