理想路径(UVA1599 图的双向广度优先搜索)

内容来自陈小玉老师算法训练营
题目:
给一个n个点m条边(2≤n≤100000, 1≤m≤200000)的无向图,每条边上都涂有一种颜色(用1到1000000000表示)。求从结点1到结点n的一条路径,使得经过的边数尽量少,在此前提下,经过边的颜色序列的字典序最小。一对结点间可能有多条边,一条边可能连接两个相同结点。输入保证至少存在一条节点1到n的路径。

输入:输入共m+1行。第一行包含两个整数n和m。之后的m行,每行都含三个整数ai、bi、ci,表示ai、bi之间有一条颜色为ci的路径。

输出:输出共两行,第一行包含正整数k,表示节点1到n至少需要经过k条边,第二行包含k个由空格隔开的正整数,表示节点1到n依次经过的边的颜色。

输入样例:
4 6
1 2 1
1 3 2
3 4 3
2 3 1
2 4 4
3 1 1

输出样例
2
1 3

输出样例解释:节点1到4至少经过两条边:1->3,颜色为1(最后输入的那条边);3->4,颜色为3。数据范围:2<=n<=105,1<=m<=105,1<=ci<=109,对于任意i∈[1,m],都有1<=ai,bi<=n。对于两个长度为k的序列a和b,若存在i∈[1,k]使ai<bi,且对于任意j∈[1,i)都有aj=bj,则a<b。

题解:本题求节点1到n的最短距离,在此前提下,色号序最小。可以先求解最短距离,然后考察色号。因为在从节点1出发的多条边中,并不知道哪条边是最短路径上的边,所以无法确定最小色号。

算法设计:从节点n反向广度优先遍历标高,节点1的高度正好为从节点1到节点n的最短距离。(广度优先搜索,只要第一次到目的地了,就是最优的,就是最短距离)

从节点1正向广度优先遍历,沿着高度减1的方向遍历,找色号小的点,如果多个点的色号最小,则考察下一个色号那个最小,直到n点结束。

完美图解
1、根据样例创建图,然后从节点n反向广度优先遍历标高,节点1的高度为2,即节点1到n的最短距离为2,输出2。
在这里插入图片描述
2、从节点1正向广度优先遍历,沿着高度减1的方向遍历,找边上色号小的邻接点,节点1到2的色号为1,节点1到3的色号也为1,节点1到3的另一条道路色号为2,最小色后为1,输出1。目前无法确定选择那条边,因此将都有可能走的两个邻接点2和3都入队暂存。
在这里插入图片描述
3、从节点2和3出发,沿着高度减1的方向遍历,找边上色号小的邻接点,节点2到4的色号为4,节点3到4的色号为3,最小色号为3,输出3。
在这里插入图片描述
代码:

#include<bits/stdc++.h>
using namespace std;
const int N=100000+5,M=200000+5,inf=0x7fffffff;
int n,m,cnt;
int head[N],dis[N];
bool vis[N];
queue<int>q1,q2,q3;
struct Edge{
	int to,c,next;
}e[M];

void add(int u,int v,int c){//添加一条边 
// 	e[++cnt].to=v;
//	e[cnt].c=c;
//	e[cnt].next=head[u];
	e[++cnt]=(Edge){v,c,head[u]};
 	head[u]=cnt;
}

void bfs1(){//逆向标高求最短距离 
	int u,v;
	memset(vis,false,sizeof(vis));
	dis[n]=0;
	q1.push(n);
	vis[n]=true;
	while(!q1.empty()){
		u=q1.front();
		q1.pop();
		vis[u]=true;
		for(int i=head[u];i;i=e[i].next){
			v=e[i].to;
			if(vis[v])
				continue;
			dis[v]=dis[u]+1;
			q1.push(v);
			vis[v]=true;
		}
	}
}
/*
正向求最小色号序列
队列q1:保存节点号
队列q2:保存色号
队列q3:保存色号最小的边关联的邻接点号
*/
void bfs2(){
	int u,v,minc,c;
	bool first=true;
	memset(vis,false,sizeof(vis));
	vis[1]=true;
	for(int i=head[1];i;i=e[i].next)//1号结点所有邻接点 
		if(dis[e[i].to]==dis[1]-1){//高度减1的邻接点
			q1.push(e[i].to);
			q2.push(e[i].c);
		}
	while(!q1.empty()){
		minc=inf;
		while(!q1.empty()){
			v=q1.front();
			q1.pop();
			c=q2.front();
			q2.pop();
			if(c<minc){
				while(!q3.empty())//发现更小队列清空 
					q3.pop();
				minc=c;
			}
			if(c==minc)
				q3.push(v);
		}
		if(first)
			first=false;
		else
			cout<<" ";
		cout<<minc;
		while(!q3.empty()){//所有等于最小色号的结点 
			u=q3.front();
			q3.pop();
			if(vis[u])
				continue;
			vis[u]=true;
			for(int i=head[u];i;i=e[i].next){//扩展每一个结点 
				v=e[i].to;
				if(dis[v]==dis[u]-1){
					q1.push(v);
					q2.push(e[i].c);
				}
			}
		}		
	}
}

int main(){
	int u,v,c;
	while(cin>>n>>m){
		memset(head,0,sizeof(head));
		cnt=0;
		for(int i=1;i<=m;i++){
			cin>>u>>v>>c;
			add(u,v,c);
			add(v,u,c); 
		}
		bfs1();
		cout<<dis[1]<<endl;
		bfs2(); 
		cout<<endl;
	 } 
	return 0;
}

注:如下图所示,按邻接表顺序遍历边时,节点1到2,节点1到3色号均最小,那么q3队列暂存2、3两个节点,后来发现有比它们更小的色号(色号为2),将q3立即清空,保存节点5。

在这里插入图片描述

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值