输入数据时记得使用strip（）

nums = input().strip()

不然你会收获WA