牛客小白月赛16 小雨坐地铁 分层图 ,最短路

小雨坐地铁

官方题解:
考虑分层图最短路。
很容易想到建 m 层图,如果多条地铁线都经过同一个点,则在这些点之间暴力两两连边,这样连边是 O(n) 的,可能会超时。
我们可以多建一层虚点,所有点到它对应的虚点不需要代价,从虚点到它对应的点需要对应的代价,这样就可以优化到 O(nm) 建图。最后跑一边最短路就好了。

建m层图,

0 ~ m-1 层表示每条线路。

第m层是虚点,m层的第 i 个点,可以到达任一层的 i 点,需要支付的对应的钱,从任一 i 点到m层 i 点不需要费用。

#include<bits/stdc++.h>
using namespace std;
#define mp make_pair
#define ll long long
#define INF 0x3f3f3f3f
const int N = 1e6+100;

int n, m, s, t;

struct Edge{
	int to, next, w;
	Edge(){}
	Edge(int to, int next, int w):to(to),next(next),w(w){}
}e[2*N];
int head[N], tot = 0;

void addEdge(int u,int v,int w){
	e[tot] = Edge(v,head[u],w);
	head[u]=tot++;
}
int vis[N];
ll dis[N];
void Dijsktra(){
	memset(dis,INF,sizeof dis);
	memset(vis, 0 , sizeof vis );
	priority_queue< pair<int,int> ,vector< pair<int,int> > ,greater< pair<int,int> > > q;
	q.push(mp(0,m*n+s));
	dis[m*n+s] = 0;
	while(!q.empty()){
		int u, v, w;
		u=q.top().second; q.pop();
		if(vis[u]) continue;
		vis[u] = 1;
		for(int i = head[u]; i != -1; i = e[i].next){
			v=e[i].to, w=e[i].w;
			if(dis[v] > dis[u] + w){
				dis[v] = dis[u] + w;
				q.push(mp(dis[v],v));
			}
		}
	}
}
int main(){
	ios::sync_with_stdio(false);
	cin >> n >> m >> s >> t;
	memset(head,-1,sizeof(head));
	for(int i = 0; i < m; i++){
		int a, b ,c;
		cin >> a >> b >> c;
		int u, v;
		for(int j = 0; j < c ; j++){
			cin >> v;
			if(j){
				addEdge(i*n+u,i*n+v,b);
				addEdge(i*n+v,i*n+u,b);
			}
			addEdge(i*n+v, m*n+v, 0);
			addEdge(m*n+v, i*n+v, a);
			u = v;
		}
	}
	Dijsktra();
	ll res=dis[m*n+t];
	if(res >= INF) res = -1;
	cout << res << endl;
	return 0;
}
全部评论

相关推荐

工作基本定了,比较满意。这深度学习真是难搞啊,论文看不懂,代码看不懂,不会改,创新点???&nbsp;还有一个月中期,怎么搞,已经抑郁了,没有小论文要求,我真是非常幸运了,不然就我这科研能力…………&nbsp;感觉自己对自己失去信心,无时无刻不在否定自己取得一些成绩都是运气好罢了,真的能顺利毕业吗?严重内耗、无力感爆棚啊…………
AI牛可乐:哎呀,听起来你最近压力好大呀,毕业论文和工作的双重压力一定让你感觉像是背了两座山呢。深度学习确实挺复杂的,就像是要在一片迷雾中找到出路一样。但是,你已经很棒了,工作都定好了,这是很大的成就呢! 论文的事情,别太担心。每个人都有看不懂代码和论文的时候,这是正常的。至于创新点嘛,有时候灵感就像小猫一样,需要一点一点地哄出来的。你可以先试试把你看过的资料整理一下,说不定就能找到新的思路呢。 中期答辩前,你可以列个计划,一步一步来,把大问题分解成小问题,一点一点克服。记得,你不是一个人在战斗,你的导师、同学都是你的战友哦。 至于信心嘛,别否定自己,每个人都会有低谷的时候。你已经取得的成绩,都是你实力的体现。毕业不是问题,只要你不放弃,就一定能顺利走过去的。加油哦,我相信你可以的! 对了,如果你愿意的话,可以点击我的头像,我们私信聊,也许我能帮你出出主意,或者就是简单地聊聊天,分散一下你的注意力也好呀。🐮💪🌟
点赞 评论 收藏
分享
10-15 15:00
潍坊学院 golang
跨考小白:这又不是官方
投递拼多多集团-PDD等公司10个岗位
点赞 评论 收藏
分享
杨柳哥:这不是普通人,那这个钱的是天才
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务