【蓝桥杯】回路计数 java实现

回路计数

题目描述

蓝桥学院由 ​​ 21栋教学楼组成，教学楼编号 ​​​1 到 21​​​。对于两栋教学楼 a​​​ 和 b​​，当 a​​ 和 b​​ 互质时，a​ 和 b 之间有一条走廊直接相连，两个方向皆可通行，否则没有直接连接的走廊。小蓝现在在第一栋教学楼，他想要访问每栋教学楼正好一次，最终回到第一栋教学楼（即走一条哈密尔顿回路），请问他有多少种不同的访问方案？两个访问方案不同是指存在某个 i​，小蓝在两个访问方法中访问完教学楼 i​ 后访问了不同的教学楼。

/*
该题涉及到使用辗转相除法求两数的最大公约数，动态规划和状态压缩
误区：
    此题容易想当然地进行深搜，但经简单思考可知若进行深搜需执行21的21次方次，无法实现。
    
那么就换一种思路，因为用上面的方法会进行大量重复的判断，所以思考如何避免这些操作？那就是由果溯因。
此题的结果就是21栋楼都被遍历了一遍，一个关键点是1与其他20个数都互质，所以说，当所有楼都被遍历一边时，不管最后在哪栋楼，都可以回到1号楼。而怎么去标识已经被遍历到的楼呢？这是另一个关键点，我们就想到使用状态压缩，将这21幢楼是否已经被遍历用0,1表示，压缩成一个21位的二进制数。关于状态转移方程，我们设f[i][h],i表示房屋的遍历状态，h表示到达第h幢房屋（代码中以下标0开始），则f[i][j]为在i状态下，最后到达h房屋的路径数。
那么我们要想得出f[i][j]就必须知道从哪个房屋可以到达j房屋，所以可以由此推导出状态转移方程
                        f[i+(1<<h)][h]+=f[i][j]

*/
public static void main(String[] args) {
        int[][] g=new int[21][21];
        for (int i = 0; i < 21; i++) {
            for (int j = 0; j < 21; j++) {
                if(gcd(i+1,j+1)==1) {
                    g[i][j]=1;//因为g[j][i]会在之后的循环遍历到，所以此处只给g[i][j]标记即可
                }
            }
        }
          //注意，此题结果超出int范围
        long[][] f=new long[1<<21][21];
        f[1][0]=1;
        for (int i = 1; i < (1<<21) ; i++) {
            for (int j = 0; j < 21; j++) {
                if((i>>j & 1) == 1) {
                    for (int h = 0; h < 21; h++) {
                        if(g[j][h]==1 && (i>>h & 1)==0) {

//g[j][h]==1表示可以从j到h,(i>>h & 1)==0则表示第h幢房屋还没有被遍历到。此时用f[i+(1<<h)][h]表示(i状态以及h幢也被遍历到）并且当前最后一幢房屋为h，这个状态可以由f[i][j]转移得到
                            f[i+(1<<h)][h]+=f[i][j]; 
                        }
                    }
                }
            }
        }
        long result=0;
        for (int i = 0; i < 21; i++) {
            result+=f[(1<<21) -1][i];

//此处(1<<21) -1表示从0到20的21个位上都为1，也就是所有房屋都被遍历一遍，又因为无论最后在哪个房屋，都可以回到1房屋，所以i从0到20累加得出结果
        }
        System.out.println(result);
          //结果为881012367360
    }
    private static int gcd(int n,int m) {
        return m!=0?gcd(m,n%m):n;
    }

 原题链接：回路计数

本人算法小白，记录我的思考过程以供大家参考，若有错误，还请各位指出，感谢！