Python3如何在OJ获取输入数据

问题提出

以前一直都是使用 C++ 在 OJ 上做题目。今天突然有人问我如何使用 Python3 在 OJ 上提交程序？经过若干讨论和测试，发现主要的问题是 Python3 如何获取 OJ 的输入数据。

经过总结，个人将输入数据模式分为三种。

Python 获取 OJ 输入

模式一

本模式适用于固定 n 行，每行一个数据，数据类型无所谓。该模式的典型题目如下图。

如上图所示，一共三行，每行一个整数。因此，我们可以利用 input() 这个函数来获取输入，然后再将数据强制转换。基本方法如下：

a = int(input())
b = int(input())
c = int(input())

AC 参考代码

#!/usr/bin/env python
# coding=utf-8
from math import acos
from math import pi

#读入数据
a = int(input())
b = int(input())
c = int(input())

#计算过程
x = a*a+b*b-c*c
y = x/2/a/b
z = acos(y)

#输出结果
print(round(z*180/pi,1))

视频

有兴趣可以参考这个视频，https://www.bilibili.com/video/av94266432/。

模式二

本模式适用于只有一行输入，在这行输入中有若干个数据，每个数据用空格或者其他字符隔开。如下图所示：

因此，我们可以利用 input() 这个函数来获取输入，然后在使用 split，将其划分到 lambda 表达式中。如下：

a = list(map(lambda x:float(x), input().split()))

注意，由于数据中有浮点数，所以全部转换成为 float，不能转换为 int，否则会出错。具体根据题目数据类型就行处理。

AC 参考代码

#!/usr/bin/env python
# coding=utf-8

#a[0]是向账号里存的钱
#a[1]是年利率
#a[2]是存钱年限
a = list(map(lambda x:float(x), input().split()))

ans = 0
i=0
while i < a[2]:
    ans = (ans+a[0]) * (1+a[1])
    i = i + 1
    
print(round(ans,2))

视频

有兴趣可以参考这个视频，https://www.bilibili.com/video/av94266945/。

模式三

本模式适用于有若干行输入。对于每行多少个数据，我们再分别套用模式一或者模式二即可。如下图所示：

由于我们不知道到底有几行，只能通过读到 EOF 的时候确认输入结束。因此我们可以使用死循环读取模式一来解决这个问题。如下所示：

while True:
    try:
        x = int(input())
    except:
        break

AC 参考代码

#!/usr/bin/env python
# coding=utf-8

ans = 0
while True:
    try:
        ans += int(input())
    except:
        break
print(ans)

视频

有兴趣可以参考这个视频，https://www.bilibili.com/video/av94267357/。

题后

感觉 Python3 对数据的输入要求非常严格，而 C++ 对数据输入格式相对没有 Python3 严格。