牛客网python编程题小白入门指南:数据读取

今天做了新浪的产品题,有两道编程题。这是第一次在产品岗做到编程题,也是第一次用牛客网编程,先开始看到有点懵逼,所以查询学习了一下,希望下次答题更加从容,也希望可以帮到有需要的同学~


数据读取问题

以题目给的两个示例为例。

例1:求a+b的和

输入描述: 多组读入,每行有两个数。

python3代码:

import sys

for line in sys.stdin:

a = line.split()

print(int(a[0]) + int(a[1]))

解析: 第一行代码:导入sys模块,sys模块包含了与Python解释器和它的环境有关的函数。第二行代码:意思是“对于输入文件中的行”,此例中输入文件中只有一行。第三行代码:对字符串进行切片。对此处,若输入为“ 1 1”,则是将其切分为“1”和“1”。第四行代码:求和。

例2:给出n阶方阵里所有数,求方阵里所有数的和。

输入描述: 输入有多个测试用例,每个测试用例第一行为一个整数n(n<=1000),表示方阵阶数为n。接下来时n行的数字,每行n个数用空格隔开。

python3代码:

import sys

if _name_ == "_main_":

# 读取第一行的n

n = int(sys.stdin.readline().strip())

ans = 0

for i in range(n):

# 读取每一行

line = sys.stdin.readline().strip()

# 把每一行的数字分隔后转化成int列表

values = list(map(int, line.split())

for v in values:

ans += v

print (ans)

解析:思路是先读取第一行的阶数n,然后再读取每行数据转化为list,再用一个for循环求和。

知识点:

sys.stdin

sys中的stdin , stdout , 以及stderr 变量包含与标准I/O 流对应的流对象。如果需要更好地控制输出,而print 不能满足你的要求, 它们就是你所需要的。你也可以替换它们, 这时候你就可以重定向输出和输入到其它设备( device ), 或者以非标准的方式处理它们。

Python split()方法

通过指定分隔符对字符串进行切片,如果参数 num 有指定值,则分隔 num+1 个子字符串。

语法: str.split(str="", num=string.count(str))

str : 分隔符,默认为所有的空字符,包括空格、换行(\n)、制表符(\t)等。

num : 分割次数。默认为 -1, 即分隔所有。

返回值: 分割后的字符串列表。

示例:

str = "Line1-abcdef \nLine2-abc \nLine4-abcd";

print str.split( );      # 以空格为分隔符,包含 \n

print str.split(' ', 1 ); # 以空格为分隔符,分隔成两个

['Line1-abcdef', 'Line2-abc', 'Line4-abcd']

['Line1-abcdef', '\nLine2-abc \nLine4-abcd']

Python int() 函数

int() 函数用于将一个字符串或数字转换为整型。

语法: class int(x, base=10)

x -- 字符串或数字。

base -- 进制数,默认十进制。

返回值: 整型数据

sys.stdin.readlines

刷题网站的后台给出的输入是一个含有多行数据的input文件,直接用sys.stdin.readlines()去逐行读取数据即可,此时lines中已经包括了所有一行一行的数据。

import sys

lines = sys.stdin.readlines()

for line in lines:

print(line)

如果第一行给出的是样例数目,需要提前读取,则可以用readline()代替 readlines() 只读取一行,而后面再用 readlines() 读取剩余行时,python是会自动从第二行开始读取的。

n = sys.stdin.readline()

lines = sys.stdin.readlines()

Python strip()方法

Python strip() 方法用于移除字符串头尾指定的字符(默认为空格或换行符)或字符序列。 注意: 该方法只能删除开头或是结尾的字符,不能删除中间部分的字符。只要头尾包含有指定字符序列中的字符就删除。

语法: str.strip([chars])

chars -- 移除字符串头尾指定的字符序列。

返回值: 移除字符串头尾指定的字符生成的新字符串。

示例:

str = "00000003210Runoob01230000000";

print str.strip( '0' );  # 去除首尾字符 0

str = "123abcrunoob321"

print (str.strip( '12' ))  # 字符序列为 12

3210Runoob0123

3abcrunoob3

Python map() 函数

map() 会根据提供的函数对指定序列做映射。第一个参数 function 以参数序列中的每一个元素调用 function 函数,返回包含每次 function 函数返回值的新列表。

语法: map(function, iterable, ...)

function -- 函数

iterable -- 一个或多个序列

返回值: Python 2.x 返回列表。Python 3.x 返回迭代器。

示例:

def square(x) : return x ** 2...

map(square, [1,2,3,4,5])  # 计算列表各个元素的平方

map(lambda x: x ** 2, [1, 2, 3, 4, 5])  # 使用 lambda 匿名函数