Python 牛客网编程题的输入用例
- 前言
- 切入正题
- 输入字符串
- 输入整数
- 输入列表
- 后记
前言
在牛客网刷题的时候,发现明明在python本地编译器能够轻松解决的输入用例问题,到牛客网在线编译就得好好斟酌,把题目所给的输入样例样式转换为我们处理时需要的数据格式,前期摸索的过程特别耗时间。因此,把自己踩过的坑总结出来,供大家参考和补充。
切入正题
以下都是基于Python3.x版本的输入用例总结。
输入字符串
直接用 input() 函数,接受一个标准输入数据,返回为 string 类型。
可能有小伙伴会说,还有一个基础输入是 raw_input( ),那我们需要明确的是,在 Python3.x 中 raw_input( ) 和 input( ) 进行了整合,去除了 raw_input( ),仅保留了 input( ) 函数。
>>> n = input()
输入整数
在上述 input 基础上,进行 int() 格式转换。
此步骤有几种情况:
- 单行输入一个整数,直接进行格式转换即可
>>> n = int(input())
- 单行输入多个整数,以特定操作符分隔,存储为多个变量
此时引入一个新工具 map() 函数 ——
map(function, iterable, …)
> 根据提供的函数(也就是我们需要转换的数据格式),对指定序列(输入用例,或者是数据列表等)做映射
举个栗子:输入样例为一行2个整数,以空格分隔,对应a, b两个变量
# 用 split() 实现分隔
>>> a, b = map(int, input().split())
如果要进一步操作,把多个输入的整数放进列表,再加一步即可
>>> nlist = list(map(int, input().split()))
- 多行连续输入,每行输入1个数
方法1:通过循环把数据 append 进一个空列表,或者直接在循环里读取数据并进行处理,数据量大的情况下特别消耗时间运行。
方法2:用系统读取行的方法,比较适用于牛客网在线判例,本地IDLE观察不到结果。
# 法1代码
>>> nlist = []
>>> for i in range(n):
nlist.append(int(input()))
# 法2代码
>>> import sys
>>> array = list(map(int, sys.stdin.readlines()))
# 不想输入一个列表容器的话👇
>>> for line in sys.stdin:
>>> a, b = line.split() #根据需要定义变量,split分隔符也是根据输入决定
输入列表
有时候会遇到输入样例直接是一个列表或者是嵌套列表的情况,如果直接用 input() 会返回一个字符串,后续进行分隔操作也繁琐麻烦,我们想一步到位得到和输入一致的列表结构。引入 eval() 函数 ——
eval() 函数用来执行一个字符串表达式,并返回表达式的值
eval(str) 的功能很强大,它既可以计算字符串中有效的表达式,又能将字符串转成相应的对象(如list、tuple、dict和string之间的转换),直接上代码~
# 1.计算字符串中有效的表达式
>>> eval('pow(2,2)')
4
# 2.字符串与其他对象的转换
# 假设输入用例为嵌套列表 [[0,0,1,1],[1,0,1,0],[0,1,1,0],[0,0,1,0]]
>>> alist = eval(input())
>>> alist
[[0,0,1,1],[1,0,1,0],[0,1,1,0],[0,0,1,0]]
后记
准备踏足互联网领域的小白的第一篇博文,希望能对大家有所帮助,也希望大家批评指正、分享促进步~!