蓝桥杯决赛 Java大学B组 第二题 反幻方

题目

反幻方

我国古籍很早就记载着

2 9 4
7 5 3
6 1 8

这是一个三阶幻方。每行每列以及对角线上的数字相加都相等。

下面考虑一个相反的问题。
可不可以用 1~9 的数字填入九宫格。
使得：每行每列每个对角线上的数字和都互不相等呢？

这应该能做到。
比如：
9 1 2
8 4 3
7 5 6

你的任务是搜索所有的三阶反幻方。并统计出一共有多少种。
旋转或镜像算同一种。

比如：
9 1 2
8 4 3
7 5 6

7 8 9
5 4 1
6 3 2

2 1 9
3 4 8
6 5 7

等都算作同一种情况。

请提交三阶反幻方一共多少种。这是一个整数，不要填写任何多余内容。

思路

最简单的办法就是全排列之后对每一种情况进行验证。
我用的是字典序全排列加上Set去重验证。

代码

import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

public class Main{
	static int data[]={1,2,3,4,5,6,7,8,9};
	static int length=data.length;
	static int cnt=0;
	public static void main(String[] args) {
		check();
		while(nextPermutation()){
			check();
		}
		System.out.println(cnt/8);
	}
	public static void check(){
		int sum1=data[0]+data[1]+data[2];
		int sum2=data[3]+data[4]+data[5];
		int sum3=data[6]+data[7]+data[8];
		int sum4=data[0]+data[3]+data[6];
		int sum5=data[1]+data[4]+data[7];
		int sum6=data[2]+data[5]+data[8];
		int sum7=data[0]+data[4]+data[8];
		int sum8=data[2]+data[4]+data[6];
		Set<Integer> set=new HashSet<Integer>();
		set.add(sum1);
		set.add(sum2);
		set.add(sum3);
		set.add(sum4);
		set.add(sum5);
		set.add(sum6);
		set.add(sum7);
		set.add(sum8);
		if(set.size()==8) cnt++;
	}
	public static boolean nextPermutation(){
		int sp1=length-1,sp2=length-1;
		while(sp1>0&&data[sp1]<=data[sp1-1]){
			sp1--;
		}
		if(sp1==0) return false;
		else{
			while(sp2>0&&data[sp2]<=data[sp1-1]) sp2--;
			swap(sp1-1,sp2);
			reverse(sp1,length-1);
			return true;
		}
	}
	public static void swap(int x,int y){
		int temp=data[x];
		data[x]=data[y];
		data[y]=temp;
	}
	public static void reverse(int x,int y){
		for(int i=x,j=y;i<j;i++,j--)
			swap(i,j);
	}
}