矩阵转置

题意

输出 $n\times m$ 矩阵的转置，即将行和列互换。

思路

zip(*matrix) 将所有行解包并按相同列下标组合，每个得到的元组就是转置矩阵的一行。

代码

cpp

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23#include <cstdio>

int n,m,k;
int a[200][200];
int b[200][200];
int c[200][200];
int main(){
    int i,j;
    scanf("%d%d",&n,&m);
    for (i=1;i<=n;i++){
        for (j=1;j<=m;j++){
            scanf("%d",&a[i][j]);
        }
    }

    for (j=1;j<=m;j++){
        for (i=1;i<=n;i++){
            printf("%d ",a[i][j]);
        }
        printf("\n");
    }
    return 0;
}

复杂度

总结

Python代码

python

1
2
3
4
5row_count, column_count = map(int, input().split())
matrix = [list(map(int, input().split())) for _ in range(row_count)]

for column in zip(*matrix):
    print(*column)

C++代码

cpp

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23#include <cstdio>

int n,m,k;
int a[200][200];
int b[200][200];
int c[200][200];
int main(){
    int i,j;
    scanf("%d%d",&n,&m);
    for (i=1;i<=n;i++){
        for (j=1;j<=m;j++){
            scanf("%d",&a[i][j]);
        }
    }

    for (j=1;j<=m;j++){
        for (i=1;i<=n;i++){
            printf("%d ",a[i][j]);
        }
        printf("\n");
    }
    return 0;
}

复杂度

时间复杂度为 $O(nm)$ ，矩阵输入空间为 $O(nm)$ 。

总结

zip(*matrix) 是规则二维列表转置的简洁 Python 写法。

目录

题意

思路

代码

复杂度

总结

Python代码

C++代码

复杂度

总结