蓝桥杯题目_九宫幻方

   小明最近在教邻居家的小朋友小学奥数，而最近正好讲述到了三阶幻方这个部分，
三阶幻方指的是将1~9不重复的填入一个3*3的矩阵当中，使得每一行、每一列和每一条对角线的和都是相同的。
    
    三阶幻方又被称作九宫格，在小学奥数里有一句非常有名的口诀：“二四为肩，六八为足，左三右七，戴九履一，五居其中”，通过这样的一句口诀就能够非常完美的构造出一个九宫格来。
    
4 9 2
3 5 7
8 1 6

    有意思的是，所有的三阶幻方，都可以通过这样一个九宫格进行若干镜像和旋转操作之后得到。现在小明准备将一个三阶幻方（不一定是上图中的那个）中的一些数抹掉，交给邻居家的小朋友来进行还原，并且希望她能够判断出究竟是不是只有一个解。
    
    而你呢，也被小明交付了同样的任务，但是不同的是，你需要写一个程序~

输入格式：
输入仅包含单组测试数据。
每组测试数据为一个3*3的矩阵，其中为0的部分表示被小明抹去的部分。
对于100%的数据，满足给出的矩阵至少能还原出一组可行的三阶幻方。

输出格式：
如果仅能还原出一组可行的三阶幻方，则将其输出，否则输出“Too Many”（不包含引号）。

样例输入
0 7 2
0 5 0
0 3 0

样例输出
6 7 2
1 5 9
8 3 4

 

public class 九宫幻方 {

	public static int a[] = new int[10];
	public static int book[] = new int[10];
	public static int b[] = new int[10];
	public static int copyB[] = new int[10]; // 拷贝数组B。用来还原B数组
	public static LinkedHashSet<Integer> linkedHashSet = new LinkedHashSet<Integer>(); // 用于存储结果
	public static String string = "";

	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);

		int i;
		for (i = 1; i <= 9; i++) {
			b[i] = scanner.nextInt();
			copyB[i] = b[i];
		}

		for (i = 1; i <= 9; i++) {
			if (b[i] != 0)
				string += b[i];
		}

		dfs(1);

		int count = 1;

		if (linkedHashSet.size() > 0) {
			// 打印结果
			for (Iterator iterator = linkedHashSet.iterator(); iterator.hasNext();) {
				Integer integer = (Integer) iterator.next();
				System.out.print(integer + "  ");

				if (count % 3 == 0)
					System.out.println();

				count++;
			}
		}
		// 如果不存在则输出“Too Many”
		else {
			System.out.println("Too Many");
		}

	}

	private static void dfs(int step) {
		int i;
		if (step == 10) {
			for (i = 1; i <= 9; i++) {
				// 0的部分表示被小明抹去的部分。获取不能填入已经存在的数值
				if (b[i] != 0 || string.contains(a[i] + ""))
					continue;

				b[i] = a[i];
			}

			int h1 = b[1] + b[2] + b[3];
			int h2 = b[4] + b[5] + b[6];
			int h3 = b[7] + b[8] + b[9];
			int l1 = b[1] + b[4] + b[7];
			int l2 = b[2] + b[5] + b[8];
			int l3 = b[3] + b[6] + b[9];
			int zhu = b[1] + b[5] + b[9];
			int fu = b[3] + b[5] + b[7];

                    //每一行、每一列和每一条对角线的和都是相同的。
			if (h1 == h2 && h1 == h3 && h1 == l1 && h1 == l2 && h1 == l3 && h1 == zhu && h1 == fu) {
				for (i = 1; i <= 9; i++)
					linkedHashSet.add(b[i]);

			}

			// 还原数组
			for (i = 1; i <= 9; i++)
				b[i] = copyB[i];

			return;
		}

		for (i = 1; i <= 9; i++) {

			if (book[i] == 0) {

				book[i] = 1;
				a[step] = i;
				dfs(step + 1);
				book[i] = 0;
			}

		}

	}

}