问题说明

求一个3×3矩阵对角线元素之和。

输入格式

输入分四行：
    第一行：一个正整数K(1<=k<=3)
    接下来三行：3×3矩阵

输出格式

输出只有一行：以空格分隔的第K行元素

3
1 2 3
1 1 1
3 2 1

3 2 1

来源/分类

⭐ Python