【编程题】构造两两相邻数之和为奇数的矩阵

liuz_notes

于 2022-08-30 14:30:07 发布

阅读量1.2k

点赞数

分类专栏：算法文章标签：矩阵 python 算法 leetcode 模拟

本文链接：https://blog.csdn.net/liuz_notes/article/details/126562811

版权

算法专栏收录该内容

3 篇文章 0 订阅

订阅专栏

题目描述

构造一个n行n列的矩阵，矩阵中元素从1到 $n^2$ ，每个数只出现一次，矩阵中每相邻两个数之和为奇数。

输入输出描述

输入一个正整数n， $1\le n \le 100$ 。

3

输出n行，每行n个正整数，有多解时输出任意合法解。

1 2 3
4 5 6
7 8 9

代码

def func(n):
    row = col = n
    x = [[0] * n for i in range(n)]
    a, b = 1, 2
    for i in range(row * col):
        if (i // row + i % row) % 2 == 0:
            x[i//row][i%row] = a
            a += 2
        else:
            x[i // row][i % row] = b
            b += 2
    return x

n = int(input())
res = func(n)
for row in res:
    tmp = []
    for x in row:
        tmp.append(str(x))
    print(' '.join(tmp))