【C++动态规划 图论】3243. 新增道路查询后的最短距离 I|1567

本文涉及知识点

打开打包代码的方法兼述单元测试
C++动态规划
C++图论

LeetCode3243. 新增道路查询后的最短距离 I

给你一个整数 n 和一个二维整数数组 queries。

有 n 个城市,编号从 0 到 n - 1。初始时,每个城市 i 都有一条单向道路通往城市 i + 1( 0 <= i < n - 1)。

queriesi = ui, vi 表示新建一条从城市 ui 到城市 vi 的单向道路。每次查询后,你需要找到从城市 0 到城市 n - 1 的最短路径的长度。

返回一个数组 answer,对于范围 0, queries.length - 1 中的每个 i,answeri 是处理完前 i + 1 个查询后,从城市 0 到城市 n - 1 的最短路径的长度。

示例 1:

输入: n = 5, queries = \[2, 4, 0, 2, 0, 4]

输出: 3, 2, 1

解释:
新增一条从 2 到 4 的道路后,从 0 到 4 的最短路径长度为 3。

新增一条从 0 到 2 的道路后,从 0 到 4 的最短路径长度为 2。

新增一条从 0 到 4 的道路后,从 0 到 4 的最短路径长度为 1。

示例 2:

输入: n = 4, queries = \[0, 3, 0, 2]

输出: 1, 1

解释:

新增一条从 0 到 3 的道路后,从 0 到 3 的最短路径长度为 1。

新增一条从 0 到 2 的道路后,从 0 到 3 的最短路径长度仍为 1。

提示:

3 <= n <= 500

1 <= queries.length <= 500

queriesi.length == 2

0 <= queriesi0 < queriesi1 < n

1 < queriesi1 - queriesi0

查询中没有重复的道路。

图论 最短路

注意 : 无论是初始路线,还是新增加的路线,都是终点>起点。 → \rightarrow → 经过的城市一定是升序。

动态规划的状态表示

dpi表示到达i的最少路径。空间复杂度:O(n)

动态规划的填表顺序

令新增加的边是i1->i2。

则 i = i2 ; i > N; i++

动态规划的转移方程

MinSelf(dpi2,dpi1+1)

for(next in neiBoi)

MinSelf(dpnext,dpi+1)

新增一条的时间复杂度:O(m),m是边数,本题是n+q。
总时间复杂度:O(q(n+q))

动态规划的初始化

dp0=0,其它INT_MAX/2。

先计算初始各边,再增加查询各边。

增加完边后,刷新dpi...封装成函数。也可以进一步简化,更新整个dp。

动态规划的返回值

dp.back

代码

核心代码

cpp 复制代码
class Solution {
		public:
			vector<int> shortestDistanceAfterQueries(int n, vector<vector<int>>& queries) {
				vector<vector<int>> neiBo(n);
				for (int i = 0; i + 1 < n; i++) {
					neiBo[i].emplace_back(i + 1);
				}
				vector<int> ans;
				for (const auto& v : queries) {
					neiBo[v[0]].emplace_back(v[1]);
					vector<int> dp(n, INT_MAX / 2);
					dp[0] = 0;
					for (int i = 0; i < n; i++) {
						for (const auto& j : neiBo[i]) {
							dp[j] = min(dp[j] , dp[i] + 1);
						}
					}
					ans.emplace_back(dp.back());
				}
				return ans;
			}
		};

代码

核心代码

cpp 复制代码
int n;
		vector<vector<int>> queries;
		TEST_METHOD(TestMethod11)
		{
			n = 5, queries = { {2, 4}, {0, 2}, {0, 4} };
			auto res = Solution().shortestDistanceAfterQueries(n, queries);
			AssertEx({ 3,2,1 }, res);
		}
		TEST_METHOD(TestMethod12)
		{
			n = 4, queries = { {0, 3}, {0, 2} };
			auto res = Solution().shortestDistanceAfterQueries(n, queries);
			AssertEx({ 1,1 }, res);
		}

单元测试

cpp 复制代码
int n;
		vector<vector<int>> queries;
		TEST_METHOD(TestMethod11)
		{
			n = 5, queries = { {2, 4}, {0, 2}, {0, 4} };
			auto res = Solution().shortestDistanceAfterQueries(n, queries);
			AssertEx({ 3,2,1 }, res);
		}
		TEST_METHOD(TestMethod12)
		{
			n = 4, queries = { {0, 3}, {0, 2} };
			auto res = Solution().shortestDistanceAfterQueries(n, queries);
			AssertEx({ 1,1 }, res);
		}

扩展阅读

我想对大家说的话
工作中遇到的问题,可以按类别查阅鄙人的算法文章,请点击《算法与数据汇总》。
学习算法:按章节学习《喜缺全书算法册》,大量的题目和测试用例,打包下载。重视操作
有效学习:明确的目标 及时的反馈 拉伸区(难度合适) 专注
闻缺陷则喜(喜缺)是一个美好的愿望,早发现问题,早修改问题,给老板节约钱。
子墨子言之:事无终始,无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙,那算法就是他的是睛
失败+反思=成功 成功+反思=成功

视频课程

先学简单的课程,请移步CSDN学院,听白银讲师(也就是鄙人)的讲解。
https://edu.csdn.net/course/detail/38771

如何你想快速形成战斗了,为老板分忧,请学习C#入职培训、C++入职培训等课程
https://edu.csdn.net/lecturer/6176

测试环境

操作系统:win7 开发环境: VS2019 C++17

或者 操作系统:win10 开发环境: VS2022 C++17

如无特殊说明,本算法用**C++**实现。

相关推荐
用户805533698036 小时前
不止三件套:QObject 属性系统全关键字与运行时反射!
c++·qt
To_OC16 小时前
LC 207 课程表:刚学图论那会儿,我连这是拓扑排序都没看出来
javascript·算法·leetcode
To_OC16 小时前
LC 208 实现 Trie 前缀树:曾被名字劝退,写完发现是送分题
javascript·算法·leetcode
BadBadBad__AK18 小时前
线段树维护区间 k 次方和
c++·数学·算法·stl
卷无止境1 天前
Eigen 库如何借助 OpenMP 加速计算
c++·后端
_清歌1 天前
DSpark 深度解读:DeepSeek-V4 如何用「半自回归」把推理速度提升 85%
算法
统计实现局1 天前
SVD 的三步走:双对角化、Givens 收敛、排序
算法
躬行见万象1 天前
《VLA 系列》UniLab 强化训练 | G1 机器人 |复现
算法
统计实现局1 天前
对称不定分解(Bunch-Kaufman):为什么 Cholesky 不够用
算法