lavos/cap3/ex22.go

package cap3

// Ex22 魔方阵，罗伯法
func Ex22(n int) [][]int {
	matrix := _makeMatrix(n)
	if n%2 == 0 {
		return matrix
	}

	// 起始位置在第一行的中间
	i, j := 0, n/2

	for num := 1; num <= n*n; num++ {
		matrix[i][j] = num // 填充数字
		num++              // 下一个数字
		i--                // 向上移动
		j++                // 向右移动

		// 如果移动后超出了上边界和右边界
		if i < 0 && j > n-1 {
			i += 2 // 向下移动两格
			j--    // 向左移动一格
		}

		// 如果超出了上边界
		if i < 0 {
			i = n - 1 // 移动到最下面一行
		}

		// 如果超出了右边界
		if j > n-1 {
			j = 0 // 移动到最左边一列
		}

		// 如果当前位置已经填充了数字
		if matrix[i][j] != 0 {
			i += 2 // 向下移动两格
			j--    // 向左移动一格
		}
	}

	return matrix
}