【ACWing】91. 最短Hamilton路径

题目地址:

https://www.acwing.com/problem/content/93/

给定一个 n n n阶带权无向图,顶点从 0 ∼ n − 1 0\sim n-1 0n1标号,求从起点 0 0 0到终点 n − 1 n-1 n1的Hamilton路径的最短长度。两个顶点之间路径的长度 A [ i ] [ j ] A[i][j] A[i][j]由一个对称方阵给出。

输入格式:
第一行输入整数 n n n。接下来 n n n行每行 n n n个整数,其中第 i i i行第 j j j个整数表示点 i i i j j j的距离(记为 A [ i ] [ j ] A[i][j] A[i][j])。对于任意的 x , y , z x,y,z x,y,z,数据保证 A [ x ] [ x ] = 0 A[x][x]=0 A[x][x]=0 A [ x ] [ y ] = A [ y ] [ x ] A[x][y]=A[y][x] A[x][y]=A[y][x]并且 A [ x ] [ y ] + A [ y ] [ z ] ≥ A [ x ] [ z ] A[x][y]+A[y][z]\ge A[x][z] A[x][y]+A[y][z]A[x][z]

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

数据范围:
1 ≤ n ≤ 20 1\le n\le 20 1n20
0 ≤ A [ i , j ] ≤ 1 0 7 0\le A[i,j]\le 10^7 0A[i,j]107

思路是动态规划。可以用状态压缩。设 f [ i ] [ j ] f[i][j] f[i][j]是当前访问顶点状态是 i i i并且当前恰好位于点 j j j的情况下,所走的最短距离。这里 i i i的各个二进制位表示每个标号的顶点是否访问过。按照走到 j j j之前上一个访问的顶点编号是多少分类,则有: f [ i ] [ j ] = min ⁡ i > > k & 1 = 1 , k ≠ j ( f [ i − ( 1 < < j ) ] [ k ] + A [ k ] [ j ] ) f[i][j]=\min_{i>>k\&1=1,k\ne j}(f[i-(1<f[i][j]=i>>k&1=1,k=jmin(f[i(1<<j)][k]+A[k][j])这里的 k k k的条件是,首先 k ≠ j k\ne j k=j,即要从一个不同于 j j j的地方走过来,其次, i i i里走过的顶点要包含 k k k。初始条件 f [ 1 ] [ 0 ] = 0 f[1][0]=0 f[1][0]=0,意味着从起点出发的时候总距离是 0 0 0。代码如下:

#include 
#include 
using namespace std;

const int N = 21, M = 1 << N;

int n;
int w[N][N], f[M][N];

int main() {
    scanf("%d", &n);
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++)
            scanf("%d", &w[i][j]);

    memset(f, 0x3f, sizeof f);
    f[1][0] = 0;

    for (int i = 0; i < 1 << n; i++) 
        for (int j = 0; j < n; j++)
            if (i >> j & 1)
                for (int k = 0; k < n; k++)
                    if (k != j && i >> k & 1)
                        f[i][j] = min(f[i][j], f[i - (1 << j)][k] + w[k][j]);

    printf("%d\n", f[(1 << n) - 1][n - 1]);

    return 0;
}

时间复杂度 O ( n 2 2 n ) O(n^22^n) O(n22n),空间 O ( n 2 n ) O(n2^n) O(n2n)

你可能感兴趣的:(AC,贪心与动态规划,算法,动态规划,c++,图论)