当前位置: 首页 > news >正文

c++分层最短路(洛谷飞行路线)acwing版

分层最短路算法是在SPFA算法的基础上,将每个点分成若干层,从而使得每个点之间的转移只在同一层次或上下两个相邻层次之间进行,减少了每轮的迭代次数,优化了算法的效率。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>using namespace std;const int MAXN = 10005;
const int MAXM = 100005;
const int INF = 0x3f3f3f3f;struct Edge {int to, nxt, w;
} e[MAXM];int head[MAXN], tot;
int dis[MAXN], vis[MAXN];inline void add(int u, int v, int w) {e[++tot].nxt = head[u];e[tot].to = v;e[tot].w = w;head[u] = tot;
}void spfa(int s) {memset(dis, 0x3f, sizeof(dis));dis[s] = 0;queue<int> q;q.push(s);while (!q.empty()) {int u = q.front();q.pop();vis[u] = false;for (int i = head[u]; i; i = e[i].nxt) {int v = e[i].to;if (dis[v] > dis[u] + e[i].w) {dis[v] = dis[u] + e[i].w;if (!vis[v]) {q.push(v);vis[v] = true;}}}}
}int main() {int n, m;scanf("%d%d", &n, &m);tot = 0;memset(head, 0, sizeof(head));for (int i = 1; i <= m; ++i) {int u, v, w;scanf("%d%d%d", &u, &v, &w);add(u, v, w);}int s, t;scanf("%d%d", &s, &t);spfa(s);printf("%d\n", dis[t]);return 0;
}
int layer[MAXN]; //记录每个点所在的层次
int check_layer[MAXN]; //记录每个点是否在队列中void layer_spfa(int s) {memset(dis, 0x3f, sizeof(dis));dis[s] = 0;layer[s] = 0;queue<int> q;q.push(s);check_layer[s] = true;while (!q.empty()) {int u = q.front();q.pop();check_layer[u] = false;for (int i = head[u]; i; i = e[i].nxt) {int v = e[i].to;if (layer[u] == layer[v]) {if (dis[v] > dis[u] + e[i].w) {dis[v] = dis[u] + e[i].w;if (!check_layer[v]) {q.push(v);check_layer[v] = true;}}} else if (layer[v] > layer[u]) { //分层if (dis[v] > dis[u] + e[i].w) {dis[v] = dis[u] + e[i].w;layer[v] = layer[u] + 1;if (!check_layer[v]) {q.push(v);check_layer[v] = true;}}}}}
}

先看题目:

Alice 和 Bob 现在要乘飞机旅行,他们选择了一家相对便宜的航空公司。该航空公司一共在 nn 个城市设有业务,设这些城市分别标记为 00 到 n-1n−1,一共有 mm 种航线,每种航线连接两个城市,并且航线有一定的价格。

Alice 和 Bob 现在要从一个城市沿着航线到达另一个城市,途中可以进行转机。航空公司对他们这次旅行也推出优惠,他们可以免费在最多 kk 种航线上搭乘飞机。那么 Alice 和 Bob 这次出行最少花费多少?

输入格式

第一行三个整数 n,m,kn,m,k,分别表示城市数,航线数和免费乘坐次数。

接下来一行两个整数 s,ts,t,分别表示他们出行的起点城市编号和终点城市编号。

接下来 mm 行,每行三个整数 a,b,ca,b,c,表示存在一种航线,能从城市 aa 到达城市 bb,或从城市 bb 到达城市 aa,价格为 cc。

输出格式

输出一行一个整数,为最少花费。

输入样例:

5 6 1
0 4
0 1 5
1 2 5
2 3 5
3 4 5
2 3 3
0 2 100

输出样例:

8

先给出具体代码:

#include<cstring>
#include<iostream>
#include<queue>using namespace std;typedef pair<int, int> PII;const int N = 2100010, INF = 0x3f3f3f3f;int n, m, k, s, t;
int dist[N];
int h[N], w[N], e[N], ne[N], idx;
bool st[N];void add(int a, int b, int c)
{e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++;
}void dijkstra(int u)
{memset(dist, INF, sizeof dist);dist[u] = 0;priority_queue<PII, vector<PII>, greater<PII>> heap;heap.push({0, u});while(heap.size()){auto t = heap.top();heap.pop();int ver = t.second ,distance = t.first;if(st[ver]) continue;st[ver]  = true;for(int i = h[ver]; ~i; i = ne[i]){int j = e[i];if(dist[j] > distance + w[i]){dist[j] = distance + w[i];heap.push({dist[j], j});}}}
}int main()
{cin >> n >> m >> k >> s >> t;memset(h, -1, sizeof h);while(m --){int a, b, c;scanf("%d%d%d", &a, &b ,&c);add(a, b, c), add(b, a, c);for(int j = 1; j <= k; j ++){add(j * n + a, j * n + b, c);add(j * n + b, j * n + a, c);add((j - 1) * n + a, j * n + b, 0);add((j - 1) * n + b, j * n + a, 0);}}for(int i = 0; i < k; i ++) add(i * n + t, (i + 1) * n + t, 0);dijkstra(s);printf("%d\n", dist[n * k + t]);return 0;
}

1:解释数据:2≤n≤10^4,1≤m≤10^5,0≤k≤10,0≤s, t, a, b<n, a != b, 0≤c≤10^3

本来数据最大值是m,双向边开两倍就可以,但是这里是分层建图,最多有十层,所以要再乘以十

2:初始化h数组

3、加边,下面的是分层建图

建图:从0到k层建k+1张图

           各层之间从上到下建边花费为0

            为防止使用小于k次权力就到达终点,在每层的终点间建花费为0的边连起来

4、dijkstra堆优化版的模板

5、答案输出:到k层的终点为答案

http://www.lryc.cn/news/167202.html

相关文章:

  • Python bs4 BeautifulSoup库使用记录
  • Jmeter系列-插件安装(5)
  • spring aop源码解析
  • 使用Unity的Input.GetAxis(““)控制物体移动、旋转
  • 【CSS】画个三角形或圆形或环
  • AI项目六:基于YOLOV5的CPU版本部署openvino
  • 记录YDLidar驱动包交叉编译时出现的一点问题
  • 嵌入式学习笔记(32)S5PV210的向量中断控制器
  • linux下安装qt、qt触摸屏校准tslib
  • C++之unordered_map,unordered_set模拟实现
  • React Router,常用API有哪些?
  • JVM类加载和双亲委派机制
  • P-MVSNet ICCV-2019 学习笔记总结 译文 深度学习三维重建
  • vueshowpdf 移动端pdf文件预览
  • C#根据excel文件中的表头创建数据库表
  • js通过xpath定位元素并且操作元素以下拉框select为例
  • 数据类型
  • vue 模板应用
  • Golang教程与Gin教程合集,入门到实战
  • 国家网络安全周 | 天空卫士荣获“2023网络安全优秀创新成果大赛优胜奖”
  • Swift学习笔记一(Array篇)
  • C++项目实战——基于多设计模式下的同步异步日志系统-②-前置知识补充-不定参函数
  • C++使用Boost库加入UDP组播时程序崩溃
  • 华为HCIA(四)
  • Qt --- Day01
  • 24.98万起,新一代AITO问界M7值得买吗?
  • Java毕业设计 SSM SpringBoot 水果蔬菜商城
  • 前端JS中的异步编程与Promise
  • Pytorch Advanced(二) Variational Auto-Encoder
  • Flask 使用 JWT(三)flask-jwt-extended