23570: 特殊密码锁

http://cs101.openjudge.cn/practice/23570/

有一种特殊的二进制密码锁，由n个相连的按钮组成（1<=n<30），按钮有凹/凸两种状态，用手按按钮会改变其状态。

然而让人头疼的是，当你按一个按钮时，跟它相邻的两个按钮状态也会反转。当然，如果你按的是最左或者最右边的按钮，该按钮只会影响到跟它相邻的一个按钮。

当前密码锁状态已知，需要解决的问题是，你至少需要按多少次按钮，才能将密码锁转变为所期望的目标状态。

输入

两行，给出两个由0、1组成的等长字符串，表示当前/目标密码锁状态，其中0代表凹，1代表凸。

输出

至少需要进行的按按钮操作次数，如果无法实现转变，则输出impossible。

样例输入

011
000

样例输出

python

"""
the toggle function is used to flip the bit, which simplifies the flip function. 
using a for-loop to iterate over the two cases: pressing the first button or not. 
"""
def toggle(bit):
    return '0' if bit == '1' else '1'

def flip(lock, i):
    if i > 0:
        lock[i-1] = toggle(lock[i-1])
    lock[i] = toggle(lock[i])
    if i + 1 < len(lock):
        lock[i+1] = toggle(lock[i+1])

def main():
    s = input()
    fin = input()
    n = len(s)
    ans = float('inf')

    for press_first in [False, True]:
        tmp = 0
        lock = list(s)
        if press_first:
            flip(lock, 0)
            tmp += 1
        for i in range(1, n):
            if lock[i-1] != fin[i-1]:
                flip(lock, i)
                tmp += 1
        if lock[n-1] == fin[n-1]:
            ans = min(ans, tmp)

    if ans == float('inf'):
        print("impossible")
    else:
        print(ans)

if __name__ == "__main__":
    main()

23570: 特殊密码锁 ​

23570: 特殊密码锁