题目
反幻方
我国古籍很早就记载着
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);
}
}