python 3全RE求助，过了样例，第一个测试本地是对的

python 3全RE

matrix = []
line = input()
matrix.append(line)
n = len(line)
for i in range(n - 1):
    matrix.append(input())
print(n, end = ' ')
flag = '0'
count = 0
for i in range(n):
    for j in range(n):
        if matrix[i][j] == flag:
            count += 1
        else:
            print(count, end = ' ')
            count = 1
            flag = matrix[i][j]
print(count)