BFS题：颜色交替的最短路径

BFS题：颜色交替的最短路径

问题：

思路：

颜色交替的解决办法：
我们每次传递时，将上一次遍历的边的颜色信息带上，在下一次遍历时，选择不同的颜色的邻接节点,节点信息在队列中的存储为（node,color)，即既存储节点的编号，又存储上一次遍历边的颜色信息

由于存在环和平行边，使用数组visit[x][y][color] = 1,代表从节点x到节点y的且颜色为color的边被访问过，防止重复访问。

对红色和蓝色边分别使用map建立邻接表，初始放入队列的信息为（0,1）和（0，0）,分别表示从起点开始，下一次边颜色为蓝和为红（上一条边颜色为红和为蓝两种情况）

在BFS函数中需判断队列头保存的上一条边的颜色，若为红，在蓝色边邻接表中进行遍历，向队列中放入新元素时，颜色信息改变为蓝色。
上一条边的信息为蓝色时处理过程类似，即在红色边邻接表中进行遍历

代码：

class Solution {
public:
    vector<int> shortestAlternatingPaths(int n, vector<vector<int>>& red_edges, vector<vector<int>>& blue_edges) {
        unordered_map<int,vector<int>> red;
        unordered_map<int,vector<int>> blue;
        int visited[100][100][2];
        int step=0;
        int i;
        vector<int> ans( n,1e+06 );
        queue< pair<int,int> > q;
        for( auto temp:red_edges )
            red[ temp[0] ].push_back( temp[1] );
        for( auto temp:blue_edges )
            blue[ temp[0] ].push_back( temp[1] );
        memset( visited,0,sizeof(visited) );
        q.push( make_pair( 0,1 ) );
        q.push( make_pair( 0,0 ) );
        while( !q.empty() )
        {   
            step++;
            int size=q.size();
            for( i=0;i<size;i++ )
            {
                int cur=q.front().first;
                int color=q.front().second;
                q.pop();
                if( color==1 )
                {
                    for( auto temp:blue[cur] )
                    {
                        if( visited[cur][temp][0]==0 )
                        {
                            ans[temp]=min( ans[temp],step );
                            visited[cur][temp][0]=1;
                            q.push( make_pair(temp,0) );
                        }
                    }
                }
                else
                {
                    for( auto temp:red[cur] )
                    {
                        if( visited[cur][temp][1]==0 )
                        {
                            ans[temp]=min( ans[temp],step );
                            visited[cur][temp][1]=1;
                            q.push( make_pair(temp,1) );
                        }
                    }
                }
            }
        }
        ans[0]=0;
        for( i=1;i<n;i++ )
            if( ans[i]>=1e+05 )
                ans[i]=-1;
        return ans;
    }
};