leetcode编程框架——在IDE上模拟leetcode环境

1、为什么写这个

为了用IDE!,这样不同的大题,复制粘贴就完事,真正帮你节约大把时间来思考题目。

import sys

# 这里写解决问题的代码,和LeetCode就完全一样了
def solve(arr):
 pass


if __name__ == '__main__':

 # 接收输入的逻辑,这里先把输入接收过来, 两种选择input()和sys.stdin.readline()
 group_nums = input()   #字符串形式,得转成int
 
 group_nums = int(group_nums)
 
 # 对于每一组
 for i in range(group_nums):
  # 接收每一组的输入, 这里不同的题目就不一样了,但一定记住我们接收的还是一行,这是一个字符串
  arr = sys.stdin.readline().strip().split(' ')
  # 元素转成int
  arr = list(map(int, arr))
  # 输入接收过来之后,这里最好打印下看看接收的是不是正确,这个很重要
  # print(arr)
  # 处理具体的问题了
  res = solve(arr)
  # 输出结果
  print(res) 

  上面这个模板,处理的是多组输入,或者多行输入;如果是一组输入,仅仅几行的话, 就不用外层for循环的, 一行行的接收即可。

2、框架模板的优势

  这个框架做到了主函数与处理问题的逻辑分离开。接收输入 -> 处理函数 -> 调整输出即可。 这么一来,我们在solve里就可以像在LeetCode上一样,直接写解决问题的函数。这样的代码看起来清晰,找bug时也好找。

3、框架模板的注意点

  逻辑通了,下面是细节上要注意的。首先是,接收输入时都是一行一行地接收,这个是字符串形式,要先接收过来,再转换格式。我个人习惯是先都接收了,然后统一转换格式,比较清晰。只要保证接收正确,大家也可以在接收时直接转。

  这里主要整理接收intput()sys.stdin.readline()两种方式的区别。这两个都是字符串格式,读入一行,区别是后者会带着一个换行符\n。啥意思?我们在输入时,是把输入输完之后,敲一个回车告诉编译器我敲完了。这俩的区别就是input()会过滤掉最后的回车,而sys.stdin.readline()会带着后面的回车。

一个最简单的例子:

arr = sys.stdin.readline().split(' ')
print(arr)    # ['1', '2', '3', '4', '5\n']

arr = input().split(' ')
#arr = sys.stdin.readline().split(' ')
print(arr)      # ['1', '2', '3', '4', '5']

  可以发现,如果用上面那个代码,接收的输入可能是不正确的。比如,有时用下面这个代码把每个元素转成int时,会出现一些不知名的错误:

list(map(int ,arr))

  所以,如果想用sys接收输入的话,更加鲁棒性的代码是:

arr = sys.stdin.readline().strip().split(' ')

  也就是加一个strip函数去掉多余的空格或者换行符。还有一种鲁棒性的写法:

arr = input().strip().split(' ')

  这两个写法都可以, 建议用input。

同样的输出也有两种方式:

# sys.stdout.write(''.join(arr))
# print(''.join(arr))

  这俩的区别是sys的那种,最后不会加一个换行,而print我们知道,输出完毕之后,会自动加一个换行符。建议还是print吧,花里胡哨的多了,就容易出错。

   有了这样的一个框架,就把ACM模式转成了我们常用的LeetCode模式了。

 

 

参考资料

笔试编程框架总结!

 

上一篇:一文教你学会python读取文本及字符串常用操作


下一篇:20201216-1 文件读与写详解3