蓝桥杯 15决赛 A2 四阶幻方(剪枝)

蓝桥杯 15决赛 A2 四阶幻方(剪枝)

标题：四阶幻方
把1~16的数字填入4x4的方格中，使得行、列以及两个对角线的和都相等，满足这样的特征时称为：四阶幻方。
四阶幻方可能有很多方案。如果固定左上角为1，请计算一共有多少种方案。
比如：
1 2 15 16
12 14 3 5
13 7 10 4
8 11 6 9
以及：
1 12 13 8
2 14 7 11
15 3 10 6
16 5 4 9
就可以算为两种不同的方案。
请提交左上角固定为1时的所有方案数字，不要填写任何多余内容或说明文字。

======================================

思路:
dfs
剪枝(这题剪枝真的很重要,快很多)

/*
 @author weng
 @build  2020年7月21日 上午10:48:30
*/

public class 剪枝_幻方_2 {
	static boolean[] b=new boolean[17];
	static int[] a =new int[16];
	static int ans;
	public static void main(String[] args) {
		a[0] =1;
		dfs(1);
		System.out.println(ans);
	}
	private static void dfs(int lev) {
		if(lev ==16) {
			ans ++;
			return ;
		}
		for(int i =2 ;i <=16 ;i ++) {
			if(b[i]) continue;
			a[lev] =i;
			//剪枝
			if((lev +1) %4 ==0) {
				if(34 !=a[lev] +a[lev -1] +a[lev -2] +a[lev -3]) continue;
			}
			if(lev >11) {
				if(34 !=a[lev] +a[lev -4] +a[lev -8] +a[lev -12]) continue;
			}
			if(lev ==12) {
				if(a[3] +a[6] +a[9] +a[12] !=34) continue ;
			}
			if(lev ==15) {
				if(a[0] +a[5] +a[10] +a[15] !=34) continue ;
			}
			b[i] =true;
			dfs(lev +1);
			b[i] =false;
		}
	}
}