python跳一跳

#-*- coding: utf8 -*-

'''
跳一跳游戏
1. 输入一个字符串，eg: 1 2 1 2 1 1 1 2 2 2 1 0
2. 输入总和：11
'''

'''
检查输入的数组是否正确
'''
def checkStr(testStr):
    if(int(inputStr[-1]) != 0):
        print("输入的结尾不为0")
        return None
    tytLst = []        
    # 输入的字符中有字符串
    for s in inputStr:
        try:
            ints = int(s)
            #print(ints)
            if(ints >2 or ints<0):
                print("输入的字符串不符合")
                return None
            tytLst.append(ints)
        except ValueError as e:
            print(e)
            return None
    return tytLst
    

if __name__ == '__main__':
    inputStr = input("请输入一个字符串， eg：1 2 2 1 1 2 0，结尾要为0 \n").split()
    test = checkStr(inputStr)
    if(test != None):
        f = lambda lst: sum(lst)
        score = f(test)
        print(score)