用Python写一个程序对给定字符串中的1进行最小交换以分组1
假设我们有一个包含0和1的二进制字符串input_str。我们的任务是通过交换给定字符串中的1来分组0和1。我们必须执行最小数量的交换操作,并返回该值。有一件事要记住,我们只能交换相邻的值。
因此,如果输入是input_str = 10110101,则输出将是4。
交换将如下所示 −
10110101 → 01110101 → 01111001 → 01111010 → 01111100
交换的总数:4。
为了解决这个问题,我们将遵循以下步骤−
- one :=一个新列表,其中包含1在input_str中出现的位置
- mid := one的大小的floor值/ 2
- res := 0
- 对于在one大小范围内的i,执行以下操作:
- 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