目录

一、题目

1、题目描述

2、输入输出

2.1输入

2.2输出

3、原题链接

二、解题报告

1、思路分析

2、复杂度

3、代码详解


一、题目

1、题目描述

2、输入输出

2.1输入

2.2输出

3、原题链接

1076D - Edge Deletion


二、解题报告

1、思路分析

考虑dijkstra实际上是利用贪心算法求了一棵最短路径树,详见:最短路合集,Dijkstra,堆优化Dijkstra,BellmanFord,SPFA,Floyd,附完整代码及OJ链接-CSDN博客

最短路径树中有n - 1条边,如果k > n - 1,那么我们保留n - 1条树边即可

否则,保留离源点最近的k个结点的前驱边

2、复杂度

时间复杂度: O(MlogM + nlogn)空间复杂度:O(N + M)

3、代码详解

 ​
#include <bits/stdc++.h>
#include <ranges>

using i64 = long long;
using i32 = unsigned int;
using u64 = unsigned long long;
using i128 = __int128;

constexpr int inf32 = 1E9 + 7;
constexpr i64 inf64 = 1E18 + 7;
constexpr int P = 1'000'000'007;

void solve() {
	int n, m, k;
	std::cin >> n >> m >> k;

	std::vector<std::vector<int>> g(n);
	std::vector<std::array<int, 2>> edges;
	
	for (int i = 0, u, v, w; i < m; ++ i) {
		std::cin >> u >> v >> w;
		-- u, -- v;
		g[u].push_back(edges.size());
		edges.push_back({v, w});
		g[v].push_back(edges.size());
		edges.push_back({u, w});	
	}

	std::vector<i64> dst(n, inf64);
	std::vector<int> pre(n, -1);

	std::priority_queue<std::pair<i64, int>, std::vector<std::pair<i64, int>>, std::greater<std::pair<i64, int>>> pq;

	pq.emplace(dst[0] = 0, 0);

	while (pq.size()) {
		auto [d, u] = pq.top();
		pq.pop();

		if (d > dst[u]) continue;

		for (int i : g[u]) {
			auto [v, w] = edges[i];
			if (d + w < dst[v]) {
				pre[v] = i;
				pq.emplace(dst[v] = d + w, v);
			}
		}
	}

	int ans = std::min(k, n - 1);

	std::cout << ans << '\n';

	std::vector<int> st(n - 1);
	std::iota(st.begin(), st.end(), 1);

	std::ranges::sort(st, [&dst](int u, int v) -> bool{
		return dst[u] < dst[v];
	});

	for (int i = 0; i < ans; ++ i) {
		std::cout << pre[st[i]] / 2 + 1 << " \n"[i + 1 == ans];
	}
}

auto FIO = []{
	std::ios::sync_with_stdio(false);
	std::cin.tie(nullptr);
	std::cout.tie(nullptr);
	return 0;
}();

int main () {
	#ifdef DEBUG
		freopen("in.txt", "r", stdin);
		freopen("out.txt", "w", stdout);
	#endif
	
	int T = 1;
	// std::cin >> T;
	while (T --)
		solve();

	return 0;
}

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐