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)

posted @ 2022-10-02 10:36  wenli7363  阅读(57)  评论(0编辑  收藏  举报