幻方：长度的任何行，列或对角线的总和始终等于相同的数字。所有9个数字都是不同的正整数。
我正在JavaScript中这样做，但是生成所有代码的最佳方法是什么？
function getMagicSquare() {
let myArray = [
[4, 9, 2],
[3, 5, 7],
[8, 1, 5]
];
for (let index1 = 1; index1 < 10; index1++) {
for (let index2 = 1; index2 < 10; index2++) {
for (let index3 = 1; index3 < 10; index3++) {
for (let index4 = 1; index4 < 10; index4++) {
for (let index5 = 1; index5 < 10; index5++) {
for (let index6 = 1; index6 < 10; index6++) {
for (let index7 = 1; index7 < 10; index7++) {
for (let index8 = 1; index8 < 10; index8++) {
for (let index9 = 1; index9 < 10; index9++)
// if numbers are not distinct for each loop, I can break the loop and make it a bit faster
{
const mySet = new Set();
mySet.add(index1).add(index2).add(index3).add(index4).add(index5).add(index6).add(index7).add(index8).add(index9)
if ((mySet.size === 9))
if (
(index1 + index2 + index3 === index4 + index5 + index6) &&
(index4 + index5 + index6 === index7 + index8 + index9) &&
(index7 + index8 + index9 === index1 + index4 + index7) &&
(index1 + index4 + index7 === index2 + index5 + index8) &&
(index2 + index5 + index8 === index3 + index6 + index9) &&
(index3 + index6 + index9 === index1 + index5 + index9) &&
(index1 + index5 + index9 === index3 + index5 + index7)
) {
myArray[0][0] = index1;
myArray[0][1] = index2;
myArray[0][2] = index3;
myArray[1][0] = index4;
myArray[1][1] = index5;
myArray[1][2] = index6;
myArray[2][0] = index7;
myArray[2][1] = index8;
myArray[2][2] = index9;
console.log(myArray);
}
}
}
}
}
}
}
}
}
}
}
第二个问题：如果我想生成NxN个魔术方块怎么办？
答案 0 :(得分：1)
这是一个非常幼稚的实现，它使用状态空间搜索和基本修剪功能来生成给定尺寸n：https://ideone.com/0aewnJ的所有可能的幻方
from collections import defaultdict, deque
from copy import copy, deepcopy
import time
def magicSum(n):
return int((n*n * (n*n + 1)) / 6)
def validate(sumDict, n):
for k, v in sumDict.items():
if v > magicSum(n):
return False
return True
def check(sumDict, n):
for k, v in sumDict.items():
if v != magicSum(n):
return False
return True
def isValid(m, n):
rowSum = defaultdict(int)
colSum = defaultdict(int)
diagSum = defaultdict(int)
isLeft = False
for i in range(n):
for j in range(n):
if m[i][j] == 0: isLeft = True
rowSum[i] += m[i][j]
colSum[j] += m[i][j]
if i == j: diagSum[0] += m[i][j]
if i + j == n - 1: diagSum[n - 1] += m[i][j]
if isLeft:
return (validate(rowSum, n) and validate(colSum, n) and validate(diagSum, n))
return (check(rowSum, n) and check(colSum, n) and check(diagSum, n))
def next(cur, m, n):
possible = []
for i in range(n):
for j in range(n):
if m[i][j] == 0:
nextM = deepcopy(m)
nextM[i][j] = cur
if isValid(nextM, n):
possible.append(nextM)
return possible
def printM(m):
for i in range(len(m)):
print(m[i])
print("\n")
def gen(n):
startM = [[0 for x in range(n)] for y in range(n)]
magic = []
Q = deque([(1, startM)])
while len(Q):
state = Q.popleft()
cur = state[0]
m = state[1]
if cur == n * n + 1:
magic.append(m)
printM(m)
continue
for w in next(cur, m, n):
Q.append((cur + 1, w))
return magic
start_time = time.time()
magic = gen(3)
elapsed_time = time.time() - start_time
print("Elapsed time: ", elapsed_time)
输出：
[6, 1, 8]
[7, 5, 3]
[2, 9, 4]
[8, 1, 6]
[3, 5, 7]
[4, 9, 2]
[6, 7, 2]
[1, 5, 9]
[8, 3, 4]
[8, 3, 4]
[1, 5, 9]
[6, 7, 2]
[2, 7, 6]
[9, 5, 1]
[4, 3, 8]
[4, 3, 8]
[9, 5, 1]
[2, 7, 6]
[2, 9, 4]
[7, 5, 3]
[6, 1, 8]
[4, 9, 2]
[3, 5, 7]
[8, 1, 6]
Elapsed time: 13.479725122451782
尽管我必须说它在运行时间方面比预期的要差一些，但我认为它对于开始还是很有好处的。稍后将尝试进一步优化它。
答案 1 :(得分：1)
下面是用于生成所有8个3x3幻方的Javascript实现。
使用的算法：
function generateMagic3x3(n) {
let i, j;
i = Math.floor(n / 2);
j = n - 1;
let baseMatrix = [
[],
[],
[]
];
baseMatrix[i][j] = 1;
for (let k = 2; k <= n * n; k++) {
i -= 1;
j += 1;
if (i < 0 && j === n) {
i = 0;
j = n - 2;
} else if (i < 0) {
i = n - 1;
} else if (j === n) {
j = 0;
}
if (typeof baseMatrix[i][j] === 'number') {
i += 1;
j -= 2;
}
baseMatrix[i][j] = k;
}
const baseMatrix2 = reflectDiag(baseMatrix);
renderMatrix(baseMatrix)
renderMatrix(reflectRows(baseMatrix));
renderMatrix(reflectColumns(baseMatrix));
renderMatrix(reflectColumns(reflectRows(baseMatrix)));
renderMatrix(baseMatrix2);
renderMatrix(reflectRows(baseMatrix2));
renderMatrix(reflectColumns(baseMatrix2));
renderMatrix(reflectColumns(reflectRows(baseMatrix2)));
};
function reflectColumns(matrix) {
var newMatrix = matrix.map(function(arr) {
return arr.slice();
});
for (let row = 0; row < matrix.length; row++) {
newMatrix[row][0] = matrix[row][2];
newMatrix[row][2] = matrix[row][0];
}
return newMatrix;
}
function reflectRows(matrix) {
var newMatrix = matrix.map(function(arr) {
return arr.slice();
});
for (let column = 0; column < matrix.length; column++) {
newMatrix[0][column] = matrix[2][column];
newMatrix[2][column] = matrix[0][column];
}
return newMatrix;
}
function reflectDiag(matrix) {
var newMatrix = matrix.map(function(arr) {
return arr.slice();
});
for (let row = 0; row < matrix.length; row++) {
for (let column = 0; column < matrix.length; column++) {
if (row !== column) {
newMatrix[row][column] = matrix[column][row];
}
}
}
return newMatrix;
}
function renderMatrix(matrix) {
const table = document.createElement('table');
let resBox = document.getElementById('res')
for (let row = 0; row < matrix.length; row++) {
const tr = table.insertRow(row);
for (let column = 0; column < matrix.length; column++) {
const cell = tr.insertCell(column);
cell.innerHTML = matrix[row][column];
}
}
resBox.appendChild(table)
}
generateMagic3x3(3);table {
border-collapse: collapse;
display:inline-block;
margin: 10px;
}
td {
border: 1px solid #000;
text-align: center;
width: 50px;
height: 50px;
}<div id='res'></div>