UVa 1599 - Ideal Path（BFS）

来源：互联网发布：巴恩斯2016总决赛数据编辑：程序博客网时间：2024/04/20 07:02

给出n个点，m条边的无向图，每条边有一种颜色，求从结点1到结点n颜色字典序最小的最短路径。

例题，书上给了思路，首先反向BFS，记录结点到n最短距离，然后从起点BFS，每次优先选择字典序最小的颜色的路径。

#include<cstdio>#include<cstring>#include<vector>#include<queue>using namespace std;const int maxn=100010;const int inf=1<<30;int n,vis[maxn],d[maxn];vector<int> g[maxn],ans;struct way{    int u,v,c;    way(int u=0,int v=0,int c=0):u(u),v(v),c(c){}};vector<way> ways;void build(int u,int v,int c){//储存边。    ways.push_back(way(u,v,c));    int i=(int)ways.size()-1;    g[u].push_back(i);}void rev_bfs(){//反向BFS，记录距离。    memset(vis,0,sizeof(vis));    d[n-1]=0;    vis[n-1]=true;    queue<int> q;    q.push(n-1);    while(!q.empty()){        int v=q.front();        q.pop();        for(int i=0;i<g[v].size();++i){            int e=g[v][i];            int u=ways[e].v;            if(!vis[u]){                vis[u]=true;                d[u]=d[v]+1;                q.push(u);            }        }    }}void bfs(){    memset(vis,0,sizeof(vis));    ans.clear();    vis[0]=true;    vector<int> next;    next.push_back(0);    for(int i=0;i<d[0];++i){        int min_color=inf;        for(int j=0;j<next.size();++j){            int u=next[j];            for(int k=0;k<g[u].size();++k){                int e=g[u][k];                int v=ways[e].v;                if(d[u]==d[v]+1)                    min_color=min(min_color,ways[e].c);            }        }        ans.push_back(min_color);        vector<int> next2;        for(int j=0;j<next.size();++j){            int u=next[j];            for(int k=0;k<g[u].size();++k){                int e=g[u][k];                int v=ways[e].v;                if(d[u]==d[v]+1&&!vis[v]&&ways[e].c==min_color){                    vis[v]=true;                    next2.push_back(v);                }            }        }        next=next2;    }    printf("%d\n",(int)ans.size());    for(int i=0;i<ans.size();++i){        if(i) printf(" ");        printf("%d",ans[i]);    }    printf("\n");}int main(){    int u,v,c,m;    while(~scanf("%d%d",&n,&m)){        ways.size();        for(int i=0;i<n;++i)            g[i].clear();        while(m--){            scanf("%d%d%d",&u,&v,&c);            build(u-1,v-1,c);            build(v-1,u-1,c);        }        rev_bfs();        bfs();    }    return 0;}

0 0