golang实现最短Hamilton路径算法

题目

给定一张 n 个点的带权无向图，点从 0~n-1 标号，求起点 0 到终点 n-1 的最短Hamilton路径。 Hamilton路径的定义是从 0 到 n-1 不重不漏地经过每个点恰好一次。
输入格式
第一行输入整数n。
接下来n行每行n个整数，其中第i行第j个整数表示点i到j的距离（记为a[i,j]）。
对于任意的x,y,z，数据保证 a[x,x]=0，a[x,y]=a[y,x] 并且 a[x,y]+a[y,z]>=a[x,z]。

输出格式
输出一个整数，表示最短Hamilton路径的长度。

数据范围
1≤n≤20
0≤a[i,j]≤10^7
输入样例：

5
0 2 4 5 1
2 0 6 5 3
4 6 0 8 3
5 5 8 0 5
1 3 3 5 0

输出样例：

18

/*
举例说明：如果是4个点
0->1->2->3
0->2->1->3
...总方案是3！，剩下的4种方法略...
假设第一种走法的方案，距离是10，第二种走法是20，
则第二种走法后面的点再怎么走的方案也就不被采纳，并且后面点的方法是可以直接套在第一种方案后的。
所以我们就可以只关注两点：
1、那些点是走过的
2、现在走在了哪个点上
状态表示：
f[status][j] 第一维表示走过的所有点，第二维度表示现在落在哪个点上
状态转移：
f[status][j]=f[status_k-j][k]+weight[k][j]  k表示当前走到的j点的前一步k点，并且k点的所有状态集合status_k中不包含j点
status的状态使用二进制状态压缩，1，0分别表示走过和没走过
*/

package main

import (
    "fmt"
    
)
const (N=20
M=1<<20)
var (n int
f[M][N] int
weight[N][N] int
)
func min(a,b int)int{
    if a>=b{
        return b
    }
    return a
}
func main(){
    fmt.Scan(&n)
    // 输入
    for i:=0;i<n;i++{
        for j:=0;j<n;j++{
            fmt.Scan(&weight[i][j])
        }
    }
    // 初始化f数组
    for i:=0;i<M;i++{
        for j:=0;j<N;j++{
            f[i][j]=0x3f3f3f3f
        }
    }
    f[1][0] = 0 // 在起点的状态，还没走，所以距离是0
    //
    for i:=0;i<1<<n;i++{
        for j:=0;j<n;j++{
            if i>>j &1 >0{ // 查看集合i中的j点有没有走过，只有走过的点的集合才是可以状态转移的合法状态
                for k:=0;k<n;k++{ //枚举所有的整数
                    if i-(1<<j)>>k &1 >0{ //当前状态i集合还不包括j点，所以集合中要减去，减去后的集合也得满足合法的状态转移条件
                    // 所以k点是已经走过的，二进制1表示。
                        f[i][j]=min(f[i][j],f[i-(1<<j)][k]+weight[k][j])
                    }
                }
            }
        }
    }
    fmt.Println(f[(1<<n)-1][n-1])
    
}