在 Python 中查找显示将所有球移动到当前位置所需的总距离的列表的程序

假设我们有一个名为nums的二进制列表，它只包含0和1，其中0表示空单元格，1表示单元格被球填充。我们必须找到一个新的sayL列表，其大小也与nums大小相同，其中L[i]设置为将所有球移动到L[i]所需的总距离。这里将球从索引j移动到索引i的距离是|j-i|。

因此，如果输入类似于nums=[1,1,0,1]，那么输出将是[4,3,4,5]，因为

L[0]=|0-0|+|1-0|+|3-0|

L[1]=|0-1|+|1-1|+|3-1|

L[2]=|0-2|+|1-2|+|3-2|

L[3]=|0-3|+|1-3|+|3-3|

因此，要将所有球移动到L[1]，我们必须将球从索引0移动到1，距离为1，并将球从索引3移动到1，距离为2。

示例

让我们看看以下实现以获得更好的理解-

def solve(nums): if not nums: return [] left_count = right_count = 0 left_sum = right_sum = 0 result = [] for index, num in enumerate(nums): if num: right_count += 1 right_sum += index for index, num in enumerate(nums): result.append(left_sum + right_sum) if num: right_count -= 1 left_count += 1 left_sum += left_count right_sum -= right_count return result nums = [1, 1, 0, 1] print(solve(nums))

输入

[1, 1, 0, 1]输出结果[4, 3, 4, 5]