牛客\小A的数字\Python

Q:C-小A的数字_牛客小白月赛90 (nowcoder.com)

#把数字分为有0无0两种情况
#如果没有0,如果最后一位数字是1 我们直接输出2最小 否则输出1
#如果有0，说明数字是有各种数字加0组成，我们将0改为1即可，其他数字改为0
t = int(input())
for _ in range(t):
    t = input()
    lt = len(t)
    cnt = t.count('0')
    if cnt == 0:
        if t[-1] == '1':
            print(2)
        else:
            print(1)
    else:
        dp = ['0'] * lt
        for i in range(lt-1,-1,-1):
            if t[i] == '0':
                dp[i] = '1'
        num = int(''.join(dp))
        print(num)

如有不足，欢迎指正