今天做了新浪的产品题,有两道编程题。这是第一次在产品岗做到编程题,也是第一次用牛客网编程,先开始看到有点懵逼,所以查询学习了一下,希望下次答题更加从容,也希望可以帮到有需要的同学~
数据读取问题
以题目给的两个示例为例。
例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 匿名函数