访问所有节点的最短路径

Posted 2021-09-03 秦枫-_-

关键点就是二进制位来记录某一位结点是否访问 关键就是state[][]数组的设置，然后用队列维护状态

class Solution {
    public int shortestPathLength(int[][] graph) {
    int n=graph.length;
    boolean [][]state=new boolean[n][1<<n];//表示访问到当前i结点时，总共访问了多少个结点，二进制i位为1表示i结点已经被访问了
    Queue<int[]> q=new LinkedList<>();//维护所有的访问可能情况
    for(int i=0;i<n;i++){
        q.offer(new int[]{i,1<<i,0});//初始状态下路径长度都为1
        state[i][1<<i]=true;//初始状态下，所有结点都是cur=i，curstate=1<<i，表示当前结点被访问
    }

    while(!q.isEmpty()){
        int []top=q.poll();//top[0]表示当前结点，top[1]表示当前访问状态，top[2]表示当前状态下的路径长度
        if(top[1]==(1<<n)-1){//(1<<n)-1)表示111111，所有二进制位都是1，表示所有结点都被访问了，那么可以直接跳出结果
         return top[2];
        }
        for(int cur:graph[top[0]]){//依次访问当前结点的关联边
            int curstate=top[1]|(1<<cur);//记录当前cur结点已经被访问
            if(!state[cur][curstate]){//更新状态
                q.offer(new int[]{cur,curstate,top[2]+1});
                state[cur][curstate]=true;
            }
        }
    }
    return 0;
    }
}