input() 与 sys.stdin.readline() 的使用区别
用python在牛客上刷题,或者是做一些面试题时,会看到系统给的默认输入是sys.stdin.readline()而不是常见的input(),因为oj系统的输入一般是从一个文件里读的,而不是从键盘输入,如果用input()会造成程序一直在等待键盘输入而造成超时,所以在一些刷题网站(比如牛客)上做题时,尽量用sys.stdin.readline()。下面用一个牛客上的题来说明上述问题。
NowCoder数列
NowCoder最近在研究一个数列:
- F(0) = 7
- F(1) = 11
- F(n) = F(n-1) + F(n-2) (n≥2)
他称之为NowCoder数列。请你帮忙确认一下数列中第n个数是否是3的倍数。
输入描述:
输入包含多组数据。
每组数据包含一个整数n,(0≤n≤1000000)。
输出描述:
对应每一组输入有一行输出。
如果F(n)是3的倍数,则输出“Yes”;否则输出“No”。
输入例子:
0
1
2
3
4
5
输出例子:
No
No
Yes
No
No
No
简单列出前几项,找规律,发现当(n-2)%4为0时,输出yes,否则输出No。代码如下:
import sys
def main():
while True:
a = sys.stdin.readline()
if a=='':
break
b=int(a)
if (b-2)%4==0:
print('Yes')
else:
print('No')
return 0
if __name__=='__main__':
main()
在大多数IDE中(spyder除外),input() 和sys.stdin.readline()的作用几乎相同,但在某些刷题网站上只能用sys.stdin.readline(),而不能用input(), 注意避坑。如果将程序里的sys.stdin.readline() 改为input(),会出现文中提到的程序运行超时,不能通过。