UVA1599-最短路+字典序-双重BFS

本来不想写了，想到明天早上还要负责早签。。。但是怕过一段时间就忘了思路了，趁热打铁吧，赶紧写完，赶紧睡。

题目链接：https://vjudge.net/problem/UVA-1599

题意：n个节点的无向图，m条路，每条路有个序号。找起点到终点的最短路且路的字典序要最小。

分析：

刚看到这道题懵了，要是用静态数组存图，需要十万乘十万的矩阵，显然不可能，占用的内存都以GB计算了。然后一直没思路。最后参考了一下Rujia Liu大神的代码，感觉各种被秀，受益匪浅。

他用动态数组存下来的，显然OJ不会输入那么多条路，便节省了不少空间。

这道题需要两次BFS，第一次反向BFS，存下每个节点到终点的最小路长度。第二次BFS从起点开始找距离为当前距离减一的下一个节点，同时要满足字典序最小。这道题可能就是这个BFS是难点了。需要用到两个数组。

Attention 算是给自己的attention吧，代码刚写完没有把所有n节点改成n-1，导致错误。 

代码：

#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>

using namespace std;
const int maxn=100000,INF=1000000000;
struct Edge
{
    int v,c;
    Edge(int v,int c):v(v),c(c){}
};

vector<int> G[maxn];
vector<Edge> e;
vector<int>ans;
bool vis[maxn];
int d[maxn],n,m;

void rev_bfs()
{
    memset(vis,0,sizeof(vis));
    memset(d,0,sizeof(d));

    queue<int> q;
    q.push(n-1);
    vis[n-1]=1;

    int v;
    while(!q.empty())
    {
        v=q.front();
        q.pop();
        for(int u:G[v])
        {
            if(!vis[e[u].v])
            {
                d[e[u].v]=d[v]+1;
                vis[e[u].v]=true;
                q.push(e[u].v);
            }
        }
    }
}
void bfs()
{
    int m_color;
    memset(vis,0,sizeof(vis));
    ans.clear();
    vector<int>next;
    next.push_back(0);

    for(int i=d[0];i>0;--i)
    {
        m_color=INF;
        for(int u:next)
            for(int v:G[u])
            {
                if(d[e[v].v]==i-1)
                    m_color=min(m_color,e[v].c);
            }
        ans.push_back(m_color);
        vector<int>next1;
        for(int u:next)
            for(int v:G[u])
            {
                if(e[v].c==m_color&&d[e[v].v]==i-1&&!vis[e[v].v])
                {
                    next1.push_back(e[v].v);
                    vis[e[v].v]=true;
                }
            }
        next=next1;
    }
}
int main()
{
    while(~scanf("%d",&n))
    {
        scanf("%d",&m);
        for(int i=0;i<n;++i)
            G[i].clear();
        e.clear();
        for(int i=0;i<m;++i)
        {
            int u,v,c;
            scanf("%d%d%d",&u,&v,&c);
            e.push_back(Edge(v-1,c));
            G[u-1].push_back(e.size()-1);
            e.push_back(Edge(u-1,c));
            G[v-1].push_back(e.size()-1);
        }
        rev_bfs();
        bfs();
        printf("%d\n",d[0]);
        for(decltype(ans.size()) i=0;i<ans.size();++i)
        {
            if(i)
               putchar(' ');
            printf("%d",ans[i]);
        }
        putchar('\n');
    }
    return 0;
}