input() 与 sys.stdin.readline() 的使用区别

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(),会出现文中提到的程序运行超时，不能通过。