【Lintcode】1306. Travel Plan II

题目地址：

https://www.lintcode.com/problem/1306/

有$n$个城市，编号$0\sim n-1$，每两个城市的距离由方阵$A$给出，从城市$0$出发，要求周游$1\sim n-1$一圈再回到城市$0$，每个城市只能访问一次，问最少路程是多少。

递推写法参考https://blog.csdn.net/qq_46105170/article/details/110155730。下面介绍一个记忆化搜索做法。

思路和递推法是一样的，也是状态压缩然后记忆化DFS。把已经访问过的城市压缩成一个整数，其每个二进制位表示某个城市是否被访问过。然后进行DFS，枚举下一个要去的城市是谁。代码如下：

import java.util.Arrays;

public class Solution {
    /**
     * @param arr: the distance between any two cities
     * @return: the minimum distance Alice needs to walk to complete the travel plan
     */
    public int travelPlanII(int[][] arr) {
        // Write your code here.
        int n = arr.length;
        int[][] dp = new int[n][1 << n];
        for (int[] row : dp) {
            Arrays.fill(row, -1);
        }
        
        return dfs(0, 1, dp, arr);
    }
    
    // 返回当前位置是cur，访问过的城市的状态是state的情况下，
    // 访问完剩余城市再回到城市0还需要的走的最小路程
    private int dfs(int cur, int state, int[][] dp, int[][] arr) {
    	// 如果有记忆则调取记忆
        if (dp[cur][state] != -1) {
            return dp[cur][state];
        }
        
        // 如果1 ~ n - 1城市都访问完了，那么只需要回到起点就行了，还需要走的就是cur与0的距离
        if (state == (1 << arr.length) - 1) {
            dp[cur][state] = arr[cur][0];
            return dp[cur][state];
        }
        
        int res = 1 << 30;
        // 接下来枚举要去的下一个城市是哪个
        for (int i = 1; i < arr.length; i++) {
        	// 如果城市i没去过，则枚举之
            if ((state >> i & 1) == 0) {
                res = Math.min(res, arr[cur][i] + dfs(i, state | (1 << i), dp, arr));
            }
        }
        
        // 返回之前存一下记忆
        dp[cur][state] = res;
        return res;
    }
}

时间复杂度$O(n^2{2}^n)$，空间$O(n{2}^n)$。