magicsquare.cpp

来自「一个外国网站上的幻方源码。希望对大家能有所帮助。」· C++ 代码 · 共 85 行

// MagicSquare.cpp : Defines the entry point for the console application.
//

#include <iostream>
#include <stdio.h>
#include <memory.h>
#include <vector>
#include "conio.h"

using namespace std;

void OddMagicSquare(vector<vector<int> > &matrix, int n);
void MagicSquare(vector<vector<int> > &matrix, int n);
void PrintMagicSquare(vector<vector<int> > &matrix, int n);

int main()
{
	int n;
	printf("Enter order of square: ");
	cin>>n;

	vector<vector<int> > matrix(n, vector<int> (n, 0));

	if (n<3)
	{
		printf("\nError: n must be greater than 3\n\n");
		return -1;
	}

	MagicSquare(matrix, n);	

	//Print results
	PrintMagicSquare(matrix, n);
		
	return 0;
}

void MagicSquare(vector<vector<int> > &matrix,int n)
{
	if (n%2==1)				//n is Odd
		OddMagicSquare(matrix, n);
	else					//n is even
		cout<<"you must order a odd!";
}

void OddMagicSquare(vector<vector<int> > &matrix, int n)
{
	int nsqr = n * n;
	int i=0, j=n/2;     // start position

	for (int k=1; k<=nsqr; ++k) 
	{
		matrix[i][j] = k;

		i--;
		j++;

		if (k%n == 0) 
		{ 
			i += 2; 
			--j; 
		}
		else 
		{
			if (j==n) 
				j -= n;
			else if (i<0) 
				i += n;
		}
	}
}

void PrintMagicSquare(vector<vector<int> > &matrix, int n)
{
	for (int i=0; i<n; i++) 
	{
		for (int j=0; j<n; j++)
			printf(" %d", matrix[i][j]);

		printf("\n");
	}

	printf("\n\n");
}