1971D - Binary Cut - CodeForces Solution

def binary_cut(s):
    no_of_pieces = 0
    ptr = 0
    while ptr < len(s):
        while ptr < len(s) and s[ptr] == '1':
            ptr += 1
        start = ptr
        while ptr < len(s) and s[ptr] == '0':
            ptr += 1
        if ptr >= len(s):
            break
        else:
            no_of_pieces += 1
            while ptr < len(s) and s[ptr] == '1':
                ptr += 1
            end = ptr
            s = s[:start] + s[end:]
        break
    ptr = 0
    while ptr < len(s):
        if s[ptr] == '0':
            while ptr < len(s) and s[ptr] == '0':
                ptr += 1
            no_of_pieces += 1
        else:
            while ptr < len(s) and s[ptr] == '1':
                ptr += 1
            no_of_pieces += 1
    return no_of_pieces


T = int(input())
for _ in range(T):
    S = input()
    print(binary_cut(S))