if __name__ == '__main__':
    n = input().strip()

    print(len(n), end=" ")

    for i in range(len(n) - 1):
        n += input().strip()

    ans = 0
    i = 0
    while i < len(n):
        ans += 1
        if i == len(n) - 1 or n[i] != n[i + 1]:  # 如果与下一个数不一样或者为最后一位时，输出统计结果
            print(ans, end=" ")
            ans = 0  # 清除统计结果
        i += 1

我感觉逻辑好像有问题，但是又想不明白哪里有问题。求助大佬能够指点迷津。