Python 中实现类似C++的cin
突然发现牛客上有些题目的测试略坑Python。因为C++中默认用cin,所以‘ ’和‘\n’不同太区分。可是Python没这个功能啊。所以有时候发现程序通不过是跪在输入上。比如明明说好有3行的输入,结果实际测试中就只有2行,其中‘\n’偷懒用空格替代了。所以在想如何实现cin。求指点有没有个好想法解决这问题。我的解决时:用一个generator,然后调用next()读入。
import sys def read(): for line in sys.stdin: for s in line: yield s