PTA 矩阵交换行 (python)

矩阵交换行

作者 于延
单位 哈尔滨师范大学

任务描述

给定一个5*5的矩阵（数学上，一个r×c的矩阵是一个由r行c列元素排列成的矩形阵列），将第n行和第m行交换，输出交换后的结果。

输入格式:

输入共6行，前5行为矩阵的每一行元素,元素与元素之间以一个空格分开。
第6行包含两个整数m、n，以一个空格分开。（1 <= m,n <= 5）

输出格式:

输出交换之后的矩阵，矩阵的每一行元素占一行，元素之间以一个空格分开。

输入样例:

1 2 2 1 2
5 6 7 8 3
9 3 0 5 3
7 2 1 4 6
3 0 8 2 4
1 5

输出样例:

3 0 8 2 4
5 6 7 8 3
9 3 0 5 3
7 2 1 4 6
1 2 2 1 2

思路：

先按行读取矩阵，将读取的矩阵存入二维列表中，因为需要交换的行也是由用户输入的，所以指定m,n交换即可。

list2 = [[] for i in range(5)]
for j in range(5):
    list1 = list(input().split())
    list2[j] = list1
# print(list2)
m , n = map(int,input().split())
list2[m-1] , list2[n-1] = list2[n-1] , list2[m-1]
for i in range(len(list2)):
    print(' '.join(list2[i]))