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

---

数据读取问题

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

例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 匿名函数

# 提供了两个列表，对相同位置的列表数据进行相加>>> map(lambda x, y: x + y, [1, 3, 5, 7, 9], [2, 4, 6, 8, 10])

输出：

[1, 4, 9, 16, 25]

[1, 4, 9, 16, 25] 

[3, 7, 11, 15, 19]