用 Python 编写程序,在给定字符串中以最少的交换次数对 1 进行分组

pythonserver side programmingprogramming更新于 2026/1/11 14:36:17

假设我们得到一个包含 0 和 1 的二进制字符串 input_str。我们的任务是通过交换给定字符串中的 1 来对 0 和 1 进行分组。我们必须执行最少数量的交换操作,并且必须返回该值。需要记住的一点是,我们只能交换相邻的值。

因此,如果输入为 input_str = 10110101,则输出为 4

交换将类似于以下 −

10110101->01110101->01111001->01111010->01111100

交换总数:4。

为了解决这个问题,我们将遵循以下步骤 −

  • one := 一个新列表,其中包含 input_str 中 1 所在的位置
  • mid := 下限值(大小为 one / 2)
  • res := 0
  • 对于 i 在 0 到 1 的范围内,执行
    • res := res + |one[mid] - one[i]| - |mid - i|
  • 如果 res < 0,则
    • 返回 0
  • 否则,
    • 返回 res

示例

让我们看看下面的实现以便更好地理解 −

def solve(input_string):
   one = [i for i in range(len(input_string)) if input_string[i] == "1"]
   mid = len(one) // 2
   res = 0
   for i in range(len(one)):
      res += abs(one[mid] - one[i]) - abs(mid - i)
   return 0 if res < 0 else res

print(solve('10110101'))

输入

'10110101'

输出

4

相关文章


有用资源