python关于算法题的输入
关于Python算法题的输入处理
最近在准备蓝桥杯,打算报Python组,所以开始尝试用Python刷算法题。
【python & ACM 输入输出的处理:sys.stdin.readline().strip().split())】
上面这个博客写得挺全面的,就特此记录一下。
当然,wyl提出一个更简洁的写法,也记录一下。
import sys
read = lambda: [int(x) for x in sys.stdin.readline().split()]
这里再给出一个创建多维数组的方法
alloc = lambda *size: [0] * size[0] if len(size) == 1 else [alloc(*size[1:]) for i in range(size[0])]
alloc=lambda*s:len(s)-1and[alloc(*s[1:])for i in range(s[0])]or[0]*s[0]
#调用该方法
alloc(2,3)